rivetkit 3.0.0-preview.4 → 3.0.0-preview.5

package/dist/tsup/{chunk-C45DIKGB.js → chunk-EOUH7MP7.js}

@@ -3,13 +3,13 @@ import {
   deserializeActorKey,
   serializeActorKey
 } from "./chunk-A2CF6PFH.js";
-import {
-  importWebSocket,
-  logger
-} from "./chunk-IPDSE33U.js";
 import {
   AsyncMutex
 } from "./chunk-JTHHCZCZ.js";
+import {
+  importWebSocket,
+  logger
+} from "./chunk-SJUGJ5DT.js";
 import {
   EncodingSchema,
   assertUnreachable,
@@ -36,7 +36,7 @@ import {
   serializeWithEncoding,
   stringifyError,
   uint8ArrayToBase64
-} from "./chunk-MAUVZIXL.js";
+} from "./chunk-KKSIGBUN.js";
 import {
   INTERNAL_ERROR_CODE,
   RivetError,
@@ -2266,6 +2266,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",
@@ -4020,6 +4028,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 = [];
@@ -4141,6 +4151,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;
     logger().debug({ msg: "action", name: opts.name, args: opts.args });
     const actionId = this.#actionIdCounter;
     this.#actionIdCounter += 1;
@@ -4228,6 +4240,7 @@ var ActorConnRaw = class {
     }
   }
   #connectWithRetry() {
+    this.#connectionOpenError = void 0;
     this.#setConnStatus("connecting");
     pRetry(this.#connectAndWait.bind(this), {
       forever: true,
@@ -4270,6 +4283,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;
       }
@@ -4279,7 +4295,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");
@@ -4311,28 +4327,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);
-    logger().warn({
-      msg: "connection did not open in time, retrying",
+    logger().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) {
       logger().debug({
         msg: "failed to close abandoned websocket",
@@ -4597,14 +4608,8 @@ var ActorConnRaw = class {
             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") {
@@ -4735,6 +4740,7 @@ var ActorConnRaw = class {
     }
   }
   #rejectPendingPromises(error, suppressUnhandled) {
+    this.#clearQueuedMessages();
     if (this.#onOpenPromise) {
       if (suppressUnhandled) {
         this.#onOpenPromise.promise.catch(() => {
@@ -6544,4 +6550,4 @@ export {
   updateRunnerConfig,
   RemoteEngineControlClient
 };
-//# sourceMappingURL=chunk-C45DIKGB.js.map
+//# sourceMappingURL=chunk-EOUH7MP7.js.map