@rivetkit/engine-runner 0.0.0-pr.4879.d508104 → 0.0.0-pr.4880.c7bfbc5

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/engine-runner",
-	"version": "0.0.0-pr.4879.d508104",
+	"version": "0.0.0-pr.4880.c7bfbc5",
 	"type": "module",
 	"files": [
 		"dist",
@@ -25,8 +25,8 @@
 		"bench": "tsx benches/actor-lifecycle.bench.ts"
 	},
 	"dependencies": {
-		"@rivetkit/virtual-websocket": "0.0.0-pr.4879.d508104",
-		"@rivetkit/engine-runner-protocol": "0.0.0-pr.4879.d508104",
+		"@rivetkit/virtual-websocket": "0.0.0-pr.4880.c7bfbc5",
+		"@rivetkit/engine-runner-protocol": "0.0.0-pr.4880.c7bfbc5",
 		"uuid": "^12.0.0",
 		"pino": "^9.9.5",
 		"ws": "^8.18.3"

package/src/mod.ts

@@ -560,18 +560,9 @@ export class Runner {
 		this.#kvRequests.clear();
 
 		// Close WebSocket
-		//
-		// A CONNECTING socket cannot send graceful stopping messages, so
-		// it is always closed directly. Without this, a shutdown that
-		// races with a reconnect leaves the in-flight WebSocket alive
-		// while the rest of the runner state (tunnel, intervals) has been
-		// torn down.
-		const pegboardWebSocket = this.#pegboardWebSocket;
-		const readyState = pegboardWebSocket?.readyState;
-		const isOpen = readyState === 1;
-		const isConnecting = readyState === 0;
-		if (pegboardWebSocket && (isOpen || isConnecting)) {
-			if (immediate || isConnecting) {
+		const pegboardWebSocket = this.getPegboardWebSocketIfReady();
+		if (pegboardWebSocket) {
+			if (immediate) {
 				// Stop immediately
 				pegboardWebSocket.close(1000, "pegboard.runner_shutdown");
 			} else {
@@ -883,10 +874,6 @@ export class Runner {
 				if (this.runnerId !== init.runnerId) {
 					this.runnerId = init.runnerId;
 
-					// Invalidate cached child logger so subsequent log
-					// lines pick up the new runnerId.
-					this.#logCached = undefined;
-
 					// Clear actors if runner id changed
 					this.#stopAllActors();
 				}