单军华
2016-12-09 1a11fb042ac01b8c045d48e6ee3abbf153bf1c36
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
// This source code is dual-licensed under the Mozilla Public License ("MPL"),
// version 1.1 and the Apache License ("ASL"), version 2.0.
//
// The ASL v2.0:
//
// ---------------------------------------------------------------------------
// Copyright 2016 Pivotal Software, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ---------------------------------------------------------------------------
//
// The MPL v1.1:
//
// ---------------------------------------------------------------------------
// The contents of this file are subject to the Mozilla Public License
// Version 1.1 (the "License"); you may not use this file except in
// compliance with the License. You may obtain a copy of the License at
// https://www.mozilla.org/MPL/
//
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
// License for the specific language governing rights and limitations
// under the License.
//
// The Original Code is RabbitMQ
//
// The Initial Developer of the Original Code is Pivotal Software, Inc.
// All Rights Reserved.
//
// Alternatively, the contents of this file may be used under the terms
// of the Apache Standard license (the "ASL License"), in which case the
// provisions of the ASL License are applicable instead of those
// above. If you wish to allow use of your version of this file only
// under the terms of the ASL License and not to allow others to use
// your version of this file under the MPL, indicate your decision by
// deleting the provisions above and replace them with the notice and
// other provisions required by the ASL License. If you do not delete
// the provisions above, a recipient may use your version of this file
// under either the MPL or the ASL License.
// ---------------------------------------------------------------------------
 
#import "RMQQueue.h"
#import "RMQMethods.h"
#import "RMQConnection.h"
#import "RMQBasicProperties.h"
#import "RMQChannel.h"
 
@interface RMQQueue ()
@property (nonatomic, copy, readwrite) NSString *name;
@property (nonatomic, readwrite) RMQQueueDeclareOptions options;
@property (nonatomic, readwrite) RMQTable *arguments;
@property (nonatomic, readwrite) id <RMQChannel> channel;
@end
 
@implementation RMQQueue
 
- (instancetype)initWithName:(NSString *)name
                     options:(RMQQueueDeclareOptions)options
                   arguments:(RMQTable *)arguments
                     channel:(id<RMQChannel>)channel {
   self = [super init];
    if (self) {
        self.name = name;
        self.options = options;
        self.arguments = arguments;
        self.channel = channel;
    }
    return self;
}
 
- (void)bind:(RMQExchange *)exchange
  routingKey:(NSString *)routingKey {
    [self.channel queueBind:self.name exchange:exchange.name routingKey:routingKey];
}
 
- (void)bind:(RMQExchange *)exchange {
    [self bind:exchange routingKey:@""];
}
 
- (void)unbind:(RMQExchange *)exchange
    routingKey:(NSString *)routingKey {
    [self.channel queueUnbind:self.name exchange:exchange.name routingKey:routingKey];
}
 
- (void)unbind:(RMQExchange *)exchange {
    [self unbind:exchange routingKey:@""];
}
 
- (void)delete:(RMQQueueDeleteOptions)options {
    [self.channel queueDelete:self.name options:options];
}
 
- (void)delete {
    [self delete:RMQQueueDeleteNoOptions];
}
 
- (NSNumber *)publish:(NSData *)data
           properties:(NSArray<RMQValue<RMQBasicValue> *> *)properties
              options:(RMQBasicPublishOptions)options {
    return [self.channel basicPublish:data
                           routingKey:self.name
                             exchange:@""
                           properties:properties
                              options:options];
}
 
- (NSNumber *)publish:(NSData *)body
           persistent:(BOOL)isPersistent
              options:(RMQBasicPublishOptions)options {
    NSMutableArray *properties = [NSMutableArray new];
    if (isPersistent) {
        [properties addObject:[[RMQBasicDeliveryMode alloc] init:2]];
    }
    return [self.channel basicPublish:body
                           routingKey:self.name
                             exchange:@""
                           properties:properties
                              options:options];
}
 
- (NSNumber *)publish:(NSData *)body
           persistent:(BOOL)isPersistent {
    return [self publish:body persistent:isPersistent options:RMQBasicPublishNoOptions];
}
 
- (NSNumber *)publish:(NSData *)body {
    return [self publish:body persistent:NO];
}
 
- (void)pop:(RMQConsumerDeliveryHandler)handler {
    [self.channel basicGet:self.name
                   options:RMQBasicGetNoOptions
         completionHandler:handler];
}
 
- (RMQConsumer *)subscribe:(RMQBasicConsumeOptions)options
                   handler:(RMQConsumerDeliveryHandler)handler {
    return [self.channel basicConsume:self.name
                              options:options
                              handler:handler];
}
 
- (RMQConsumer *)subscribe:(RMQConsumerDeliveryHandler)handler {
    return [self subscribe:RMQBasicConsumeNoAck
                   handler:handler];
}
 
@end