|
17 | 17 | NSString const *FLEEndOfEventStream = @"EndOfEventStream";
|
18 | 18 |
|
19 | 19 | @implementation FLEEventChannel {
|
20 |
| - NSString *_name; |
21 |
| - __weak id<FLEBinaryMessenger> _messenger; |
22 |
| - id<FLEMethodCodec> _codec; |
| 20 | + NSString *_name; |
| 21 | + __weak id<FLEBinaryMessenger> _messenger; |
| 22 | + id<FLEMethodCodec> _codec; |
23 | 23 | }
|
24 | 24 |
|
25 | 25 | + (instancetype)eventChannelWithName:(NSString *)name
|
26 | 26 | binaryMessenger:(id<FLEBinaryMessenger>)messenger
|
27 | 27 | codec:(id<FLEMethodCodec>)codec {
|
28 |
| - return [[[self class] alloc] initWithName:name binaryMessenger:messenger codec:codec]; |
| 28 | + return [[[self class] alloc] initWithName:name binaryMessenger:messenger codec:codec]; |
29 | 29 | }
|
30 | 30 |
|
31 | 31 | - (instancetype)initWithName:(NSString *)name
|
32 | 32 | binaryMessenger:(id<FLEBinaryMessenger>)messenger
|
33 | 33 | codec:(id<FLEMethodCodec>)codec {
|
34 |
| - self = [super init]; |
35 |
| - if (self) { |
36 |
| - _name = [name copy]; |
37 |
| - _messenger = messenger; |
38 |
| - _codec = codec; |
39 |
| - } |
40 |
| - return self; |
| 34 | + self = [super init]; |
| 35 | + if (self) { |
| 36 | + _name = [name copy]; |
| 37 | + _messenger = messenger; |
| 38 | + _codec = codec; |
| 39 | + } |
| 40 | + return self; |
41 | 41 | }
|
42 | 42 |
|
43 | 43 | - (void)setStreamHandler:(NSObject<FLEStreamHandler> *)handler {
|
44 |
| - if (!handler) { |
45 |
| - [_messenger setMessageHandlerOnChannel:_name binaryMessageHandler:nil]; |
46 |
| - return; |
47 |
| - } |
| 44 | + if (!handler) { |
| 45 | + [_messenger setMessageHandlerOnChannel:_name binaryMessageHandler:nil]; |
| 46 | + return; |
| 47 | + } |
48 | 48 |
|
49 |
| - id<FLEMethodCodec> codec = _codec; |
50 |
| - __weak id<FLEBinaryMessenger> messenger = _messenger; |
51 |
| - NSString *name = _name; |
52 |
| - FLEEventSink eventSink = ^(id event) { |
53 |
| - if (event == FLEEndOfEventStream) |
54 |
| - [messenger sendOnChannel:name message:nil]; |
55 |
| - else if ([event isKindOfClass:[FLEMethodError class]]) |
56 |
| - [messenger sendOnChannel:name |
57 |
| - message:[codec encodeErrorEnvelope:(FLEMethodError *)event]]; |
58 |
| - else |
59 |
| - [messenger sendOnChannel:name message:[codec encodeSuccessEnvelope:event]]; |
60 |
| - }; |
| 49 | + id<FLEMethodCodec> codec = _codec; |
| 50 | + __weak id<FLEBinaryMessenger> messenger = _messenger; |
| 51 | + NSString *name = _name; |
| 52 | + FLEEventSink eventSink = ^(id event) { |
| 53 | + if (event == FLEEndOfEventStream) { |
| 54 | + [messenger sendOnChannel:name message:nil]; |
| 55 | + } else if ([event isKindOfClass:[FLEMethodError class]]) { |
| 56 | + [messenger sendOnChannel:name message:[codec encodeErrorEnvelope:(FLEMethodError *)event]]; |
| 57 | + } else { |
| 58 | + [messenger sendOnChannel:name message:[codec encodeSuccessEnvelope:event]]; |
| 59 | + } |
| 60 | + }; |
61 | 61 |
|
62 |
| - __block FLEEventSink currentSink = nil; |
63 |
| - FLEBinaryMessageHandler messageHandler = ^(NSData *message, FLEBinaryReply callback) { |
64 |
| - FLEMethodCall* call = [codec decodeMethodCall:message]; |
65 |
| - if ([call.methodName isEqual:@"listen"]) { |
66 |
| - if (currentSink) { |
67 |
| - FLEMethodError* error = [handler onCancelWithArguments:nil]; |
68 |
| - if (error) |
69 |
| - NSLog(@"Failed to cancel existing stream: %@. %@ (%@)", error.code, error.message, |
70 |
| - error.details); |
71 |
| - } |
72 |
| - currentSink = eventSink; |
73 |
| - FLEMethodError *error = [handler onListenWithArguments:call.arguments eventSink:currentSink]; |
74 |
| - if (error) |
75 |
| - callback([codec encodeErrorEnvelope:error]); |
76 |
| - else |
77 |
| - callback([codec encodeSuccessEnvelope:nil]); |
78 |
| - } else if ([call.methodName isEqual:@"cancel"]) { |
79 |
| - if (!currentSink) { |
80 |
| - callback( |
81 |
| - [codec encodeErrorEnvelope:[[FLEMethodError alloc] initWithCode:@"error" |
82 |
| - message:@"No active stream to cancel" |
83 |
| - details:nil]]); |
84 |
| - return; |
85 |
| - } |
86 |
| - currentSink = nil; |
87 |
| - FLEMethodError *error = [handler onCancelWithArguments:call.arguments]; |
88 |
| - if (error) |
89 |
| - callback([codec encodeErrorEnvelope:error]); |
90 |
| - else |
91 |
| - callback([codec encodeSuccessEnvelope:nil]); |
92 |
| - } else { |
93 |
| - callback(nil); |
| 62 | + __block FLEEventSink currentSink = nil; |
| 63 | + FLEBinaryMessageHandler messageHandler = ^(NSData *message, FLEBinaryReply callback) { |
| 64 | + FLEMethodCall *call = [codec decodeMethodCall:message]; |
| 65 | + if ([call.methodName isEqual:@"listen"]) { |
| 66 | + if (currentSink) { |
| 67 | + FLEMethodError *error = [handler onCancelWithArguments:nil]; |
| 68 | + if (error) { |
| 69 | + NSLog(@"Failed to cancel existing stream: %@. %@ (%@)", error.code, error.message, |
| 70 | + error.details); |
94 | 71 | }
|
95 |
| - }; |
96 |
| - [_messenger setMessageHandlerOnChannel:_name binaryMessageHandler:messageHandler]; |
| 72 | + } |
| 73 | + currentSink = eventSink; |
| 74 | + FLEMethodError *error = [handler onListenWithArguments:call.arguments eventSink:currentSink]; |
| 75 | + if (error) { |
| 76 | + callback([codec encodeErrorEnvelope:error]); |
| 77 | + } else { |
| 78 | + callback([codec encodeSuccessEnvelope:nil]); |
| 79 | + } |
| 80 | + } else if ([call.methodName isEqual:@"cancel"]) { |
| 81 | + if (!currentSink) { |
| 82 | + callback([codec |
| 83 | + encodeErrorEnvelope:[[FLEMethodError alloc] initWithCode:@"error" |
| 84 | + message:@"No active stream to cancel" |
| 85 | + details:nil]]); |
| 86 | + return; |
| 87 | + } |
| 88 | + currentSink = nil; |
| 89 | + FLEMethodError *error = [handler onCancelWithArguments:call.arguments]; |
| 90 | + if (error) { |
| 91 | + callback([codec encodeErrorEnvelope:error]); |
| 92 | + } else { |
| 93 | + callback([codec encodeSuccessEnvelope:nil]); |
| 94 | + } |
| 95 | + } else { |
| 96 | + callback(nil); |
| 97 | + } |
| 98 | + }; |
| 99 | + [_messenger setMessageHandlerOnChannel:_name binaryMessageHandler:messageHandler]; |
97 | 100 | }
|
98 | 101 |
|
99 | 102 | @end
|
0 commit comments