@rivetkit/engine-runner 0.0.0-main.41efcce → 0.0.0-main.4994268

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;
@@ -1537,7 +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;
@@ -1815,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 {
@@ -2072,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;
@@ -2213,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);
     }
@@ -2302,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);