Skip to content

Commit 3a63654

Browse files
committed
chore(rivetkit): log close code & reason in disconnect warning
1 parent d6e9414 commit 3a63654

File tree

2 files changed

+5
-3
lines changed
  • engine/sdks/typescript/runner/src
  • rivetkit-typescript/packages/rivetkit/src/drivers/engine

2 files changed

+5
-3
lines changed

engine/sdks/typescript/runner/src/mod.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ export interface RunnerConfig {
4343
prepopulateActorNames: Record<string, { metadata: Record<string, any> }>;
4444
metadata?: Record<string, any>;
4545
onConnected: () => void;
46-
onDisconnected: () => void;
46+
onDisconnected: (code: number, reason: string) => void;
4747
onShutdown: () => void;
4848
fetch: (
4949
runner: Runner,
@@ -640,7 +640,7 @@ export class Runner {
640640
reason: ev.reason.toString(),
641641
});
642642

643-
this.#config.onDisconnected();
643+
this.#config.onDisconnected(ev.code, ev.reason);
644644

645645
if (ev.reason.toString().startsWith("ws.eviction")) {
646646
logger()?.info({

rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,11 +125,13 @@ export class EngineActorDriver implements ActorDriver {
125125

126126
this.#runnerStarted.resolve(undefined);
127127
},
128-
onDisconnected: () => {
128+
onDisconnected: (code, reason) => {
129129
logger().warn({
130130
msg: "runner disconnected",
131131
namespace: this.#runConfig.namespace,
132132
runnerName: this.#runConfig.runnerName,
133+
code,
134+
reason,
133135
});
134136
hasDisconnected = true;
135137
},

0 commit comments

Comments
 (0)