feat: import/export actors

Author: NathanFlurry

.gitattributes

@@ -7,6 +7,7 @@
 *.tar.gz filter=lfs diff=lfs merge=lfs -text
 *.tgz filter=lfs diff=lfs merge=lfs -text
 engine/packages/test-snapshot-gen/snapshots/** filter=lfs diff=lfs merge=lfs -text
+rivetkit-typescript/packages/rivetkit/fixtures/driver-test-suite/snapshots/**/kv.bin filter=lfs diff=lfs merge=lfs -text
 
 **/Cargo.lock linguist-generated=true
 

Cargo.lock

@@ -28,6 +28,8 @@ pub async fn create(
 			runner_name_selector: body.runner_name_selector,
 			input: body.input.clone(),
 			crash_policy: body.crash_policy,
+			start_immediately: true,
+			create_ts: None,
 			// NOTE: This can forward if the user attempts to create an actor with a target dc and this dc
 			// ends up forwarding to another.
 			forward_request: true,

engine/artifacts/openapi.json

@@ -48,6 +48,8 @@ pub async fn get_or_create(
 			runner_name_selector: body.runner_name_selector,
 			input: body.input.clone(),
 			crash_policy: body.crash_policy,
+			start_immediately: true,
+			create_ts: None,
 			// NOTE: This can forward if the user attempts to create an actor with a target dc and this dc
 			// ends up forwarding to another.
 			forward_request: true,

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

@@ -0,0 +1,61 @@
+use anyhow::Result;
+use rivet_api_builder::ApiCtx;
+use rivet_types::actors::CrashPolicy;
+use rivet_util::Id;
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+#[serde(deny_unknown_fields)]
+pub struct ImportCreateQuery {
+	pub namespace: String,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+#[serde(deny_unknown_fields)]
+pub struct ImportCreateRequest {
+	pub actor_id: Id,
+	pub name: String,
+	pub key: Option<String>,
+	pub runner_name_selector: String,
+	pub crash_policy: CrashPolicy,
+	pub create_ts: i64,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+#[serde(deny_unknown_fields)]
+pub struct ImportCreateResponse {
+	pub actor: rivet_types::actors::Actor,
+}
+
+#[tracing::instrument(skip_all)]
+pub async fn create(
+	ctx: ApiCtx,
+	_path: (),
+	query: ImportCreateQuery,
+	body: ImportCreateRequest,
+) -> Result<ImportCreateResponse> {
+	let namespace = ctx
+		.op(namespace::ops::resolve_for_name_global::Input {
+			name: query.namespace,
+		})
+		.await?
+		.ok_or_else(|| namespace::errors::Namespace::NotFound.build())?;
+
+	let res = ctx
+		.op(pegboard::ops::actor::create::Input {
+			actor_id: body.actor_id,
+			namespace_id: namespace.namespace_id,
+			name: body.name,
+			key: body.key,
+			runner_name_selector: body.runner_name_selector,
+			crash_policy: body.crash_policy,
+			input: None,
+			forward_request: false,
+			datacenter_name: None,
+			start_immediately: false,
+			create_ts: Some(body.create_ts),
+		})
+		.await?;
+
+	Ok(ImportCreateResponse { actor: res.actor })
+}

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

@@ -1,6 +1,7 @@
 pub mod create;
 pub mod delete;
 pub mod get_or_create;
+pub mod import_create;
 pub mod kv_get;
 pub mod list;
 pub mod list_names;

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

@@ -24,6 +24,7 @@ pub async fn router(
 			.route("/actors", get(actors::list::list))
 			.route("/actors", post(actors::create::create))
 			.route("/actors", put(actors::get_or_create::get_or_create))
+			.route("/actors/import-create", post(actors::import_create::create))
 			.route("/actors/{actor_id}", delete(actors::delete::delete))
 			.route("/actors/names", get(actors::list_names::list_names))
 			.route(

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

@@ -22,12 +22,14 @@ rivet-api-types.workspace = true
 rivet-api-util.workspace = true
 rivet-config.workspace = true
 rivet-data.workspace = true
+rivet-envoy-protocol.workspace = true
 rivet-error.workspace = true
 rivet-pools.workspace = true
 rivet-types.workspace = true
 rivet-util.workspace = true
 serde_json.workspace = true
 serde.workspace = true
+serde_bare.workspace = true
 subtle.workspace = true
 tokio.workspace = true
 tower-http.workspace = true