From f76b79cd53b12f06b24272bb7fc5b3d0c976e482 Mon Sep 17 00:00:00 2001
From: Nathan Flurry <developer@nathanflurry.com>
Date: Fri, 31 Oct 2025 14:50:06 -0700
Subject: [PATCH] chore(rivetkit): log close code & reason in disconnect
 warning

---
 engine/sdks/typescript/runner/src/mod.ts                      | 4 ++--
 .../packages/rivetkit/src/drivers/engine/actor-driver.ts      | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/engine/sdks/typescript/runner/src/mod.ts b/engine/sdks/typescript/runner/src/mod.ts
index 64c4e3c309..7e4955471e 100644
--- a/engine/sdks/typescript/runner/src/mod.ts
+++ b/engine/sdks/typescript/runner/src/mod.ts
@@ -43,7 +43,7 @@ export interface RunnerConfig {
 	prepopulateActorNames: Record<string, { metadata: Record<string, any> }>;
 	metadata?: Record<string, any>;
 	onConnected: () => void;
-	onDisconnected: () => void;
+	onDisconnected: (code: number, reason: string) => void;
 	onShutdown: () => void;
 	fetch: (
 		runner: Runner,
@@ -640,7 +640,7 @@ export class Runner {
 				reason: ev.reason.toString(),
 			});
 
-			this.#config.onDisconnected();
+			this.#config.onDisconnected(ev.code, ev.reason);
 
 			if (ev.reason.toString().startsWith("ws.eviction")) {
 				logger()?.info({
diff --git a/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts b/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts
index 508f0566ce..230f60c616 100644
--- a/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts
+++ b/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts
@@ -125,11 +125,13 @@ export class EngineActorDriver implements ActorDriver {
 
 				this.#runnerStarted.resolve(undefined);
 			},
-			onDisconnected: () => {
+			onDisconnected: (code, reason) => {
 				logger().warn({
 					msg: "runner disconnected",
 					namespace: this.#runConfig.namespace,
 					runnerName: this.#runConfig.runnerName,
+					code,
+					reason,
 				});
 				hasDisconnected = true;
 			},