rivetkit 3.0.0-preview.3 → 3.0.0-preview.4

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.4",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -5578,6 +5578,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 +5640,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 +5660,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 +5672,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 +5739,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 +5747,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,6 +5820,9 @@ var ActorConnRaw = class {
             this.#actorResolutionState,
             this.#driver
           );
+          if (this.#websocket !== socket || this.#abandonedSockets.has(socket)) {
+            return;
+          }
           if (schedulingError) {
             errorToThrow = schedulingError;
           }
@@ -5930,9 +5942,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) {