@rivetkit/engine-runner 2.3.0-rc.1 → 2.3.0-rc.11

package/dist/mod.cjs

@@ -1,5 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class;// src/mod.ts
 var _enginerunnerprotocol = require('@rivetkit/engine-runner-protocol'); var protocol = _interopRequireWildcard(_enginerunnerprotocol);
+var _uuid = require('uuid');
 
 // src/log.ts
 var LOGGER;
@@ -513,6 +514,8 @@ function stringifyKvResponseData(data) {
 }
 
 // src/websocket-tunnel-adapter.ts
+
+
 var _virtualwebsocket = require('@rivetkit/virtual-websocket');
 var HIBERNATABLE_SYMBOL = /* @__PURE__ */ Symbol("hibernatable");
 var WebSocketTunnelAdapter = class {
@@ -570,14 +573,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 +600,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 +641,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 +656,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 +1538,6 @@ async function importWebSocket() {
 }
 
 // src/mod.ts
-
-
-var _uuid = require('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;
@@ -2194,7 +2217,7 @@ var Runner = class {
     );
     for (const [_, actor] of this.#actors) {
       const checkpoint = ack.lastEventCheckpoints.find(
-        (x) => x.actorId == actor.actorId
+        (x) => x.actorId === actor.actorId
       );
       if (checkpoint) actor.handleAckEvents(checkpoint.index);
     }
@@ -2283,18 +2306,18 @@ var Runner = class {
       });
       await this.#config.onActorStart(actorId, generation, actorConfig);
       instance.actorStartPromise.resolve();
-    } catch (err) {
+    } catch (error) {
       (_d = this.log) == null ? void 0 : _d.error({
         msg: "error starting runner actor",
         actorId,
-        err
+        error
       });
-      instance.actorStartPromise.reject(err);
+      instance.actorStartPromise.reject(error);
       await this.forceStopActor(actorId, generation);
     }
   }
   async #handleCommandStopActor(commandWrapper) {
-    const stopCommand = commandWrapper.inner.val;
+    const _stopCommand = commandWrapper.inner.val;
     const actorId = commandWrapper.checkpoint.actorId;
     const generation = commandWrapper.checkpoint.generation;
     await this.forceStopActor(actorId, generation);