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..1878d4cfb30ec3547f9ed36a144ccc00ed23e305 |
--- /dev/null |
+++ b/lib/src/lazy_stream.dart |
@@ -0,0 +1,51 @@ |
+// 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 { |
+ // Explicitly check for null because we null out [_callback] internally. |
+ if (_callback == null) throw new ArgumentError.notNull('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."); |
+ } |
+ |
+ // 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); |
+ } |
+} |