Add StreamController.multiplex constructor.

sdk/lib/async/stream_controller.dart

 // Copyright (c) 2012, 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.
 
 part of dart.async;
 
 // -------------------------------------------------------------------
 // Controller for creating and adding events to a stream.
 // -------------------------------------------------------------------
 
@@ skipping 28 matching lines @@
  * Whether to invoke a callback depends only on the state before and after
  * a stream action, for example firing an event. If the state changes multiple
  * times during the action, and then ends up in the same state as before, no
  * callback is performed.
  *
  * If listeners are added after the stream has completed (sent a "done" event),
  * the listeners will be sent a "done" event eventually, but they won't affect
  * the stream at all, and won't trigger callbacks. From the controller's point
  * of view, the stream is completely inert when has completed.
  */
-class StreamController<T> extends EventSink<T> {
+abstract class StreamController<T> implements EventSink<T> {
+  /** The stream that this controller is controlling. */
+  Stream<T> get stream;
+
+  /**
+   * A controller with a [stream] that supports only one single subscriber.
+   *
+   * The controller will buffer all incoming events until the subscriber is
+   * registered.
+   *
+   * The [onPause] function is called when the stream becomes
+   * paused. [onResume] is called when the stream resumed.
+   *
+   * The [onListen] callback is called when the stream
+   * receives its listener and [onCancel] when the listener ends
+   * its subscription.
+   *
+   * If the stream is canceled before the controller needs new data the
+   * [onResume] call might not be executed.
+   */
+  factory StreamController({void onListen(),
+                            void onPause(),
+                            void onResume(),
+                            void onCancel()})
+      => new _StreamControllerImpl<T>(onListen, onPause, onResume, onCancel);
+
+  /**
+   * A controller where [stream] creates new stream each time it is read.
+   *
+   * The controller distributes any events to all currently subscribed streams.
+   *
+   * The [onListen] callback is called when the first listener is subscribed,
+   * and the [onCancel] is called when there is no longer any active listeners.
+   * If a listener is added again later, after the [onCancel] was called,
+   * the [onListen] will be called again.
+   */
+  factory StreamController.multiplex({void onListen(), void onCancel()}) {
+    return new _MultiplexStreamController<T>(onListen, onCancel);
+  }
+
+  /**
+   * Returns a view of this object that only exposes the [EventSink] interface.
+   */
+  EventSink<T> get sink;
+}
+
+
+abstract class _StreamControllerLifecycle<T> {
+  void _recordListen(StreamSubscription<T> subscription) {}
+  void _recordPause(StreamSubscription<T> subscription) {}
+  void _recordResume(StreamSubscription<T> subscription) {}
+  void _recordCancel(StreamSubscription<T> subscription) {}
+}
+
+/**
+ * Default implementation of [StreamController].
+ *
+ * Controls a stream that only supports a single controller.
+ */
+class _StreamControllerImpl<T> implements StreamController<T>,
+                                          _StreamControllerLifecycle<T> {
   static const int _STATE_OPEN = 0;
   static const int _STATE_CANCELLED = 1;
   static const int _STATE_CLOSED = 2;
 
   final _NotificationHandler _onListen;
   final _NotificationHandler _onPause;
   final _NotificationHandler _onResume;
   final _NotificationHandler _onCancel;
   _StreamImpl<T> _stream;
 
   // An active subscription on the stream, or null if no subscripton is active.
   _ControllerSubscription<T> _subscription;
 
   // Whether we have sent a "done" event.
   int _state = _STATE_OPEN;
 
   // Events added to the stream before it has an active subscription.
   _PendingEvents _pendingEvents = null;
 
-  /**
-   * A controller with a [stream] that supports only one single subscriber.
-   *
-   * The controller will buffer all incoming events until the subscriber is
-   * registered.
-   *
-   * The [onPause] function is called when the stream becomes
-   * paused. [onResume] is called when the stream resumed.
-   *
-   * The [onListen] callback is called when the stream
-   * receives its listener and [onCancel] when the listener ends
-   * its subscription.
-   *
-   * If the stream is canceled before the controller needs new data the
-   * [onResume] call might not be executed.
-   */
-  StreamController({void onListen(),
-                    void onPause(),
-                    void onResume(),
-                    void onCancel()})
-      : _onListen = onListen,
-        _onPause = onPause,
-        _onResume = onResume,
-        _onCancel = onCancel {
+  _StreamControllerImpl(this._onListen,
+                        this._onPause,
+                        this._onResume,
+                        this._onCancel) {
     _stream = new _ControllerStream<T>(this);
   }
 
   Stream<T> get stream => _stream;
 
   /**
    * Returns a view of this object that only exposes the [EventSink] interface.
    */
   EventSink<T> get sink => new _EventSinkView<T>(this);
 
@@ skipping 70 matching lines @@
 
   void _addPendingEvent(_DelayedEvent event) {
     if (_isCancelled) return;
     _StreamImplEvents events = _pendingEvents;
     if (events == null) {
       events = _pendingEvents = new _StreamImplEvents();
     }
     events.add(event);
   }
 
-  void _recordListen(_BufferingStreamSubscription subscription) {
+  void _recordListen(StreamSubscription<T> subscription) {

[floitsch, 2013/05/24 20:51:42]

Why not keep the type?

[Lasse Reichstein Nielsen, 2013/05/27 08:04:12]

True, it is true for this subclass, and I avoid a cast below.

+    _BufferingStreamSubscription bufferingSubscription = subscription;
     assert(_subscription == null);
-    _subscription = subscription;
-    subscription._setPendingEvents(_pendingEvents);
+    _subscription = bufferingSubscription;
+    bufferingSubscription._setPendingEvents(_pendingEvents);
     _pendingEvents = null;
-    subscription._guardCallback(() {
+    bufferingSubscription._guardCallback(() {
       _runGuarded(_onListen);
     });
   }
 
-  void _recordCancel() {
+  void _recordCancel(StreamSubscription<T> subscription) {
     _subscription = null;

[floitsch, 2013/05/24 20:51:42]

assert that the subscription is the same?

[Lasse Reichstein Nielsen, 2013/05/27 08:04:12]

Done.

     _state |= _STATE_CANCELLED;
     _runGuarded(_onCancel);
   }
 
-  void _recordPause() {
+  void _recordPause(StreamSubscription<T> subscription) {
     _runGuarded(_onPause);
   }
 
-  void _recordResume() {
+  void _recordResume(StreamSubscription<T> subscription) {
     _runGuarded(_onResume);
   }
 }
 
 typedef void _NotificationHandler();
 
 void _runGuarded(_NotificationHandler notificationHandler) {
   if (notificationHandler == null) return;
   try {
     notificationHandler();
   } catch (e, s) {
     _throwDelayed(e, s);
   }
 }
 
 class _ControllerStream<T> extends _StreamImpl<T> {
-  StreamController _controller;
+  _StreamControllerLifecycle<T> _controller;
   bool _hasListener = false;
 
   _ControllerStream(this._controller);
 
   StreamSubscription<T> _createSubscription(
       void onData(T data),
       void onError(Object error),
       void onDone(),
       bool cancelOnError) {
     if (_hasListener) {
       throw new StateError("The stream has already been listened to.");
     }
     _hasListener = true;
     return new _ControllerSubscription<T>(
         _controller, onData, onError, onDone, cancelOnError);
   }
 
   void _onListen(_BufferingStreamSubscription subscription) {
     _controller._recordListen(subscription);
   }
 }
 
 class _ControllerSubscription<T> extends _BufferingStreamSubscription<T> {
-  final StreamController _controller;
+  final _StreamControllerLifecycle<T> _controller;
 
-  _ControllerSubscription(StreamController controller,
+  _ControllerSubscription(this._controller,
                           void onData(T data),
                           void onError(Object error),
                           void onDone(),
                           bool cancelOnError)
-      : _controller = controller,
-        super(onData, onError, onDone, cancelOnError);
+      : super(onData, onError, onDone, cancelOnError);
 
   void _onCancel() {
-    _controller._recordCancel();
+    _controller._recordCancel(this);
   }
 
   void _onPause() {
-    _controller._recordPause();
+    _controller._recordPause(this);
   }
 
   void _onResume() {
-    _controller._recordResume();
+    _controller._recordResume(this);
   }
 }
+
+class _MultiplexStreamController<T> implements StreamController<T>,
+                                               _StreamControllerLifecycle<T> {
+  final _NotificationHandler _onListen;
+  final _NotificationHandler _onCancel;
+  // TODO(lrn): Make a more efficient implementation of these subscriptions,
+  // e.g., the traditional double-linked list with concurrent add and remove
+  // while firing.
+  Set<_BufferingStreamSubscription<T>> _streams;
+
+  _MultiplexStreamController(this._onListen, this._onCancel)
+      : _streams = new Set<_BufferingStreamSubscription<T>>();
+
+  // StreamController interface.
+
+  Stream<T> get stream => new _ControllerStream<T>(this);
+
+  EventSink<T> get sink => new _EventSinkView<T>(this);
+
+  // _StreamControllerLifecycle interface.
+
+  void _recordListen(_BufferingStreamSubscription<T> subscription) {
+    bool isFirst = _streams.isEmpty;
+    _streams.add(subscription);
+    if (isFirst) {
+      _runGuarded(_onListen);
+    }
+  }
+
+  void _recordCancel(_BufferingStreamSubscription<T> subscription) {
+    _streams.remove(subscription);
+    if (_streams.isEmpty) {
+      _runGuarded(_onCancel);
+    }
+  }
+
+  void _recordPause(StreamSubscription<T> subscription) {}
+  void _recordResume(StreamSubscription<T> subscription) {}
+
+  // EventSink interface.
+
+  void add(T data) {
+    if (_streams.isEmpty) return;
+    _forEachListener((_BufferingStreamSubscription<T> subscription) {
+      subscription._add(data);
+    });
+  }
+
+  void addError(Object error) {
+    if (_streams.isEmpty) return;
+    _forEachListener((_BufferingStreamSubscription<T> subscription) {
+      subscription._addError(error);
+    });
+  }
+
+  void close() {
+    if (_streams.isEmpty) return;
+    _forEachListener((_BufferingStreamSubscription<T> subscription) {
+      _streams.remove(subscription);
+      subscription._close();
+    });
+  }
+
+  void _forEachListener(
+      void action(_BufferingStreamSubscription<T> subscription)) {
+    List<_BufferingStreamSubscription<T>> subscriptions = _streams.toList();
+    for (_BufferingStreamSubscription<T> subscription in subscriptions) {
+      if (_streams.contains(subscription)) {
+        action(subscription);
+      }
+    }
+  }
+}
+