Chromium Code Reviews| Index: lib/src/lazy_stream.dart |
| diff --git a/lib/src/lazy_stream.dart b/lib/src/lazy_stream.dart |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..ce47590927027c69d144e22e7b309532f57b4e7d |
| --- /dev/null |
| +++ b/lib/src/lazy_stream.dart |
| @@ -0,0 +1,48 @@ |
| +// Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file |
| +// for details. All rights reserved. Use of this source code is governed by a |
| +// BSD-style license that can be found in the LICENSE file. |
| + |
| +library async.lazy_stream; |
| + |
| +import "dart:async"; |
| + |
| +import "stream_completer.dart"; |
| + |
| +/// A [Stream] wrapper that forwards to another [Stream] that's initialized |
| +/// lazily. |
| +/// |
| +/// This class allows a concrete `Stream` to be created only once it has a |
| +/// listener. It's useful to wrapping APIs that do expensive computation to |
| +/// produce a `Stream`. |
| +class LazyStream<T> extends Stream<T> { |
| + /// The callback that's called to create the inner stream. |
| + ZoneCallback _callback; |
| + |
| + /// Creates a single-subscription `Stream` that calls [callback] when it gets |
| + /// a listener and forwards to the returned stream. |
| + /// |
| + /// The [callback] may return a `Stream` or a `Future<Stream>`. |
| + LazyStream(callback()) : _callback = callback; |
| + |
| + StreamSubscription<T> listen(void onData(T event), |
| + {Function onError, |
| + void onDone(), |
| + bool cancelOnError}) { |
| + if (_callback == null) { |
| + throw new StateError("Stream has already been listened to."); |
|
Lasse Reichstein Nielsen
2015/12/11 12:33:21
This error is incorrect if you passed a null callb
nweiz
2015/12/14 21:10:18
Got it. Done.
|
| + } |
| + |
| + // Null out the callback before we invoke it to ensure that even while |
| + // running it, this can't be called twice. |
| + var callback = _callback; |
| + _callback = null; |
| + var result = callback(); |
| + |
| + Stream stream = result is Future |
| + ? StreamCompleter.fromFuture(result) |
| + : result; |
| + |
| + return stream.listen(onData, |
| + onError: onError, onDone: onDone, cancelOnError: cancelOnError); |
| + } |
| +} |