cross-tab-worker-databus 0.6.0 → 0.8.0

package/dist/cjs/index.cjs

@@ -562,6 +562,7 @@ var WorkerClusterRuntime = class {
       projectedLoads.set(owner.workerId, (projectedLoads.get(owner.workerId) ?? owner.load) + 1);
       const generation = (previous?.generation ?? 0) + 1;
       this.writeRoute(topicKey, owner, previous?.workerId, generation);
+      this.handlers.onDiagnostic?.({ operation: "route_migration", topic });
       this.flushStorage();
       this.handlers.onControl("UNSUBSCRIBE", topic);
       this.sendRouteReleased(owner.workerId, topic, topicKey, generation);
@@ -573,20 +574,20 @@ var WorkerClusterRuntime = class {
    * posted to a remote owner, so the caller can surface the failure instead of
    * silently dropping the publication.
    */
-  publish(topic, data) {
+  publish(topic, data, messageId) {
     const topicKey = this.rememberTopic(topic);
     if (this.assignedTopics.has(topicKey)) {
-      return this.sendControl(this.workerId, "PUBLISH", topic, topicKey, data);
+      return this.sendControl(this.workerId, "PUBLISH", topic, topicKey, data, messageId);
     }
     for (const pattern of this.assignedTopics.values()) {
       if (pattern !== topic && topicMatchesPattern(pattern, topic)) {
-        return this.sendControl(this.workerId, "PUBLISH", topic, topicKey, data);
+        return this.sendControl(this.workerId, "PUBLISH", topic, topicKey, data, messageId);
       }
     }
     const workers = this.readWorkers();
     const route = this.readRoute(topicKey);
     const target = this.routeOwnerIsLive(route, workers) ? route?.workerId ?? this.workerId : this.workerId;
-    return this.sendControl(target, "PUBLISH", topic, topicKey, data);
+    return this.sendControl(target, "PUBLISH", topic, topicKey, data, messageId);
   }
   /** True when `route` exists and its owner worker is among `workers`.
    * Shared by subscribe (skip re-assignment) and publish (route to owner).
@@ -708,7 +709,11 @@ var WorkerClusterRuntime = class {
       default:
         break;
     }
-    this.handlers.onControl(message.action, message.topic, message.data);
+    if (message.messageId === void 0) {
+      this.handlers.onControl(message.action, message.topic, message.data);
+    } else {
+      this.handlers.onControl(message.action, message.topic, message.data, message.messageId);
+    }
     if (message.action !== "PUBLISH") this.updateLoad();
   }
   /**
@@ -825,7 +830,7 @@ var WorkerClusterRuntime = class {
    * Local execution updates the assignment map and route synchronously, bypassing
    * the BroadcastChannel latency.
    */
-  sendControl(targetWorkerId, action, topic, topicKey, data) {
+  sendControl(targetWorkerId, action, topic, topicKey, data, messageId) {
     if (targetWorkerId === this.workerId) {
       switch (action) {
         case "SUBSCRIBE":
@@ -839,7 +844,8 @@ var WorkerClusterRuntime = class {
         default:
           break;
       }
-      this.handlers.onControl(action, topic, data);
+      if (messageId === void 0) this.handlers.onControl(action, topic, data);
+      else this.handlers.onControl(action, topic, data, messageId);
       if (action !== "PUBLISH") this.updateLoad();
       return true;
     }
@@ -850,7 +856,8 @@ var WorkerClusterRuntime = class {
       action,
       topic,
       topicKey,
-      ...data === void 0 ? {} : { data }
+      ...data === void 0 ? {} : { data },
+      ...messageId === void 0 ? {} : { messageId }
     });
   }
   /** Post a message on the BroadcastChannel. Returns false on postMessage failure. */
@@ -946,6 +953,8 @@ var WorkerClusterRuntime = class {
       ...route,
       confirmedAt: this.environment.now()
     });
+    const topic = this.knownTopics.get(topicKey);
+    if (topic) this.handlers.onDiagnostic?.({ operation: "route_ack", topic });
   }
   /** Remove routes whose topic has no subscribers and whose TTL has expired. */
   cleanupOrphanedRoutes(workers) {
@@ -1239,6 +1248,10 @@ var CrossTabDataBus = class _CrossTabDataBus {
   initialConfig;
   hasInitialConfig;
   trace;
+  dedupMaxEntries;
+  dedupTtlMs;
+  dedupEnabled;
+  seenMessageIds = /* @__PURE__ */ new Map();
   activeConfig;
   status = "disconnected";
   started = false;
@@ -1277,17 +1290,26 @@ var CrossTabDataBus = class _CrossTabDataBus {
     this.replayBuffers = replay ? /* @__PURE__ */ new Map() : null;
     this.replayPersistence = replay?.persistence ?? null;
     this.replayHydration = this.hydrateReplay();
-    const { autoStart, initialConfig, trace, transport, ...clusterOptions } = options;
+    const { autoStart, initialConfig, trace, transport, dedup, ...clusterOptions } = options;
     this.transport = transport;
     this.initialConfig = initialConfig;
     this.hasInitialConfig = "initialConfig" in options;
     this.trace = new DataBusTraceReporter(trace);
+    this.dedupMaxEntries = dedup?.maxEntries ?? 1e3;
+    this.dedupTtlMs = dedup?.ttlMs ?? 6e4;
+    this.dedupEnabled = dedup !== void 0;
+    if (!Number.isSafeInteger(this.dedupMaxEntries) || this.dedupMaxEntries <= 0) {
+      throw new TypeError("dedup.maxEntries must be a positive safe integer.");
+    }
+    if (!Number.isFinite(this.dedupTtlMs) || this.dedupTtlMs <= 0) {
+      throw new TypeError("dedup.ttlMs must be a positive finite number.");
+    }
     this.cluster = new WorkerClusterRuntime({
       ...clusterOptions,
       handlers: {
         // The cluster calls `onControl` when it receives a SUBSCRIBE/UNSUBSCRIBE/PUBLISH
         // control message — meaning the owning Worker has delegated the action to us.
-        onControl: (action, topic, data) => {
+        onControl: (action, topic, data, messageId) => {
           switch (action) {
             case "SUBSCRIBE":
               if (this.subscribeTransport(topic)) this.traceSubscription("subscribe", topic);
@@ -1296,7 +1318,7 @@ var CrossTabDataBus = class _CrossTabDataBus {
               if (this.unsubscribeTransport(topic)) this.traceSubscription("unsubscribe", topic);
               break;
             case "PUBLISH":
-              this.runTransport(() => this.transport.publish(topic, data));
+              this.runTransport(() => this.transport.publish(topic, data, messageId ? { messageId } : void 0));
               break;
             default:
               break;
@@ -1321,6 +1343,9 @@ var CrossTabDataBus = class _CrossTabDataBus {
           this.trace.event({ type: "lifecycle", action: "resume" });
           this.trace.start();
           this.resumeTransport();
+        },
+        onDiagnostic: (event) => {
+          this.trace.event({ type: "reliability", ...event });
         }
       }
     });
@@ -1467,12 +1492,27 @@ var CrossTabDataBus = class _CrossTabDataBus {
     if (handlers.size > 0) return;
     this.topicHandlers.delete(topic);
     this.replayBuffers?.delete(topic);
+    if (this.replayPersistence?.clearTopic) {
+      void this.replayPersistence.clearTopic(topic).catch((error) => this.reportError(error));
+    }
     this.cluster.unsubscribe(topic);
   }
+  /** Clear all in-memory replay buffers and, when supported, durable history. */
+  async clearReplay() {
+    this.replayBuffers?.clear();
+    if (this.replayPersistence?.clear) {
+      try {
+        await this.replayPersistence.clear();
+      } catch (error) {
+        this.reportError(error);
+        throw error;
+      }
+    }
+  }
   /** Publish a message to `topic`. The owning Worker delivers it to the transport. */
-  publish(topic, data) {
+  publish(topic, data, options) {
     this.ensureStarted();
-    if (!this.cluster.publish(topic, data)) {
+    if (!this.cluster.publish(topic, data, options?.messageId)) {
       this.reportError(
         new Error("Failed to send the publish control message to the owning worker.")
       );
@@ -1522,6 +1562,7 @@ var CrossTabDataBus = class _CrossTabDataBus {
       else await this.transport.stop();
     } finally {
       this.transportSubscribedTopics.clear();
+      this.seenMessageIds.clear();
       this.started = false;
       this.stopping = false;
       this.suspended = false;
@@ -1539,6 +1580,7 @@ var CrossTabDataBus = class _CrossTabDataBus {
    * and dispatches locally.
    */
   handleTransportMessage(message) {
+    if (this.isDuplicate(message)) return;
     this.trace.recordReceived(message.topic);
     if (!this.cluster.isAssigned(message.topic)) {
       this.trace.recordDiscarded(message.topic);
@@ -1551,6 +1593,24 @@ var CrossTabDataBus = class _CrossTabDataBus {
     }
     this.trace.recordDiscarded(message.topic);
   }
+  isDuplicate(message) {
+    if (!this.dedupEnabled || !message.messageId) return false;
+    const now = Date.now();
+    for (const [id, timestamp] of this.seenMessageIds) {
+      if (now - timestamp > this.dedupTtlMs) this.seenMessageIds.delete(id);
+    }
+    if (this.seenMessageIds.has(message.messageId)) {
+      this.trace.event({ type: "reliability", operation: "dedup_suppressed", topic: message.topic });
+      return true;
+    }
+    this.seenMessageIds.set(message.messageId, now);
+    while (this.seenMessageIds.size > this.dedupMaxEntries) {
+      const oldest = this.seenMessageIds.keys().next().value;
+      if (oldest === void 0) break;
+      this.seenMessageIds.delete(oldest);
+    }
+    return false;
+  }
   /** Deliver a message to every local handler registered for its topic,
    * plus every handler registered with a wildcard subscription that matches
    * (e.g. a handler subscribed to "chat.*" receives "chat.room.1"). */
@@ -1635,6 +1695,7 @@ var CrossTabDataBus = class _CrossTabDataBus {
       const now = Date.now();
       if (now - this.lastRecoveryAt >= _CrossTabDataBus.RECOVERY_COOLDOWN_MS) {
         this.lastRecoveryAt = now;
+        this.trace.event({ type: "reliability", operation: "transport_recovery", attempt: 1 });
         setTimeout(() => {
           if (this.stopping || !this.started || this.suspended) return;
           if (this.status !== "error") return;
@@ -1834,6 +1895,24 @@ function createIndexedDbReplayPersistence(options) {
         transaction.oncomplete = () => resolve();
         transaction.onerror = () => reject(transaction.error ?? new Error("Failed to persist replay history."));
       });
+    },
+    async clear() {
+      const db = await open();
+      await new Promise((resolve, reject) => {
+        const transaction = db.transaction(storeName, "readwrite");
+        transaction.objectStore(storeName).clear();
+        transaction.oncomplete = () => resolve();
+        transaction.onerror = () => reject(transaction.error ?? new Error("Failed to clear replay history."));
+      });
+    },
+    async clearTopic(topic) {
+      const db = await open();
+      await new Promise((resolve, reject) => {
+        const transaction = db.transaction(storeName, "readwrite");
+        transaction.objectStore(storeName).delete(topic);
+        transaction.oncomplete = () => resolve();
+        transaction.onerror = () => reject(transaction.error ?? new Error("Failed to clear topic replay history."));
+      });
     }
   };
 }
@@ -1885,12 +1964,12 @@ var WebSocketTransport = class {
     this.sendFrame({ op: "unsubscribe", topic });
   }
   /** Publish `data` to `topic` as a JSON frame. Requires an open socket. */
-  publish(topic, data) {
+  publish(topic, data, options) {
     if (data instanceof ArrayBuffer) {
-      this.sendBinaryFrame(topic, data);
+      this.sendBinaryFrame(topic, data, options?.messageId);
       return;
     }
-    this.sendFrame({ op: "publish", topic, data });
+    this.sendFrame({ op: "publish", topic, data, ...options?.messageId ? { messageId: options.messageId } : {} });
   }
   /** Close the socket and drop all state. Safe to call multiple times. */
   stop() {
@@ -1910,7 +1989,11 @@ var WebSocketTransport = class {
     }
     this.socket.send(JSON.stringify(payload));
   }
-  sendBinaryFrame(topic, data) {
+  sendBinaryFrame(topic, data, messageId) {
+    if (messageId) {
+      this.sendFrame({ op: "publish", topic, data: Array.from(new Uint8Array(data)), messageId });
+      return;
+    }
     if (this.socket?.readyState !== WS_OPEN) {
       this.handlers?.onError(new Error('WebSocket is not open; dropped "publish" frame.'));
       return;
@@ -1952,7 +2035,12 @@ var WebSocketTransport = class {
     if (!parsed || typeof parsed !== "object") return;
     const frame = parsed;
     if (typeof frame.topic !== "string") return;
-    this.handlers?.onMessage({ topic: frame.topic, data: frame.data });
+    const message = {
+      topic: frame.topic,
+      data: frame.data,
+      ...typeof frame.messageId === "string" ? { messageId: frame.messageId } : {}
+    };
+    this.handlers?.onMessage(message);
   }
 };
 function defaultWebSocketFactory(url, protocols) {