rivetkit 2.3.0-rc.5 → 2.3.0-rc.6

package/src/registry/runtime.ts

@@ -257,9 +257,10 @@ export interface RuntimeServerlessResponseHead {
 	headers: Record<string, string>;
 }
 
-export interface RuntimeRegistryDiagnostics {
-	mode: string;
-	envoyActiveActorCount?: number | null;
+export interface RuntimeRegistryRouteResponse {
+	status: number;
+	headers: Record<string, string>;
+	body: RuntimeBytes;
 }
 
 export type RuntimeServerlessStreamEvent =
@@ -309,6 +310,9 @@ export interface CoreRuntime {
 		config: RuntimeServeConfig,
 	): Promise<void>;
 	shutdownRegistry(registry: RegistryHandle): Promise<void>;
+	registryActorStopThresholdMs?(
+		registry: RegistryHandle,
+	): Promise<number | undefined>;
 	handleServerlessRequest(
 		registry: RegistryHandle,
 		req: RuntimeServerlessRequest,
@@ -316,9 +320,15 @@ export interface CoreRuntime {
 		cancelToken: CancellationTokenHandle,
 		config: RuntimeServeConfig,
 	): Promise<RuntimeServerlessResponseHead>;
-	registryDiagnostics?(
+	registryHealth?(
+		registry: RegistryHandle,
+	): Promise<RuntimeRegistryRouteResponse>;
+	registryMetadata?(
+		registry: RegistryHandle,
+	): Promise<RuntimeRegistryRouteResponse>;
+	registryMetrics?(
 		registry: RegistryHandle,
-	): Promise<RuntimeRegistryDiagnostics>;
+	): Promise<RuntimeRegistryRouteResponse>;
 	createActorFactory(
 		callbacks: object,
 		config?: RuntimeActorConfig | undefined | null,
@@ -388,6 +398,7 @@ export interface CoreRuntime {
 		args: RuntimeBytes,
 	): void;
 	actorWaitUntil(ctx: ActorContextHandle, promise: Promise<unknown>): void;
+	actorWaitForTrackedShutdownWork(ctx: ActorContextHandle): Promise<boolean>;
 	actorKeepAwake(ctx: ActorContextHandle, promise: Promise<unknown>): void;
 	actorBeginKeepAwake(ctx: ActorContextHandle): number;
 	actorEndKeepAwake(ctx: ActorContextHandle, regionId: number): void;

package/src/registry/wasm-runtime.ts

@@ -33,6 +33,7 @@ import type {
 	RuntimeQueueTryNextBatchOptions,
 	RuntimeQueueWaitOptions,
 	RuntimeRequestSaveOpts,
+	RuntimeRegistryRouteResponse,
 	RuntimeServeConfig,
 	RuntimeServerlessRequest,
 	RuntimeServerlessResponseHead,
@@ -270,8 +271,18 @@ export class WasmCoreRuntime implements CoreRuntime {
 		await callWasm(() => asWasmRegistry(registry).shutdown());
 	}
 
-	async registryDiagnostics(): Promise<{ mode: string; envoyActiveActorCount: null }> {
-		return { mode: "wasm", envoyActiveActorCount: null };
+	async registryHealth(): Promise<RuntimeRegistryRouteResponse> {
+		return {
+			status: 200,
+			headers: { "content-type": "application/json" },
+			body: new TextEncoder().encode(
+				JSON.stringify({
+					status: "ok",
+					runtime: "rivetkit",
+					version: "wasm",
+				}),
+			),
+		};
 	}
 
 	async handleServerlessRequest(
@@ -499,6 +510,15 @@ export class WasmCoreRuntime implements CoreRuntime {
 		callHandle(asWasmActorContext(ctx), "waitUntil", promise);
 	}
 
+	async actorWaitForTrackedShutdownWork(
+		ctx: ActorContextHandle,
+	): Promise<boolean> {
+		return await callHandle<Promise<boolean>>(
+			asWasmActorContext(ctx),
+			"waitForTrackedShutdownWork",
+		);
+	}
+
 	actorKeepAwake(ctx: ActorContextHandle, promise: Promise<unknown>): void {
 		const wasmCtx = asWasmActorContext(ctx);
 		const regionId = callHandle<number>(wasmCtx, "beginKeepAwake");