feat(serverless): regional serverless configs (#3101)

Author: MasterPtato

Cargo.lock

@@ -1,4 +1,3 @@
-use anyhow::anyhow;
 use axum::{
 	extract::{
 		Request,

out/openapi.json

@@ -2,4 +2,5 @@ pub mod actors;
 pub mod datacenters;
 pub mod namespaces;
 pub mod pagination;
+pub mod runner_configs;
 pub mod runners;

packages/common/api-builder/src/extract.rs

@@ -0,0 +1,29 @@
+use std::collections::HashMap;
+
+use serde::{Deserialize, Serialize};
+use utoipa::IntoParams;
+
+use crate::pagination::Pagination;
+
+#[derive(Debug, Serialize, Deserialize, Clone, IntoParams)]
+#[serde(deny_unknown_fields)]
+#[into_params(parameter_in = Query)]
+pub struct ListQuery {
+	pub namespace: String,
+	pub limit: Option<usize>,
+	pub cursor: Option<String>,
+	pub variant: Option<rivet_types::keys::namespace::runner_config::RunnerConfigVariant>,
+	#[serde(default)]
+	pub runner_names: Option<String>,
+}
+
+#[derive(Deserialize, Clone)]
+#[serde(deny_unknown_fields)]
+pub struct ListPath {}
+
+#[derive(Deserialize, Serialize)]
+#[serde(deny_unknown_fields)]
+pub struct ListResponse {
+	pub runner_configs: HashMap<String, rivet_types::runner_configs::RunnerConfig>,
+	pub pagination: Pagination,
+}

packages/common/api-types/src/lib.rs

@@ -0,0 +1 @@
+pub mod list;

packages/common/api-types/src/runner_configs/list.rs

@@ -102,7 +102,7 @@ where
 	Q: Serialize + Clone + Send + 'static,
 	F: Fn(ApiCtx, Q) -> Fut + Clone + Send + 'static,
 	Fut: Future<Output = Result<I>> + Send,
-	A: Fn(I, &mut R),
+	A: Fn(u16, I, &mut R),
 	R: Default + Send + 'static,
 {
 	let dcs = &ctx.config().topology().datacenters;
@@ -117,19 +117,22 @@ where
 		async move {
 			if dc.datacenter_label == ctx.config().dc_label() {
 				// Local datacenter - use direct API call
-				local_handler(ctx, query).await
+				(dc.datacenter_label, local_handler(ctx, query).await)
 			} else {
 				// Remote datacenter - HTTP request
-				request_remote_datacenter::<I>(
-					ctx.config(),
+				(
 					dc.datacenter_label,
-					&endpoint,
-					Method::GET,
-					headers,
-					Some(&query),
-					Option::<&()>::None,
+					request_remote_datacenter::<I>(
+						ctx.config(),
+						dc.datacenter_label,
+						&endpoint,
+						Method::GET,
+						headers,
+						Some(&query),
+						Option::<&()>::None,
+					)
+					.await,
 				)
-				.await
 			}
 		}
 	}))
@@ -141,11 +144,11 @@ where
 	let result_count = results.len();
 	let mut errors = Vec::new();
 	let mut aggregated = R::default();
-	for res in results {
+	for (dc_label, res) in results {
 		match res {
-			Ok(data) => aggregator(data, &mut aggregated),
+			Ok(data) => aggregator(dc_label, data, &mut aggregated),
 			Err(err) => {
-				tracing::error!(?err, "failed to request edge dc");
+				tracing::error!(?dc_label, ?err, "failed to request edge dc");
 				errors.push(err);
 			}
 		}

packages/common/api-types/src/runner_configs/mod.rs

@@ -13,6 +13,7 @@ rivet-data.workspace = true
 rivet-runner-protocol.workspace = true
 rivet-util.workspace = true
 serde.workspace = true
+strum.workspace = true
 universaldb.workspace = true
 utoipa.workspace = true
 vbare.workspace = true

packages/common/api-util/src/lib.rs

@@ -1 +1,2 @@
+pub mod namespace;
 pub mod pegboard;

packages/common/types/Cargo.toml

@@ -0,0 +1 @@
+pub mod runner_config;