chore: rename retry -> hibernate in runner protocol

engine/packages/api-peer/src/actors/kv_get.rs

@@ -1,6 +1,6 @@
 use anyhow::*;
-use base64::prelude::BASE64_STANDARD;
 use base64::Engine;
+use base64::prelude::BASE64_STANDARD;
 use pegboard_actor_kv as actor_kv;
 use rivet_api_builder::ApiCtx;
 use rivet_util::Id;

engine/packages/api-public/src/actors/kv_get.rs

@@ -65,7 +65,11 @@ async fn kv_get_inner(ctx: ApiCtx, path: KvGetPath) -> Result<Response> {
 		request_remote_datacenter_raw(
 			&ctx,
 			path.actor_id.label(),
-			&format!("/actors/{}/kv/keys/{}", path.actor_id, urlencoding::encode(&path.key)),
+			&format!(
+				"/actors/{}/kv/keys/{}",
+				path.actor_id,
+				urlencoding::encode(&path.key)
+			),
 			axum::http::Method::GET,
 			Option::<&()>::None,
 			Option::<&()>::None,

engine/packages/pegboard-gateway/src/lib.rs

@@ -401,10 +401,10 @@ impl CustomServeTrait for PegboardGateway {
 										protocol::ToServerTunnelMessageKind::ToServerWebSocketClose(close) => {
 										tracing::debug!(?close, "server closed websocket");
 
-										if can_hibernate && close.retry {
-											// Successful closure
+										if can_hibernate && close.hibernate {
 											return Err(WebSocketServiceHibernate.build());
 										} else {
+											// Successful closure
 											return Ok(LifecycleResult::ServerClose(close));
 										}
 									}

engine/packages/pegboard-runner/src/lib.rs

@@ -257,7 +257,7 @@ impl CustomServeTrait for PegboardRunnerWsCustomServe {
 					protocol::ToServerWebSocketClose {
 						code: Some(close_code),
 						reason: close_reason,
-						retry: true,
+						hibernate: true,
 					},
 				)
 			} else {

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

@@ -1,8 +1,8 @@
 import type * as protocol from "@rivetkit/engine-runner-protocol";
 import type { MessageId, RequestId } from "@rivetkit/engine-runner-protocol";
 import type { Logger } from "pino";
 import { stringify as uuidstringify, v4 as uuidv4 } from "uuid";
 import { logger } from "./log";
 import type { ActorInstance, Runner } from "./mod";
 import {
 	stringifyToClientTunnelMessageKind,
@@ -13,7 +13,7 @@
 
 const GC_INTERVAL = 60000; // 60 seconds
 const MESSAGE_ACK_TIMEOUT = 5000; // 5 seconds
 const WEBSOCKET_STATE_PERSIST_TIMEOUT = 30000; // 30 seconds
 
 interface PendingRequest {
 	resolve: (response: Response) => void;
@@ -401,8 +401,8 @@
 							existing.actorId = req.actorId;
 						} else {
 							this.#actorPendingRequests.set(requestIdStr, {
-								resolve: () => {},
-								reject: () => {},
+								resolve: () => { },
+								reject: () => { },
 								streamController: controller,
 								actorId: req.actorId,
 							});
@@ -552,7 +552,7 @@
 				val: {
 					code: 1011,
 					reason: "Actor not found",
-					retry: false,
+					hibernate: false,
 				},
 			});
 			return;
@@ -570,7 +570,7 @@
 				val: {
 					code: 1011,
 					reason: "Not Implemented",
-					retry: false,
+					hibernate: false,
 				},
 			});
 			return;
@@ -608,7 +608,7 @@
 					const dataBuffer =
 						typeof data === "string"
 							? (new TextEncoder().encode(data)
-									.buffer as ArrayBuffer)
+								.buffer as ArrayBuffer)
 							: data;
 
 					this.#sendMessage(requestId, {
@@ -619,14 +619,14 @@
 						},
 					});
 				},
-				(code?: number, reason?: string, retry: boolean = false) => {
+				(code?: number, reason?: string, hibernate: boolean = false) => {
 					// Send close through tunnel
 					this.#sendMessage(requestId, {
 						tag: "ToServerWebSocketClose",
 						val: {
 							code: code || null,
 							reason: reason || null,
-							retry,
+							hibernate,
 						},
 					});
 
@@ -656,8 +656,8 @@
 					headerInit[k] = v;
 				}
 			}
 			headerInit["Upgrade"] = "websocket";
 			headerInit["Connection"] = "Upgrade";
 
 			const request = new Request(`http://localhost${open.path}`, {
 				method: "GET",
@@ -700,7 +700,7 @@
 				val: {
 					code: 1011,
 					reason: "Server Error",
-					retry: false,
+					hibernate: false,
 				},
 			});
 

engine/sdks/typescript/runner/src/websocket-tunnel-adapter.ts

@@ -4,7 +4,7 @@
 import { logger } from "./log";
 
 export class WebSocketTunnelAdapter {
 	#webSocketId: string;
 	#readyState: number = 0; // CONNECTING
 	#eventListeners: Map<string, Set<(event: any) => void>> = new Map();
 	#onopen: ((this: any, ev: any) => any) | null = null;
@@ -215,7 +215,7 @@
 	closeInner(
 		code: number | undefined,
 		reason: string | undefined,
-		retry: boolean,
+		hibernate: boolean,
 		callback: boolean,
 	): void {
 		if (
@@ -229,7 +229,7 @@
 
 		// Send close through tunnel
 		if (callback) {
-			this.#closeCallback(code, reason, retry);
+			this.#closeCallback(code, reason, hibernate);
 		}
 
 		// Update state and fire event
@@ -249,7 +249,7 @@
 	addEventListener(
 		type: string,
 		listener: (event: any) => void,
 		options?: boolean | any,
 	): void {
 		if (typeof listener === "function") {
 			let listeners = this.#eventListeners.get(type);
@@ -267,7 +267,7 @@
 	removeEventListener(
 		type: string,
 		listener: (event: any) => void,
 		options?: boolean | any,
 	): void {
 		if (typeof listener === "function") {
 			const listeners = this.#eventListeners.get(type);
@@ -561,7 +561,7 @@
 		if (cb) cb(new Error("Ping not supported in tunnel adapter"));
 	}
 
 	pong(data?: any, mask?: boolean, cb?: (err: Error) => void): void {
 		// Not implemented for tunnel - could be added if needed
 		if (cb) cb(new Error("Pong not supported in tunnel adapter"));
 	}