diff --git a/src/api/health.rs b/src/api/health.rs index 1ac73d1..28beaca 100644 --- a/src/api/health.rs +++ b/src/api/health.rs @@ -4,3 +4,12 @@ use serde_json::{json, Value}; pub async fn health_check() -> (StatusCode, Json) { (StatusCode::OK, Json(json!({ "status": "ok", "version": env!("CARGO_PKG_VERSION") }))) } + +pub async fn readiness_check( + axum::extract::State(pool): axum::extract::State, +) -> axum::http::StatusCode { + match sqlx::query("SELECT 1").fetch_one(&pool).await { + Ok(_) => axum::http::StatusCode::OK, + Err(_) => axum::http::StatusCode::SERVICE_UNAVAILABLE, + } +}