Add /health endpoint

crates/client-api/src/routes/health.rs

@@ -0,0 +1,47 @@
+use crate::{ControlStateDelegate, NodeDelegate};
+use axum::extract::State;
+use axum::response::IntoResponse;
+use http::StatusCode;
+
+static VERSION: &str = env!("CARGO_PKG_VERSION");
+static PACKAGE_NAME: &str = env!("CARGO_PKG_NAME");
+
+pub async fn health<S: ControlStateDelegate + NodeDelegate>(
+    State(ctx): State<S>,
+) -> axum::response::Result<impl IntoResponse> {
+    let nodes: Vec<u64> = ctx
+        .get_nodes()
+        .map_err(|_| {
+            (
+                StatusCode::INTERNAL_SERVER_ERROR,
+                "Couldn't connect to the control database",
+            )
+        })?
+        .iter()
+        .map(|n| n.id)
+        .collect();
+    let schedulable = !ctx
+        .get_node_by_id(
+            ctx.get_node_id()
+                .ok_or((StatusCode::INTERNAL_SERVER_ERROR, "Can't get node id"))?,
+        )
+        .map_err(|_| (StatusCode::INTERNAL_SERVER_ERROR, "Couldn't get node info"))?
+        .map(|n| n.unschedulable)
+        .unwrap_or(false);
+
+    Ok(serde_json::json!({
+        "package_name": PACKAGE_NAME,
+        "version": VERSION,
+        "nodes": nodes,
+        "schedulable": schedulable,
+    })
+    .to_string())
+}
+
+pub fn router<S>() -> axum::Router<S>
+where
+    S: ControlStateDelegate + NodeDelegate + Clone + 'static,
+{
+    use axum::routing::get;
+    axum::Router::new().route("/", get(health::<S>))
+}

crates/client-api/src/routes/mod.rs

@@ -1,5 +1,6 @@
 pub mod database;
 pub mod energy;
+pub mod health;
 pub mod identity;
 pub mod metrics;
 pub mod prometheus;

crates/standalone/src/routes/mod.rs

@@ -2,7 +2,7 @@ use http::header::{ACCEPT, AUTHORIZATION};
 use tower_http::cors::{Any, CorsLayer};
 
 use spacetimedb_client_api::{
-    routes::{database, energy, identity, metrics, prometheus},
+    routes::{database, energy, health, identity, metrics, prometheus},
     ControlStateDelegate, NodeDelegate,
 };
 
@@ -19,7 +19,8 @@ where
         .nest("/identity", identity::router(ctx.clone()))
         .nest("/energy", energy::router())
         .nest("/prometheus", prometheus::router())
-        .nest("/metrics", metrics::router());
+        .nest("/metrics", metrics::router())
+        .nest("/health", health::router());
 
     let cors = CorsLayer::new()
         .allow_headers([AUTHORIZATION, ACCEPT])