rivetkit 2.3.0-rc.8 → 2.3.0

package/dist/tsup/{chunk-EMFKMVJR.js → chunk-4JDSFJS5.js}

@@ -1,7 +1,7 @@
 import {
   deserializeActorKey,
   serializeActorKey
-} from "./chunk-3YY5S6TV.js";
+} from "./chunk-HXUEHHJF.js";
 import {
   EncodingSchema,
   contentTypeForEncoding,
@@ -12,14 +12,14 @@ import {
   jsonStringifyCompat,
   serializeWithEncoding,
   uint8ArrayToBase64
-} from "./chunk-KY3CERZR.js";
+} from "./chunk-2OTRTA3J.js";
 import {
   AsyncMutex
-} from "./chunk-PCBNKI2J.js";
+} from "./chunk-JZ7TWV65.js";
 import {
   importWebSocket,
   logger
-} from "./chunk-D3T3ZBSY.js";
+} from "./chunk-BEI24WTI.js";
 import {
   assertUnreachable,
   bufferToArrayBuffer,
@@ -37,13 +37,13 @@ import {
   noopNext,
   promiseWithResolvers,
   stringifyError
-} from "./chunk-GBG63SUG.js";
+} from "./chunk-VTTFNQQI.js";
 import {
   INTERNAL_ERROR_CODE,
   RivetError,
   actorNotFound,
   invalidRequest
-} from "./chunk-TMLOKTRB.js";
+} from "./chunk-BRP62GZC.js";
 
 // src/common/actor-router-consts.ts
 var PATH_CONNECT = "/connect";
@@ -2246,7 +2246,7 @@ function classifyActorError(error) {
   return void 0;
 }
 function isRetryableGuardGatewayHttpError(code) {
-  return code === "service_unavailable" || code === "actor_stopped_while_waiting" || code === "tunnel_request_aborted" || code === "tunnel_message_timeout" || code === "tunnel_response_closed" || code === "gateway_response_start_timeout";
+  return code === "service_unavailable" || code === "actor_wake_retries_exceeded" || code === "actor_stopped_while_waiting" || code === "tunnel_request_aborted" || code === "tunnel_message_timeout" || code === "tunnel_response_closed" || code === "gateway_response_start_timeout";
 }
 function classifyTransportError(error) {
   if (error.message.includes("database accessed after actor stopped")) {
@@ -2336,10 +2336,7 @@ async function retryOnLifecycleBoundary(run, opts) {
       if (attempt === maxAttempts - 1) {
         break;
       }
-      const delayMs = Math.min(
-        initialDelayMs * 2 ** attempt,
-        maxDelayMs
-      );
+      const delayMs = Math.min(initialDelayMs * 2 ** attempt, maxDelayMs);
       await waitWithSignal(delayMs, opts == null ? void 0 : opts.signal);
     }
   }
@@ -2543,7 +2540,9 @@ function createQueueSender(senderOptions) {
       }),
       requestToBare: (value) => ({
         name: value.name ?? name,
-        body: bufferToArrayBuffer(encodeCborCompat(value.body)),
+        body: bufferToArrayBuffer(
+          encodeCborCompat(value.body)
+        ),
         wait: value.wait ?? false,
         timeout: value.timeout !== void 0 ? BigInt(value.timeout) : null
       }),
@@ -2782,10 +2781,7 @@ var ActorHandleRaw = class {
             }
           }).send(name, body, options);
         } catch (err) {
-          const { group, code, message, metadata, actor } = deconstructError(
-            err,
-            true
-          );
+          const { group, code, message, metadata, actor } = deconstructError(err, true);
           if (this.#shouldRetryQueueDispatchOverload(
             group,
             code,
@@ -2818,7 +2814,10 @@ var ActorHandleRaw = class {
             await this.#waitForRetryWindow();
             continue;
           }
-          const invalidated = this.#invalidateResolvedActorId(group, code);
+          const invalidated = this.#invalidateResolvedActorId(
+            group,
+            code
+          );
           if (invalidated && attempt < maxAttempts - 1) {
             useQueryTarget = code === "starting" || code === "stopping" || code.startsWith("destroyed_");
             if (useQueryTarget) {
@@ -2826,7 +2825,10 @@ var ActorHandleRaw = class {
             }
             continue;
           }
-          throw new RivetError(group, code, message, { metadata, actor });
+          throw new RivetError(group, code, message, {
+            metadata,
+            actor
+          });
         }
       }
       throw new Error("unreachable queue retry state");
@@ -2906,7 +2908,9 @@ var ActorHandleRaw = class {
             args
           }),
           requestToBare: (args) => ({
-            args: bufferToArrayBuffer(encodeCborCompat(args))
+            args: bufferToArrayBuffer(
+              encodeCborCompat(args)
+            )
           }),
           responseFromJson: (json) => json.output,
           responseFromBare: (bare5) => decodeCborCompat(new Uint8Array(bare5.output))
@@ -2916,10 +2920,7 @@ var ActorHandleRaw = class {
         }
         return output;
       } catch (err) {
-        const { group, code, message, metadata, actor } = deconstructError(
-          err,
-          true
-        );
+        const { group, code, message, metadata, actor } = deconstructError(err, true);
         if (await this.#shouldRetrySchedulingError(
           group,
           code,
@@ -2950,11 +2951,16 @@ var ActorHandleRaw = class {
             { metadata, actor }
           );
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code
+        );
         if (invalidated && attempt < maxAttempts - 1) {
           if (group === "actor" && (code === "starting" || code === "stopping")) {
             useQueryTarget = true;
-            await new Promise((resolve) => setTimeout(resolve, 100));
+            await new Promise(
+              (resolve) => setTimeout(resolve, 100)
+            );
           }
           continue;
         }
@@ -3131,10 +3137,7 @@ var ActorHandleRaw = class {
         }
         return response;
       } catch (err) {
-        const { group, code, message, metadata, actor } = deconstructError(
-          err,
-          true
-        );
+        const { group, code, message, metadata, actor } = deconstructError(err, true);
         if (await this.#shouldRetrySchedulingError(
           group,
           code,
@@ -3157,7 +3160,10 @@ var ActorHandleRaw = class {
           await this.#waitForRetryWindow();
           continue;
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code
+        );
         if (invalidated && attempt < maxAttempts - 1) {
           useQueryTarget = code === "starting" || code === "stopping" || code.startsWith("destroyed_");
           if (useQueryTarget) {
@@ -3257,8 +3263,9 @@ var ActorHandleRaw = class {
       this.#gatewayOptions,
       options
     );
+    const useQueryTarget = isDynamicActorQuery(this.#actorResolutionState);
     const target = await this.#resolveGatewayRequestTarget(
-      false,
+      useQueryTarget,
       gatewayOptions
     );
     return await rawWebSocket(
@@ -3281,7 +3288,9 @@ var ActorHandleRaw = class {
     if ("directId" in target) {
       return target.directId;
     }
-    throw new Error("dynamic actor resolution did not produce a direct actor id");
+    throw new Error(
+      "dynamic actor resolution did not produce a direct actor id"
+    );
   }
   /**
    * Returns the raw URL for routing traffic to the actor.
@@ -3638,7 +3647,6 @@ var ActorConnRaw = class {
    * @protected
    */
   constructor(client, driver, params, getParams, encoding, actorResolutionState, gatewayOptions = {}) {
-    var _a, _b;
     this.#client = client;
     this.#driver = driver;
     this.#params = params;
@@ -3646,6 +3654,9 @@ var ActorConnRaw = class {
     this.#encoding = encoding;
     this.#actorResolutionState = actorResolutionState;
     this.#gatewayOptions = resolveActorGatewayOptions(gatewayOptions);
+    if ("getForId" in actorResolutionState) {
+      this.#actorId = actorResolutionState.getForId.actorId;
+    }
     this.#readyPromise = promiseWithResolvers(
       (reason) => logger().warn({
         msg: "unhandled ready promise rejection",
@@ -3663,9 +3674,7 @@ var ActorConnRaw = class {
         );
       }
     });
-    this.#keepNodeAliveInterval = setInterval(() => {
-    }, 6e4);
-    (_b = (_a = this.#keepNodeAliveInterval).unref) == null ? void 0 : _b.call(_a);
+    this.#keepNodeAliveInterval = setInterval(() => 6e4);
   }
   #clearResolvedActorIdentity() {
     this.#actorId = void 0;
@@ -4073,12 +4082,10 @@ var ActorConnRaw = class {
           );
           return;
         }
-        let errorToThrow = new RivetError(
-          group,
-          code,
-          message,
-          { metadata, actor }
-        );
+        let errorToThrow = new RivetError(group, code, message, {
+          metadata,
+          actor
+        });
         if (isSchedulingError(group, code) && this.#actorId) {
           const schedulingError = await checkForSchedulingError(
             group,
@@ -4966,7 +4973,7 @@ async function sendHttpRequestToGateway(runConfig, gatewayUrl, actorRequest, opt
     if (reqBody.byteLength !== 0) {
       bodyToSend = reqBody;
       guardHeaders.delete("transfer-encoding");
-      guardHeaders.set("content-length", String(bodyToSend.byteLength));
+      guardHeaders.delete("content-length");
     }
   }
   const guardRequest = new Request(gatewayUrl, {
@@ -5365,11 +5372,11 @@ async function lookupMetadataCached(config) {
 }
 
 // src/engine-client/ws-proxy.ts
-async function createWebSocketProxy(c, targetUrl, protocols) {
+async function createWebSocketProxy(_c, targetUrl, protocols) {
   const WebSocket2 = await importWebSocket();
   const state = {};
   return {
-    onOpen: async (event, clientWs) => {
+    onOpen: async (_event, clientWs) => {
       logger2().debug({ msg: "client websocket connected", targetUrl });
       if (clientWs.readyState !== 1) {
         logger2().warn({
@@ -5407,23 +5414,25 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
           reject(error);
         });
       });
-      state.targetWs.addEventListener("message", (event2) => {
-        if (typeof event2.data === "string" || event2.data instanceof ArrayBuffer) {
-          clientWs.send(event2.data);
-        } else if (event2.data instanceof Blob) {
-          event2.data.arrayBuffer().then((buffer) => {
+      state.connectPromise.catch(() => {
+      });
+      state.targetWs.addEventListener("message", (event) => {
+        if (typeof event.data === "string" || event.data instanceof ArrayBuffer) {
+          clientWs.send(event.data);
+        } else if (event.data instanceof Blob) {
+          event.data.arrayBuffer().then((buffer) => {
             clientWs.send(buffer);
           });
         }
       });
-      state.targetWs.addEventListener("close", (event2) => {
+      state.targetWs.addEventListener("close", (event) => {
         logger2().debug({
           msg: "target websocket closed",
           targetUrl,
-          code: event2.code,
-          reason: event2.reason
+          code: event.code,
+          reason: event.reason
         });
-        closeWebSocketIfOpen(clientWs, event2.code, event2.reason);
+        closeWebSocketIfOpen(clientWs, event.code, event.reason);
       });
       state.targetWs.addEventListener("error", (error) => {
         logger2().error({
@@ -5466,7 +5475,7 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
         );
       }
     },
-    onClose: (event, clientWs) => {
+    onClose: (event, _clientWs) => {
       logger2().debug({
         msg: "client websocket closed",
         targetUrl,
@@ -5483,7 +5492,7 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
         }
       }
     },
-    onError: (event, clientWs) => {
+    onError: (event, _clientWs) => {
       logger2().error({ msg: "client websocket error", targetUrl, event });
       if (state.targetWs) {
         if (state.targetWs.readyState === WebSocket2.OPEN) {
@@ -5599,7 +5608,9 @@ var RemoteEngineControlClient = class {
       name,
       key: serializeActorKey(key),
       runner_name_selector: this.#config.poolName,
-      input: actorInput ? uint8ArrayToBase64(encodeCborCompat(actorInput)) : void 0,
+      input: actorInput ? uint8ArrayToBase64(
+        encodeCborCompat(actorInput)
+      ) : void 0,
       crash_policy: crashPolicy ?? "sleep"
     });
     logger2().info({
@@ -5849,4 +5860,4 @@ export {
   updateRunnerConfig,
   RemoteEngineControlClient
 };
-//# sourceMappingURL=chunk-EMFKMVJR.js.map
+//# sourceMappingURL=chunk-4JDSFJS5.js.map