chore: rename retry -> hibernate in runner protocol

Author: MasterPtato

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/rust/runner-protocol/src/versioned.rs

@@ -401,8 +401,8 @@ export class Tunnel {
 							existing.actorId = req.actorId;
 						} else {
 							this.#actorPendingRequests.set(requestIdStr, {
-								resolve: () => {},
-								reject: () => {},
+								resolve: () => { },
+								reject: () => { },
 								streamController: controller,
 								actorId: req.actorId,
 							});
@@ -552,7 +552,7 @@ export class Tunnel {
 				val: {
 					code: 1011,
 					reason: "Actor not found",
-					retry: false,
+					hibernate: false,
 				},
 			});
 			return;
@@ -570,7 +570,7 @@ export class Tunnel {
 				val: {
 					code: 1011,
 					reason: "Not Implemented",
-					retry: false,
+					hibernate: false,
 				},
 			});
 			return;
@@ -608,7 +608,7 @@ export class Tunnel {
 					const dataBuffer =
 						typeof data === "string"
 							? (new TextEncoder().encode(data)
-									.buffer as ArrayBuffer)
+								.buffer as ArrayBuffer)
 							: data;
 
 					this.#sendMessage(requestId, {
@@ -619,14 +619,14 @@ export class Tunnel {
 						},
 					});
 				},
-				(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,
 						},
 					});
 
@@ -700,7 +700,7 @@ export class Tunnel {
 				val: {
 					code: 1011,
 					reason: "Server Error",
-					retry: false,
+					hibernate: false,
 				},
 			});
 

engine/sdks/schemas/runner-protocol/v3.bare

@@ -215,7 +215,7 @@ export class WebSocketTunnelAdapter {
 	closeInner(
 		code: number | undefined,
 		reason: string | undefined,
-		retry: boolean,
+		hibernate: boolean,
 		callback: boolean,
 	): void {
 		if (
@@ -229,7 +229,7 @@ export class WebSocketTunnelAdapter {
 
 		// Send close through tunnel
 		if (callback) {
-			this.#closeCallback(code, reason, retry);
+			this.#closeCallback(code, reason, hibernate);
 		}
 
 		// Update state and fire event