@dxos/messaging 0.3.1-main.5e68e9d → 0.3.1-main.b9bb100

package/dist/lib/browser/index.mjs

@@ -16,6 +16,8 @@ var MESSAGE_TIMEOUT = 1e4;
 var __dxlog_file = "/home/runner/work/dxos/dxos/packages/core/mesh/messaging/src/messenger.ts";
 var ReliablePayload = schema.getCodecForType("dxos.mesh.messaging.ReliablePayload");
 var Acknowledgement = schema.getCodecForType("dxos.mesh.messaging.Acknowledgement");
+var ERROR_LIMIT = 5;
+var NETWORK_REBOOT_DELAY = 3e3;
 var RECEIVED_MESSAGES_GC_INTERVAL = 12e4;
 var Messenger = class {
   constructor({ signalManager, retryDelay = 300 }) {
@@ -30,6 +32,8 @@ var Messenger = class {
     */
     this._toClear = new ComplexSet(PublicKey.hash);
     this._closed = true;
+    this._errorCount = 0;
+    this._rebooting = false;
     this._signalManager = signalManager;
     this._retryDelay = retryDelay;
     this.open();
@@ -43,14 +47,14 @@ var Messenger = class {
       id: traceId
     }), {
       F: __dxlog_file,
-      L: 69,
+      L: 74,
       S: this,
       C: (f, a) => f(...a)
     });
     this._ctx = new Context({
       onError: (err) => log.catch(err, void 0, {
         F: __dxlog_file,
-        L: 71,
+        L: 76,
         S: this,
         C: (f, a) => f(...a)
       })
@@ -60,7 +64,7 @@ var Messenger = class {
         from: message.author
       }, {
         F: __dxlog_file,
-        L: 75,
+        L: 80,
         S: this,
         C: (f, a) => f(...a)
       });
@@ -74,7 +78,7 @@ var Messenger = class {
       id: traceId
     }), {
       F: __dxlog_file,
-      L: 90,
+      L: 95,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -89,7 +93,7 @@ var Messenger = class {
   async sendMessage({ author, recipient, payload }) {
     invariant(!this._closed, "Closed", {
       F: __dxlog_file,
-      L: 102,
+      L: 107,
       S: this,
       A: [
         "!this._closed",
@@ -103,7 +107,7 @@ var Messenger = class {
     };
     invariant(!this._onAckCallbacks.has(reliablePayload.messageId), void 0, {
       F: __dxlog_file,
-      L: 109,
+      L: 114,
       S: this,
       A: [
         "!this._onAckCallbacks.has(reliablePayload.messageId!)",
@@ -116,7 +120,7 @@ var Messenger = class {
       recipient
     }, {
       F: __dxlog_file,
-      L: 110,
+      L: 115,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -131,7 +135,7 @@ var Messenger = class {
         messageId: reliablePayload.messageId
       }, {
         F: __dxlog_file,
-        L: 124,
+        L: 129,
         S: this,
         C: (f, a) => f(...a)
       });
@@ -143,21 +147,22 @@ var Messenger = class {
         err
       }, {
         F: __dxlog_file,
-        L: 126,
+        L: 131,
         S: this,
         C: (f, a) => f(...a)
       }));
     }, this._retryDelay);
-    scheduleTask(messageContext, () => {
+    scheduleTask(messageContext, async () => {
       log("message not delivered", {
         messageId: reliablePayload.messageId
       }, {
         F: __dxlog_file,
-        L: 135,
+        L: 140,
         S: this,
         C: (f, a) => f(...a)
       });
       this._onAckCallbacks.delete(reliablePayload.messageId);
+      await this.errorLimitCheck();
       timeoutHit(new ProtocolTimeoutError("signaling message not delivered", new TimeoutError(MESSAGE_TIMEOUT, "Message not delivered")));
       void messageContext.dispose();
     }, MESSAGE_TIMEOUT);
@@ -173,6 +178,54 @@ var Messenger = class {
     });
     return promise;
   }
+  async errorLimitCheck() {
+    log(`checking error limit ${this._errorCount}`, void 0, {
+      F: __dxlog_file,
+      L: 165,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    if (this._errorCount++ > ERROR_LIMIT) {
+      await this.rebootNetwork();
+    }
+  }
+  async rebootNetwork() {
+    if (this._rebooting) {
+      return;
+    }
+    this._rebooting = true;
+    log("rebooting Messenger/SignalManager", void 0, {
+      F: __dxlog_file,
+      L: 177,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    await this.close();
+    await this._signalManager.close();
+    log("pausing", void 0, {
+      F: __dxlog_file,
+      L: 180,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    await new Promise((resolve) => setTimeout(resolve, NETWORK_REBOOT_DELAY));
+    log("done pausing", void 0, {
+      F: __dxlog_file,
+      L: 182,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    await this.open();
+    await this._signalManager.open();
+    log("done rebooting", void 0, {
+      F: __dxlog_file,
+      L: 185,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    this._errorCount = 0;
+    this._rebooting = false;
+  }
   /**
   * Subscribes onMessage function to messages that contains payload with payloadType.
   * @param payloadType if not specified, onMessage will be subscribed to all types of messages.
@@ -180,7 +233,7 @@ var Messenger = class {
   async listen({ peerId, payloadType, onMessage }) {
     invariant(!this._closed, "Closed", {
       F: __dxlog_file,
-      L: 171,
+      L: 204,
       S: this,
       A: [
         "!this._closed",
@@ -244,7 +297,7 @@ var Messenger = class {
   async _handleReliablePayload({ author, recipient, payload }) {
     invariant(payload.type_url === "dxos.mesh.messaging.ReliablePayload", void 0, {
       F: __dxlog_file,
-      L: 232,
+      L: 265,
       S: this,
       A: [
         "payload.type_url === 'dxos.mesh.messaging.ReliablePayload'",
@@ -258,7 +311,7 @@ var Messenger = class {
       messageId: reliablePayload.messageId
     }, {
       F: __dxlog_file,
-      L: 235,
+      L: 268,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -280,7 +333,7 @@ var Messenger = class {
   async _handleAcknowledgement({ payload }) {
     invariant(payload.type_url === "dxos.mesh.messaging.Acknowledgement", void 0, {
       F: __dxlog_file,
-      L: 257,
+      L: 290,
       S: this,
       A: [
         "payload.type_url === 'dxos.mesh.messaging.Acknowledgement'",
@@ -296,7 +349,7 @@ var Messenger = class {
       to: author
     }, {
       F: __dxlog_file,
-      L: 270,
+      L: 303,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -347,7 +400,7 @@ var Messenger = class {
         elapsed
       }, {
         F: __dxlog_file,
-        L: 318,
+        L: 351,
         S: this,
         C: (f, a) => f(...a)
       });
@@ -1331,12 +1384,12 @@ var dec = (payload) => {
 };
 
 // packages/core/mesh/messaging/src/signal-manager/websocket-signal-manager.ts
-import { Event as Event3, sleep as sleep2, synchronized } from "@dxos/async";
+import { Event as Event3, synchronized } from "@dxos/async";
 import { Context as Context4 } from "@dxos/context";
 import { invariant as invariant5 } from "@dxos/invariant";
 import { PublicKey as PublicKey5 } from "@dxos/keys";
 import { log as log5 } from "@dxos/log";
-import { RateLimitExceededError, trace as trace4 } from "@dxos/protocols";
+import { trace as trace4 } from "@dxos/protocols";
 function _ts_decorate(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
@@ -1348,14 +1401,11 @@ function _ts_decorate(decorators, target, key, desc) {
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
 var __dxlog_file5 = "/home/runner/work/dxos/dxos/packages/core/mesh/messaging/src/signal-manager/websocket-signal-manager.ts";
-var MAX_SERVER_FAILURES = 5;
-var WSS_SIGNAL_SERVER_REBOOT_DELAY = 3e3;
 var WebsocketSignalManager = class {
   constructor(_hosts) {
     this._hosts = _hosts;
     this._servers = /* @__PURE__ */ new Map();
     this._opened = false;
-    this.failureCount = /* @__PURE__ */ new Map();
     this.statusChanged = new Event3();
     this.commandTrace = new Event3();
     this.swarmEvent = new Event3();
@@ -1365,7 +1415,7 @@ var WebsocketSignalManager = class {
       hosts: this._hosts
     }, {
       F: __dxlog_file5,
-      L: 46,
+      L: 43,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1376,7 +1426,6 @@ var WebsocketSignalManager = class {
       const server = new SignalClient(host.server, async (message) => this.onMessage.emit(message), async (data) => this.swarmEvent.emit(data));
       server.statusChanged.on(() => this.statusChanged.emit(this.getStatus()));
       this._servers.set(host.server, server);
-      this.failureCount.set(host.server, 0);
       server.commandTrace.on((trace5) => this.commandTrace.emit(trace5));
     }
   }
@@ -1388,7 +1437,7 @@ var WebsocketSignalManager = class {
       hosts: this._hosts
     }, {
       F: __dxlog_file5,
-      L: 69,
+      L: 65,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1396,7 +1445,7 @@ var WebsocketSignalManager = class {
       id: this._instanceId
     }), {
       F: __dxlog_file5,
-      L: 70,
+      L: 66,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1409,7 +1458,7 @@ var WebsocketSignalManager = class {
       id: this._instanceId
     }), {
       F: __dxlog_file5,
-      L: 77,
+      L: 73,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1422,38 +1471,6 @@ var WebsocketSignalManager = class {
     await this._ctx.dispose();
     await Promise.all(Array.from(this._servers.values()).map((server) => server.close()));
   }
-  async restartServer(serverName) {
-    log5("Restarting server", {
-      serverName
-    }, {
-      F: __dxlog_file5,
-      L: 93,
-      S: this,
-      C: (f, a) => f(...a)
-    });
-    invariant5(this._opened, "server already closed", {
-      F: __dxlog_file5,
-      L: 94,
-      S: this,
-      A: [
-        "this._opened",
-        "'server already closed'"
-      ]
-    });
-    const server = this._servers.get(serverName);
-    invariant5(server, "server not found", {
-      F: __dxlog_file5,
-      L: 97,
-      S: this,
-      A: [
-        "server",
-        "'server not found'"
-      ]
-    });
-    await server.close();
-    await sleep2(WSS_SIGNAL_SERVER_REBOOT_DELAY);
-    await server.open();
-  }
   getStatus() {
     return Array.from(this._servers.values()).map((server) => server.getStatus());
   }
@@ -1463,13 +1480,13 @@ var WebsocketSignalManager = class {
       peerId
     }, {
       F: __dxlog_file5,
-      L: 110,
+      L: 93,
       S: this,
       C: (f, a) => f(...a)
     });
     invariant5(this._opened, "Closed", {
       F: __dxlog_file5,
-      L: 111,
+      L: 94,
       S: this,
       A: [
         "this._opened",
@@ -1487,13 +1504,13 @@ var WebsocketSignalManager = class {
       peerId
     }, {
       F: __dxlog_file5,
-      L: 117,
+      L: 100,
       S: this,
       C: (f, a) => f(...a)
     });
     invariant5(this._opened, "Closed", {
       F: __dxlog_file5,
-      L: 118,
+      L: 101,
       S: this,
       A: [
         "this._opened",
@@ -1506,75 +1523,44 @@ var WebsocketSignalManager = class {
     }));
   }
   async sendMessage({ author, recipient, payload }) {
-    log5(`Signal ${recipient.truncate()}`, void 0, {
+    log5(`Signal ${recipient}`, void 0, {
       F: __dxlog_file5,
-      L: 132,
+      L: 115,
       S: this,
       C: (f, a) => f(...a)
     });
     invariant5(this._opened, "Closed", {
       F: __dxlog_file5,
-      L: 133,
+      L: 116,
       S: this,
       A: [
         "this._opened",
         "'Closed'"
       ]
     });
-    void this._forEachServer(async (server, serverName) => {
+    void this._forEachServer(async (server) => {
       void server.sendMessage({
         author,
         recipient,
         payload
-      }).catch((err) => {
-        if (err instanceof RateLimitExceededError) {
-          log5("WSS rate limit exceeded", {
-            err
-          }, {
-            F: __dxlog_file5,
-            L: 138,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-        } else {
-          log5(`error sending to ${serverName}`, {
-            err
-          }, {
-            F: __dxlog_file5,
-            L: 140,
-            S: this,
-            C: (f, a) => f(...a)
-          });
-          void this.checkServerFailure(serverName);
-        }
-      });
-    });
-  }
-  async checkServerFailure(serverName) {
-    const failureCount = this.failureCount.get(serverName) ?? 0;
-    if (failureCount > MAX_SERVER_FAILURES) {
-      log5.warn(`Too many failures sending to ${serverName} (${failureCount} > ${MAX_SERVER_FAILURES}), restarting`, void 0, {
+      }).catch((err) => log5(err, void 0, {
         F: __dxlog_file5,
-        L: 151,
+        L: 119,
         S: this,
         C: (f, a) => f(...a)
-      });
-      await this.restartServer(serverName);
-      this.failureCount.set(serverName, 0);
-      return;
-    }
-    this.failureCount.set(serverName, (this.failureCount.get(serverName) ?? 0) + 1);
+      }));
+    });
   }
   async subscribeMessages(peerId) {
     log5(`Subscribed for message stream peerId=${peerId}`, void 0, {
       F: __dxlog_file5,
-      L: 160,
+      L: 124,
       S: this,
       C: (f, a) => f(...a)
     });
     invariant5(this._opened, "Closed", {
       F: __dxlog_file5,
-      L: 161,
+      L: 125,
       S: this,
       A: [
         "this._opened",
@@ -1586,13 +1572,13 @@ var WebsocketSignalManager = class {
   async unsubscribeMessages(peerId) {
     log5(`Subscribed for message stream peerId=${peerId}`, void 0, {
       F: __dxlog_file5,
-      L: 167,
+      L: 131,
       S: this,
       C: (f, a) => f(...a)
     });
     invariant5(this._opened, "Closed", {
       F: __dxlog_file5,
-      L: 168,
+      L: 132,
       S: this,
       A: [
         "this._opened",
@@ -1605,31 +1591,25 @@ var WebsocketSignalManager = class {
     this._ctx = new Context4({
       onError: (err) => log5.catch(err, void 0, {
         F: __dxlog_file5,
-        L: 175,
+        L: 139,
         S: this,
         C: (f, a) => f(...a)
       })
     });
   }
   async _forEachServer(fn) {
-    return Promise.all(Array.from(this._servers.entries()).map(([serverName, server]) => fn(server, serverName)));
+    return Promise.all(Array.from(this._servers.values()).map(fn));
   }
 };
 _ts_decorate([
   synchronized
 ], WebsocketSignalManager.prototype, "open", null);
-_ts_decorate([
-  synchronized
-], WebsocketSignalManager.prototype, "close", null);
 _ts_decorate([
   synchronized
 ], WebsocketSignalManager.prototype, "join", null);
 _ts_decorate([
   synchronized
 ], WebsocketSignalManager.prototype, "leave", null);
-_ts_decorate([
-  synchronized
-], WebsocketSignalManager.prototype, "checkServerFailure", null);
 export {
   MemorySignalManager,
   MemorySignalManagerContext,