OLD | NEW |
| (Empty) |
1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file | |
2 // for details. All rights reserved. Use of this source code is governed by a | |
3 // BSD-style license that can be found in the LICENSE file. | |
4 | |
5 part of dart.async; | |
6 | |
7 /** | |
8 * Wraps an [_EventSink] so it exposes only the [EventSink] interface. | |
9 */ | |
10 class _EventSinkWrapper<T> implements EventSink<T> { | |
11 _EventSink _sink; | |
12 _EventSinkWrapper(this._sink); | |
13 | |
14 void add(T data) { _sink._add(data); } | |
15 void addError(error, [StackTrace stackTrace]) { | |
16 _sink._addError(error, stackTrace); | |
17 } | |
18 void close() { _sink._close(); } | |
19 } | |
20 | |
21 /** | |
22 * A StreamSubscription that pipes data through a sink. | |
23 * | |
24 * The constructor of this class takes a [_SinkMapper] which maps from | |
25 * [EventSink] to [EventSink]. The input to the mapper is the output of | |
26 * the transformation. The returned sink is the transformation's input. | |
27 */ | |
28 class _SinkTransformerStreamSubscription<S, T> | |
29 extends _BufferingStreamSubscription<T> { | |
30 /// The transformer's input sink. | |
31 EventSink<S> _transformerSink; | |
32 | |
33 /// The subscription to the input stream. | |
34 StreamSubscription<S> _subscription; | |
35 | |
36 _SinkTransformerStreamSubscription(Stream<S> source, | |
37 _SinkMapper<S, T> mapper, | |
38 void onData(T data), | |
39 Function onError, | |
40 void onDone(), | |
41 bool cancelOnError) | |
42 // We set the adapter's target only when the user is allowed to send data. | |
43 : super(onData, onError, onDone, cancelOnError) { | |
44 _EventSinkWrapper<T> eventSink = new _EventSinkWrapper<T>(this); | |
45 _transformerSink = mapper(eventSink); | |
46 _subscription = source.listen(_handleData, | |
47 onError: _handleError, | |
48 onDone: _handleDone); | |
49 } | |
50 | |
51 /** Whether this subscription is still subscribed to its source. */ | |
52 bool get _isSubscribed => _subscription != null; | |
53 | |
54 // _EventSink interface. | |
55 | |
56 /** | |
57 * Adds an event to this subscriptions. | |
58 * | |
59 * Contrary to normal [_BufferingStreamSubscription]s we may receive | |
60 * events when the stream is already closed. Report them as state | |
61 * error. | |
62 */ | |
63 void _add(T data) { | |
64 if (_isClosed) { | |
65 throw new StateError("Stream is already closed"); | |
66 } | |
67 super._add(data); | |
68 } | |
69 | |
70 /** | |
71 * Adds an error event to this subscriptions. | |
72 * | |
73 * Contrary to normal [_BufferingStreamSubscription]s we may receive | |
74 * events when the stream is already closed. Report them as state | |
75 * error. | |
76 */ | |
77 void _addError(Object error, StackTrace stackTrace) { | |
78 if (_isClosed) { | |
79 throw new StateError("Stream is already closed"); | |
80 } | |
81 super._addError(error, stackTrace); | |
82 } | |
83 | |
84 /** | |
85 * Adds a close event to this subscriptions. | |
86 * | |
87 * Contrary to normal [_BufferingStreamSubscription]s we may receive | |
88 * events when the stream is already closed. Report them as state | |
89 * error. | |
90 */ | |
91 void _close() { | |
92 if (_isClosed) { | |
93 throw new StateError("Stream is already closed"); | |
94 } | |
95 super._close(); | |
96 } | |
97 | |
98 // _BufferingStreamSubscription hooks. | |
99 | |
100 void _onPause() { | |
101 if (_isSubscribed) _subscription.pause(); | |
102 } | |
103 | |
104 void _onResume() { | |
105 if (_isSubscribed) _subscription.resume(); | |
106 } | |
107 | |
108 Future _onCancel() { | |
109 if (_isSubscribed) { | |
110 StreamSubscription subscription = _subscription; | |
111 _subscription = null; | |
112 subscription.cancel(); | |
113 } | |
114 return null; | |
115 } | |
116 | |
117 void _handleData(S data) { | |
118 try { | |
119 _transformerSink.add(data); | |
120 } catch (e, s) { | |
121 _addError(e, s); | |
122 } | |
123 } | |
124 | |
125 void _handleError(error, [stackTrace]) { | |
126 try { | |
127 _transformerSink.addError(error, stackTrace); | |
128 } catch (e, s) { | |
129 if (identical(e, error)) { | |
130 _addError(error, stackTrace); | |
131 } else { | |
132 _addError(e, s); | |
133 } | |
134 } | |
135 } | |
136 | |
137 void _handleDone() { | |
138 try { | |
139 _subscription = null; | |
140 _transformerSink.close(); | |
141 } catch (e, s) { | |
142 _addError(e, s); | |
143 } | |
144 } | |
145 } | |
146 | |
147 | |
148 typedef EventSink<S> _SinkMapper<S, T>(EventSink<T> output); | |
149 | |
150 /** | |
151 * A StreamTransformer for Sink-mappers. | |
152 * | |
153 * A Sink-mapper takes an [EventSink] (its output) and returns another | |
154 * EventSink (its input). | |
155 * | |
156 * Note that this class can be `const`. | |
157 */ | |
158 class _StreamSinkTransformer<S, T> implements StreamTransformer<S, T> { | |
159 final _SinkMapper<S, T> _sinkMapper; | |
160 const _StreamSinkTransformer(this._sinkMapper); | |
161 | |
162 Stream<T> bind(Stream<S> stream) | |
163 => new _BoundSinkStream<S, T>(stream, _sinkMapper); | |
164 } | |
165 | |
166 /** | |
167 * The result of binding a StreamTransformer for Sink-mappers. | |
168 * | |
169 * It contains the bound Stream and the sink-mapper. Only when the user starts | |
170 * listening to this stream is the sink-mapper invoked. The result is used | |
171 * to create a StreamSubscription that transforms events. | |
172 */ | |
173 class _BoundSinkStream<S, T> extends Stream<T> { | |
174 final _SinkMapper<S, T> _sinkMapper; | |
175 final Stream<S> _stream; | |
176 | |
177 bool get isBroadcast => _stream.isBroadcast; | |
178 | |
179 _BoundSinkStream(this._stream, this._sinkMapper); | |
180 | |
181 StreamSubscription<T> listen(void onData(T event), | |
182 { Function onError, | |
183 void onDone(), | |
184 bool cancelOnError }) { | |
185 cancelOnError = identical(true, cancelOnError); | |
186 StreamSubscription<T> subscription = | |
187 new _SinkTransformerStreamSubscription<S, T>( | |
188 _stream, _sinkMapper, onData, onError, onDone, cancelOnError); | |
189 return subscription; | |
190 } | |
191 } | |
192 | |
193 /// Data-handler coming from [StreamTransformer.fromHandlers]. | |
194 typedef void _TransformDataHandler<S, T>(S data, EventSink<T> sink); | |
195 /// Error-handler coming from [StreamTransformer.fromHandlers]. | |
196 typedef void _TransformErrorHandler<T>( | |
197 Object error, StackTrace stackTrace, EventSink<T> sink); | |
198 /// Done-handler coming from [StreamTransformer.fromHandlers]. | |
199 typedef void _TransformDoneHandler<T>(EventSink<T> sink); | |
200 | |
201 /** | |
202 * Wraps handlers (from [StreamTransformer.fromHandlers]) into an `EventSink`. | |
203 * | |
204 * This way we can reuse the code from [_StreamSinkTransformer]. | |
205 */ | |
206 class _HandlerEventSink<S, T> implements EventSink<S> { | |
207 final _TransformDataHandler<S, T> _handleData; | |
208 final _TransformErrorHandler<T> _handleError; | |
209 final _TransformDoneHandler<T> _handleDone; | |
210 | |
211 /// The output sink where the handlers should send their data into. | |
212 final EventSink<T> _sink; | |
213 | |
214 _HandlerEventSink(this._handleData, this._handleError, this._handleDone, | |
215 this._sink); | |
216 | |
217 void add(S data) { _handleData(data, _sink); } | |
218 void addError(Object error, [StackTrace stackTrace]) { | |
219 _handleError(error, stackTrace, _sink); | |
220 } | |
221 void close() { _handleDone(_sink); } | |
222 } | |
223 | |
224 /** | |
225 * A StreamTransformer that transformers events with the given handlers. | |
226 * | |
227 * Note that this transformer can only be used once. | |
228 */ | |
229 class _StreamHandlerTransformer<S, T> extends _StreamSinkTransformer<S, T> { | |
230 | |
231 _StreamHandlerTransformer({ | |
232 void handleData(S data, EventSink<T> sink), | |
233 void handleError(Object error, StackTrace stackTrace, EventSink<T> sink), | |
234 void handleDone(EventSink<T> sink)}) | |
235 : super((EventSink<T> outputSink) { | |
236 if (handleData == null) handleData = _defaultHandleData; | |
237 if (handleError == null) handleError = _defaultHandleError; | |
238 if (handleDone == null) handleDone = _defaultHandleDone; | |
239 return new _HandlerEventSink<S, T>( | |
240 handleData, handleError, handleDone, outputSink); | |
241 }); | |
242 | |
243 Stream<T> bind(Stream<S> stream) { | |
244 return super.bind(stream); | |
245 } | |
246 | |
247 /** Default data handler forwards all data. */ | |
248 static void _defaultHandleData(var data, EventSink sink) { | |
249 sink.add(data); | |
250 } | |
251 | |
252 /** Default error handler forwards all errors. */ | |
253 static void _defaultHandleError(error, StackTrace stackTrace, | |
254 EventSink sink) { | |
255 sink.addError(error, stackTrace); | |
256 } | |
257 | |
258 /** Default done handler forwards done. */ | |
259 static void _defaultHandleDone(EventSink sink) { | |
260 sink.close(); | |
261 } | |
262 } | |
263 | |
264 /// A closure mapping a stream and cancelOnError to a StreamSubscription. | |
265 typedef StreamSubscription<T> _SubscriptionTransformer<S, T>( | |
266 Stream<S> stream, bool cancelOnError); | |
267 | |
268 /** | |
269 * A [StreamTransformer] that minimizes the number of additional classes. | |
270 * | |
271 * Instead of implementing three classes: a [StreamTransformer], a [Stream] | |
272 * (as the result of a `bind` call) and a [StreamSubscription] (which does the | |
273 * actual work), this class only requires a fincution that is invoked when the | |
274 * last bit (the subscription) of the transformer-workflow is needed. | |
275 * | |
276 * The given transformer function maps from Stream and cancelOnError to a | |
277 * `StreamSubscription`. As such it can also act on `cancel` events, making it | |
278 * fully general. | |
279 */ | |
280 class _StreamSubscriptionTransformer<S, T> implements StreamTransformer<S, T> { | |
281 final _SubscriptionTransformer<S, T> _transformer; | |
282 | |
283 const _StreamSubscriptionTransformer(this._transformer); | |
284 | |
285 Stream<T> bind(Stream<S> stream) => | |
286 new _BoundSubscriptionStream<S, T>(stream, _transformer); | |
287 } | |
288 | |
289 /** | |
290 * A stream transformed by a [_StreamSubscriptionTransformer]. | |
291 * | |
292 * When this stream is listened to it invokes the [_transformer] function with | |
293 * the stored [_stream]. Usually the transformer starts listening at this | |
294 * moment. | |
295 */ | |
296 class _BoundSubscriptionStream<S, T> extends Stream<T> { | |
297 final _SubscriptionTransformer<S, T> _transformer; | |
298 final Stream<S> _stream; | |
299 | |
300 _BoundSubscriptionStream(this._stream, this._transformer); | |
301 | |
302 StreamSubscription<T> listen(void onData(T event), | |
303 { Function onError, | |
304 void onDone(), | |
305 bool cancelOnError }) { | |
306 cancelOnError = identical(true, cancelOnError); | |
307 StreamSubscription<T> result = _transformer(_stream, cancelOnError); | |
308 result.onData(onData); | |
309 result.onError(onError); | |
310 result.onDone(onDone); | |
311 return result; | |
312 } | |
313 } | |
OLD | NEW |