feat(inspector): insert and clear actor queue

Author: abcxff

rivetkit-rust/packages/rivetkit-core/src/actor/queue.rs

@@ -535,6 +535,46 @@ impl ActorContext {
 		self.config().max_queue_size
 	}
 
+	/// Removes all messages from the queue and resets metadata.
+	pub async fn reset(&self) -> Result<()> {
+		self.ensure_initialized().await?;
+
+		// List and delete all message keys
+		let entries = self.list_message_entries().await?;
+		if !entries.is_empty() {
+			let keys: Vec<Vec<u8>> = entries.iter().map(|(k, _)| k.clone()).collect();
+			let key_refs: Vec<&[u8]> = keys.iter().map(Vec::as_slice).collect();
+			self.0
+				.kv
+				.batch_delete(&key_refs)
+				.await
+				.context("delete all queue messages")?;
+		}
+
+		// Reset metadata
+		let metadata = QueueMetadata {
+			next_id: 0,
+			size: 0,
+		};
+		let encoded_metadata =
+			encode_queue_metadata(&metadata).context("encode reset queue metadata")?;
+		self.0
+			.kv
+			.put(&QUEUE_METADATA_KEY, &encoded_metadata)
+			.await
+			.context("persist reset queue metadata")?;
+		*self.0.queue_metadata.lock().await = metadata;
+
+		// Drop all completion waiters
+		self.0.queue_completion_waiters.clear_async().await;
+
+		// Update metrics and notify inspector
+		self.0.metrics.set_queue_depth(0);
+		self.notify_inspector_update(0);
+
+		Ok(())
+	}
+
 	pub(crate) fn configure_queue(&self, config: ActorConfig) {
 		*self.0.queue_config.lock() = config;
 	}

rivetkit-rust/packages/rivetkit-core/src/registry/inspector.rs

@@ -1,7 +1,7 @@
 use super::dispatch::*;
 use super::http::*;
 use super::*;
-use crate::error::{client_error_message, ProtocolError};
+use crate::error::{ProtocolError, client_error_message};
 use ::http;
 
 #[derive(rivet_error::RivetError, serde::Serialize)]
@@ -127,6 +127,30 @@ impl RegistryDispatcher {
 				};
 				json_http_response(StatusCode::OK, &payload)
 			}
+			(http::Method::DELETE, "/inspector/queue") => {
+				match instance.ctx.queue().reset().await {
+					Ok(_) => json_http_response(StatusCode::OK, &json!({})),
+					Err(error) => Err(error).context("reset inspector queue"),
+				}
+			}
+			(http::Method::POST, "/inspector/queue") => {
+				let body: InspectorEnqueueBody = match parse_json_body(request) {
+					Ok(body) => body,
+					Err(response) => return Ok(Some(response)),
+				};
+				let cbor_body = encode_json_as_cbor(&body.body.unwrap_or(serde_json::Value::Null))?;
+				match instance.ctx.queue().send(&body.name, &cbor_body).await {
+					Ok(message) => json_http_response(
+						StatusCode::OK,
+						&json!({
+							"id": message.id,
+							"name": message.name,
+							"createdAtMs": message.created_at,
+						}),
+					),
+					Err(error) => Err(error).context("enqueue inspector queue message"),
+				}
+			}
 			(http::Method::GET, "/inspector/workflow-history") => self
 				.inspector_workflow_history(instance)
 				.await
@@ -346,9 +370,7 @@ impl RegistryDispatcher {
 	) -> Result<(bool, Option<Vec<u8>>)> {
 		let result = instance
 			.ctx
-			.internal_keep_awake(dispatch_workflow_history_through_task(
-				&instance.dispatch,
-			))
+			.internal_keep_awake(dispatch_workflow_history_through_task(&instance.dispatch))
 			.await
 			.context("load inspector workflow history");
 

rivetkit-rust/packages/rivetkit-core/src/registry/mod.rs

@@ -212,6 +212,13 @@ struct InspectorWorkflowReplayBody {
 	entry_id: Option<String>,
 }
 
+#[derive(Debug, Default, Deserialize)]
+#[serde(default)]
+struct InspectorEnqueueBody {
+	name: String,
+	body: Option<JsonValue>,
+}
+
 #[derive(Debug, Serialize)]
 #[serde(rename_all = "camelCase")]
 struct InspectorQueueMessageJson {
@@ -625,11 +632,8 @@ impl RegistryDispatcher {
 
 		let (start_tx, start_rx) = oneshot::channel();
 		let result: Result<Arc<ActorTaskHandle>> = async {
-			try_send_lifecycle_command(
-				&lifecycle_tx,
-				LifecycleCommand::Start { reply: start_tx },
-			)
-			.context("send actor task start command")?;
+			try_send_lifecycle_command(&lifecycle_tx, LifecycleCommand::Start { reply: start_tx })
+				.context("send actor task start command")?;
 			start_rx
 				.await
 				.context("receive actor task start reply")?

rivetkit-typescript/packages/rivetkit-napi/src/queue.rs

@@ -216,6 +216,11 @@ impl Queue {
 		self.inner.max_size()
 	}
 
+	#[napi]
+	pub async fn reset(&self) -> napi::Result<()> {
+		self.inner.reset().await.map_err(napi_anyhow_error)
+	}
+
 	#[napi]
 	pub async fn inspect_messages(&self) -> napi::Result<Vec<JsQueueInspectMessage>> {
 		self.inner

rivetkit-typescript/packages/rivetkit/src/registry/napi-runtime.ts

@@ -687,6 +687,10 @@ export class NapiCoreRuntime implements CoreRuntime {
 		return await asNativeActorContext(ctx).queue().inspectMessages();
 	}
 
+	async actorQueueReset(ctx: ActorContextHandle): Promise<void> {
+		await asNativeActorContext(ctx).queue().reset();
+	}
+
 	actorScheduleAfter(
 		ctx: ActorContextHandle,
 		durationMs: number,

rivetkit-typescript/packages/rivetkit/src/registry/native.ts

@@ -3461,6 +3461,27 @@ export function buildNativeFactory(
 					messages,
 				});
 			}
+			if (
+				url.pathname === "/inspector/queue" &&
+				jsRequest.method === "DELETE"
+			) {
+				await runtime.actorQueueReset(ctx);
+				return jsonResponse({});
+			}
+			if (
+				url.pathname === "/inspector/queue" &&
+				jsRequest.method === "POST"
+			) {
+				const body = await jsRequest.json() as { name?: string; body?: unknown };
+				const name = body.name ?? "";
+				const cbor = encodeCborCompat((body.body ?? null) as JsonCompatValue);
+				const message = await runtime.actorQueueSend(ctx, name, cbor);
+				return jsonResponse({
+					id: Number(message.id()),
+					name: message.name(),
+					createdAtMs: message.createdAt(),
+				});
+			}
 			if (
 				url.pathname === "/inspector/traces" &&
 				jsRequest.method === "GET"

rivetkit-typescript/packages/rivetkit/src/registry/runtime.ts

@@ -498,6 +498,7 @@ export interface CoreRuntime {
 	actorQueueInspectMessages(
 		ctx: ActorContextHandle,
 	): Promise<RuntimeQueueInspectMessage[]>;
+	actorQueueReset(ctx: ActorContextHandle): Promise<void>;
 
 	actorScheduleAfter(
 		ctx: ActorContextHandle,

rivetkit-typescript/packages/rivetkit/src/registry/wasm-runtime.ts

@@ -803,6 +803,11 @@ export class WasmCoreRuntime implements CoreRuntime {
 		return await callHandleAsync(queue, "inspectMessages");
 	}
 
+	async actorQueueReset(ctx: ActorContextHandle): Promise<void> {
+		const queue = childHandle(asWasmActorContext(ctx), "queue");
+		await callHandleAsync(queue, "reset");
+	}
+
 	actorScheduleAfter(
 		ctx: ActorContextHandle,
 		durationMs: number | bigint,