@rivetkit/engine-runner 2.3.0-rc.4 → 2.3.0-rc.5

package/dist/mod.js

@@ -513,7 +513,9 @@ function stringifyKvResponseData(data) {
 }
 
 // src/websocket-tunnel-adapter.ts
-import { VirtualWebSocket } from "@rivetkit/virtual-websocket";
+import {
+  VirtualWebSocket
+} from "@rivetkit/virtual-websocket";
 var HIBERNATABLE_SYMBOL = /* @__PURE__ */ Symbol("hibernatable");
 var WebSocketTunnelAdapter = class {
   constructor(tunnel, actorId, requestId, serverMessageIndex, hibernatable, isRestoringHibernatable, request, sendCallback, closeCallback) {
@@ -570,14 +572,23 @@ var WebSocketTunnelAdapter = class {
       }
       messageData = data;
     } else if (data instanceof ArrayBuffer) {
-      if (data.byteLength > MAX_PAYLOAD_SIZE) throw new Error("WebSocket message too large");
+      if (data.byteLength > MAX_PAYLOAD_SIZE)
+        throw new Error("WebSocket message too large");
       isBinary = true;
       messageData = data;
     } else if (ArrayBuffer.isView(data)) {
-      if (data.byteLength > MAX_PAYLOAD_SIZE) throw new Error("WebSocket message too large");
+      if (data.byteLength > MAX_PAYLOAD_SIZE)
+        throw new Error("WebSocket message too large");
       isBinary = true;
       const view = data;
-      const buffer = view.buffer instanceof SharedArrayBuffer ? new Uint8Array(view.buffer, view.byteOffset, view.byteLength).slice().buffer : view.buffer.slice(view.byteOffset, view.byteOffset + view.byteLength);
+      const buffer = view.buffer instanceof SharedArrayBuffer ? new Uint8Array(
+        view.buffer,
+        view.byteOffset,
+        view.byteLength
+      ).slice().buffer : view.buffer.slice(
+        view.byteOffset,
+        view.byteOffset + view.byteLength
+      );
       messageData = buffer;
     } else {
       throw new Error("Unsupported data type");
@@ -588,7 +599,11 @@ var WebSocketTunnelAdapter = class {
   _handleOpen(requestId) {
     if (this.#readyState !== 0) return;
     this.#readyState = 1;
-    this.#ws.dispatchEvent({ type: "open", rivetRequestId: requestId, target: this.#ws });
+    this.#ws.dispatchEvent({
+      type: "open",
+      rivetRequestId: requestId,
+      target: this.#ws
+    });
   }
   // Called by Tunnel when message is received
   _handleMessage(requestId, data, serverMessageIndex, isBinary) {
@@ -625,7 +640,10 @@ var WebSocketTunnelAdapter = class {
           expectedIndex,
           receivedIndex: serverMessageIndex,
           closeReason,
-          gap: wrappingSubU16(wrappingSubU16(serverMessageIndex, previousIndex), 1)
+          gap: wrappingSubU16(
+            wrappingSubU16(serverMessageIndex, previousIndex),
+            1
+          )
         });
         this.#close(1008, closeReason, true);
         return true;
@@ -637,7 +655,10 @@ var WebSocketTunnelAdapter = class {
       if (this.#binaryType === "nodebuffer") {
         messageData = Buffer.from(data);
       } else if (this.#binaryType === "arraybuffer") {
-        messageData = data.buffer.slice(data.byteOffset, data.byteOffset + data.byteLength);
+        messageData = data.buffer.slice(
+          data.byteOffset,
+          data.byteOffset + data.byteLength
+        );
       }
     }
     this.#ws.dispatchEvent({
@@ -1516,9 +1537,7 @@ async function importWebSocket() {
 }
 
 // src/mod.ts
-import {
-  v4 as uuidv4
-} from "uuid";
+import { v4 as uuidv4 } from "uuid";
 var KV_EXPIRE = 3e4;
 var PROTOCOL_VERSION = 7;
 var EVENT_BACKLOG_WARN_THRESHOLD = 1e4;
@@ -1796,9 +1815,12 @@ var Runner = class {
       );
     }
     this.#kvRequests.clear();
-    const pegboardWebSocket = this.getPegboardWebSocketIfReady();
-    if (pegboardWebSocket) {
-      if (immediate) {
+    const pegboardWebSocket = this.#pegboardWebSocket;
+    const readyState = pegboardWebSocket == null ? void 0 : pegboardWebSocket.readyState;
+    const isOpen = readyState === 1;
+    const isConnecting = readyState === 0;
+    if (pegboardWebSocket && (isOpen || isConnecting)) {
+      if (immediate || isConnecting) {
         pegboardWebSocket.close(1e3, "pegboard.runner_shutdown");
       } else {
         try {
@@ -2053,6 +2075,7 @@ var Runner = class {
         const init = message.val;
         if (this.runnerId !== init.runnerId) {
           this.runnerId = init.runnerId;
+          this.#logCached = void 0;
           this.#stopAllActors();
         }
         this.#protocolMetadata = init.metadata;