rivetkit 2.3.0-rc.9 → 2.3.1

package/dist/tsup/{chunk-F3Q5BFQ6.js → chunk-PT6OIW5E.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-W7EYSYVI.js";
+} from "./chunk-R6KPN5EW.js";
 import {
   AsyncMutex
-} from "./chunk-PCBNKI2J.js";
+} from "./chunk-JZ7TWV65.js";
 import {
   importWebSocket,
   logger
-} from "./chunk-H37XQU3I.js";
+} from "./chunk-JBUZRPY5.js";
 import {
   assertUnreachable,
   bufferToArrayBuffer,
@@ -37,13 +37,13 @@ import {
   noopNext,
   promiseWithResolvers,
   stringifyError
-} from "./chunk-VRCIXJRN.js";
+} from "./chunk-L2X3YFER.js";
 import {
   INTERNAL_ERROR_CODE,
   RivetError,
   actorNotFound,
   invalidRequest
-} from "./chunk-KIWH5H3K.js";
+} from "./chunk-ZZ3WBRPD.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.
@@ -3594,7 +3603,6 @@ function createActorProxy(handle) {
 }
 
 // src/client/actor-conn.ts
-var DEFAULT_MAX_INCOMING_MESSAGE_SIZE = 65536;
 var CONNECT_SYMBOL = /* @__PURE__ */ Symbol("connect");
 var ActorConnRaw = class {
   #disposed = false;
@@ -4074,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,
@@ -4477,31 +4483,8 @@ var ActorConnRaw = class {
               }
             }
           );
-          const serializedLength = messageLength(messageSerialized);
-          if (serializedLength > DEFAULT_MAX_INCOMING_MESSAGE_SIZE && message.body.tag === "ActionRequest") {
-            const actionId = Number(message.body.val.id);
-            const inFlight = this.#takeActionInFlight(actionId);
-            const error = new RivetError(
-              "message",
-              "incoming_too_long",
-              "Incoming message too long",
-              {
-                maxSize: DEFAULT_MAX_INCOMING_MESSAGE_SIZE,
-                actualSize: serializedLength
-              }
-            );
-            logger().warn({
-              msg: "rejecting oversized connection action request",
-              actionId,
-              actionName: inFlight.name,
-              actualSize: serializedLength,
-              maxSize: DEFAULT_MAX_INCOMING_MESSAGE_SIZE
-            });
-            inFlight.reject(error);
-            this.#dispatchActorError(error);
-            return;
-          }
           this.#websocket.send(messageSerialized);
+          const serializedLength = messageLength(messageSerialized);
           logger().trace({
             msg: "sent websocket message",
             len: serializedLength
@@ -4990,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, {
@@ -5389,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({
@@ -5431,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({
@@ -5490,7 +5475,7 @@ async function createWebSocketProxy(c, targetUrl, protocols) {
         );
       }
     },
-    onClose: (event, clientWs) => {
+    onClose: (event, _clientWs) => {
       logger2().debug({
         msg: "client websocket closed",
         targetUrl,
@@ -5507,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) {
@@ -5623,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({
@@ -5873,4 +5860,4 @@ export {
   updateRunnerConfig,
   RemoteEngineControlClient
 };
-//# sourceMappingURL=chunk-F3Q5BFQ6.js.map
+//# sourceMappingURL=chunk-PT6OIW5E.js.map