relay-runtime 2.0.0-rc.2 → 5.0.0

package/lib/RelayReplaySubject.js

@@ -0,0 +1,134 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * 
+ * @format
+ */
+'use strict';
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+
+var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
+
+var RelayObservable = require("./RelayObservable");
+
+var invariant = require("fbjs/lib/invariant");
+
+/**
+ * An implementation of a `ReplaySubject` for Relay Observables.
+ *
+ * Records events provided and synchronously plays them back to new subscribers,
+ * as well as forwarding new asynchronous events.
+ */
+var RelayReplaySubject =
+/*#__PURE__*/
+function () {
+  function RelayReplaySubject() {
+    var _this = this;
+
+    (0, _defineProperty2["default"])(this, "_complete", false);
+    (0, _defineProperty2["default"])(this, "_events", []);
+    (0, _defineProperty2["default"])(this, "_sinks", new Set());
+    this._observable = RelayObservable.create(function (sink) {
+      _this._sinks.add(sink);
+
+      var events = _this._events;
+
+      for (var i = 0; i < events.length; i++) {
+        if (sink.closed) {
+          // Bail if an event made the observer unsubscribe.
+          break;
+        }
+
+        var event = events[i];
+
+        switch (event.kind) {
+          case 'complete':
+            sink.complete();
+            break;
+
+          case 'error':
+            sink.error(event.error);
+            break;
+
+          case 'next':
+            sink.next(event.data);
+            break;
+
+          default:
+            event.kind;
+            !false ? process.env.NODE_ENV !== "production" ? invariant(false, 'RelayReplaySubject: Unknown event kind `%s`.', event.kind) : invariant(false) : void 0;
+        }
+      }
+
+      return function () {
+        _this._sinks["delete"](sink);
+      };
+    });
+  }
+
+  var _proto = RelayReplaySubject.prototype;
+
+  _proto.complete = function complete() {
+    if (this._complete === true) {
+      return;
+    }
+
+    this._complete = true;
+
+    this._events.push({
+      kind: 'complete'
+    });
+
+    this._sinks.forEach(function (sink) {
+      return sink.complete();
+    });
+  };
+
+  _proto.error = function error(_error) {
+    if (this._complete === true) {
+      return;
+    }
+
+    this._complete = true;
+
+    this._events.push({
+      kind: 'error',
+      error: _error
+    });
+
+    this._sinks.forEach(function (sink) {
+      return sink.error(_error);
+    });
+  };
+
+  _proto.next = function next(data) {
+    if (this._complete === true) {
+      return;
+    }
+
+    this._events.push({
+      kind: 'next',
+      data: data
+    });
+
+    this._sinks.forEach(function (sink) {
+      return sink.next(data);
+    });
+  };
+
+  _proto.subscribe = function subscribe(observer) {
+    return this._observable.subscribe(observer);
+  };
+
+  _proto.getObserverCount = function getObserverCount() {
+    return this._sinks.size;
+  };
+
+  return RelayReplaySubject;
+}();
+
+module.exports = RelayReplaySubject;