rivetkit 3.0.0-preview.3 → 3.0.0-preview.5

package/dist/browser/client.js

@@ -230,7 +230,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "3.0.0-preview.3",
+  version: "3.0.0-preview.5",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -4934,6 +4934,14 @@ function classifyActorError(error) {
       }
     );
   }
+  if (error.group === "ups" && error.code === "consumer_closed" || error.group === "depot" && error.code === "request_deadline_exceeded") {
+    return buildLifecycleBoundaryInfo(
+      "reconnect_only",
+      "actor_error",
+      error.message,
+      { group: error.group, code: error.code }
+    );
+  }
   if (error.group === "guard" && isRetryableGuardGatewayHttpError(error.code)) {
     return buildLifecycleBoundaryInfo(
       "request_retry",
@@ -5254,6 +5262,8 @@ var ActorConnRaw = class {
   /* Will be aborted on dispose. */
   #abortController = new AbortController();
   #connStatus = "idle";
+  /** The opening failure that stopped retries. Later actions must fail instead of queueing. */
+  #connectionOpenError;
   #actorId;
   #connId;
   #messageQueue = [];
@@ -5372,6 +5382,8 @@ var ActorConnRaw = class {
         `Invalid action call: expected an options object { name, args }, got ${typeof opts}. Use conn.actionName(...args) for the shorthand API.`
       );
     }
+    if (this.#disposed) throw new ActorConnDisposed();
+    if (this.#connectionOpenError) throw this.#connectionOpenError;
     logger2().debug({ msg: "action", name: opts.name, args: opts.args });
     const actionId = this.#actionIdCounter;
     this.#actionIdCounter += 1;
@@ -5459,6 +5471,7 @@ var ActorConnRaw = class {
     }
   }
   #connectWithRetry() {
+    this.#connectionOpenError = void 0;
     this.#setConnStatus("connecting");
     pRetry2(this.#connectAndWait.bind(this), {
       forever: true,
@@ -5501,6 +5514,9 @@ var ActorConnRaw = class {
       await this.#connectWebSocket();
       await this.#waitForOpen(this.#onOpenPromise.promise);
     } catch (error) {
+      this.#abandonCurrentSocket(
+        isConnectionOpenTimeout(error) ? "Connect timeout" : "Connection opening failed"
+      );
       if (this.#shouldRetryConnectionOpenError(error)) {
         throw error;
       }
@@ -5510,7 +5526,7 @@ var ActorConnRaw = class {
         `Failed to open connection: ${stringifyError(error)}`,
         { error: stringifyError(error) }
       );
-      this.#clearQueuedMessages();
+      this.#connectionOpenError = actorError;
       this.#rejectPendingPromises(actorError, false);
       this.#dispatchActorError(actorError);
       this.#setConnStatus("idle");
@@ -5542,28 +5558,23 @@ var ActorConnRaw = class {
     });
     try {
       await Promise.race([open, timedOut]);
-    } catch (error) {
-      if (isConnectionOpenTimeout(error)) {
-        this.#abandonCurrentSocket();
-      }
-      throw error;
     } finally {
       clearTimeout(timer);
     }
   }
-  #abandonCurrentSocket() {
+  #abandonCurrentSocket(reason) {
     const ws = this.#websocket;
     if (!ws) return;
     this.#websocket = void 0;
     this.#abandonedSockets.add(ws);
-    logger2().warn({
-      msg: "connection did not open in time, retrying",
+    logger2().debug({
+      msg: "closing failed connection attempt",
       connId: this.#connId,
       readyState: ws.readyState,
-      timeoutMs: CONNECT_TIMEOUT_MS
+      reason
     });
     try {
-      ws.close(1e3, "Connect timeout");
+      ws.close(1e3, reason);
     } catch (err) {
       logger2().debug({
         msg: "failed to close abandoned websocket",
@@ -5578,6 +5589,9 @@ var ActorConnRaw = class {
     if (isConnectionOpenTimeout(error)) {
       return true;
     }
+    if (error instanceof RivetError && error.group === "client" && error.code === "connection_open_interrupted") {
+      return true;
+    }
     if (error instanceof RivetError && isInvalidToken(error.group, error.code) && this.#driver.refreshAuthToken && this.#authRetryAvailable) {
       this.#authRetryAvailable = false;
       return true;
@@ -5637,16 +5651,18 @@ var ActorConnRaw = class {
     });
     this.#websocket = ws;
     ws.addEventListener("open", () => {
-      if (this.#abandonedSockets.has(ws)) return;
+      if (this.#websocket !== ws || this.#abandonedSockets.has(ws))
+        return;
       logger2().debug({
         msg: "client websocket open",
         connId: this.#connId
       });
     });
     ws.addEventListener("message", async (ev) => {
-      if (this.#abandonedSockets.has(ws)) return;
+      if (this.#websocket !== ws || this.#abandonedSockets.has(ws))
+        return;
       try {
-        await this.#handleOnMessage(ev.data);
+        await this.#handleOnMessage(ev.data, ws);
       } catch (err) {
         logger2().error({
           msg: "error in websocket message handler",
@@ -5655,7 +5671,8 @@ var ActorConnRaw = class {
       }
     });
     ws.addEventListener("close", async (ev) => {
-      if (this.#abandonedSockets.has(ws)) return;
+      if (this.#websocket !== ws || this.#abandonedSockets.has(ws))
+        return;
       try {
         await this.#handleOnClose(ev);
       } catch (err) {
@@ -5666,7 +5683,8 @@ var ActorConnRaw = class {
       }
     });
     ws.addEventListener("error", () => {
-      if (this.#abandonedSockets.has(ws)) return;
+      if (this.#websocket !== ws || this.#abandonedSockets.has(ws))
+        return;
       try {
         this.#handleOnError();
       } catch (err) {
@@ -5732,7 +5750,7 @@ var ActorConnRaw = class {
     });
   }
   /** Called by the onmessage event from drivers. */
-  async #handleOnMessage(data) {
+  async #handleOnMessage(data, socket) {
     logger2().trace({
       msg: "received message",
       dataType: typeof data,
@@ -5740,6 +5758,8 @@ var ActorConnRaw = class {
       isArrayBuffer: data instanceof ArrayBuffer
     });
     const response = await this.#parseMessage(data);
+    if (this.#websocket !== socket || this.#abandonedSockets.has(socket))
+      return;
     logger2().trace(
       getLogMessage() ? {
         msg: "parsed message",
@@ -5811,18 +5831,15 @@ var ActorConnRaw = class {
             this.#actorResolutionState,
             this.#driver
           );
+          if (this.#websocket !== socket || this.#abandonedSockets.has(socket)) {
+            return;
+          }
           if (schedulingError) {
             errorToThrow = schedulingError;
           }
         }
-        if (this.#onOpenPromise) {
-          this.#onOpenPromise.reject(errorToThrow);
-        }
         this.#invalidateActorIfStale(group, code);
-        for (const [id, inFlight] of this.#actionsInFlight.entries()) {
-          inFlight.reject(errorToThrow);
-          this.#actionsInFlight.delete(id);
-        }
+        this.#rejectPendingPromises(errorToThrow, false);
         this.#dispatchActorError(errorToThrow);
       }
     } else if (response.body.tag === "ActionResponse") {
@@ -5930,9 +5947,12 @@ var ActorConnRaw = class {
         }
         this.#invalidateActorIfStale(group, code);
       } else {
-        error = new Error(
-          `${wasClean ? "Connection closed" : "Connection lost"} (code: ${closeEvent.code}, reason: ${reason})`
-        );
+        const message = `${wasClean ? "Connection closed" : "Connection lost"} (code: ${closeEvent.code}, reason: ${reason})`;
+        error = !wasConnected && closeEvent.code === 1006 ? new RivetError(
+          "client",
+          "connection_open_interrupted",
+          message
+        ) : new Error(message);
       }
       this.#rejectPendingPromises(error, false);
       if (error instanceof RivetError) {
@@ -5948,6 +5968,7 @@ var ActorConnRaw = class {
     }
   }
   #rejectPendingPromises(error, suppressUnhandled) {
+    this.#clearQueuedMessages();
     if (this.#onOpenPromise) {
       if (suppressUnhandled) {
         this.#onOpenPromise.promise.catch(() => {