diff --git a/dstack/kms/src/main.rs b/dstack/kms/src/main.rs index 82a9d6b8c..fde993d95 100644 --- a/dstack/kms/src/main.rs +++ b/dstack/kms/src/main.rs @@ -63,7 +63,7 @@ async fn run_onboard_service(kms_config: KmsConfig, figment: Figment) -> Result< // Remove section tls let rocket = rocket::custom(figment) - .mount("/", rocket::routes![index, finish]) + .mount("/", rocket::routes![index, finish, health]) .mount( "/prpc", ra_rpc::prpc_routes!(OnboardState, OnboardHandler, trim: "Onboard."), @@ -80,6 +80,11 @@ async fn run_onboard_service(kms_config: KmsConfig, figment: Figment) -> Result< Ok(()) } +#[rocket::get("/health")] +fn health() -> RawText<&'static str> { + RawText("OK") +} + #[rocket::get("/metrics")] fn metrics(state: &State) -> RawText { RawText(state.metrics().render_prometheus()) @@ -184,6 +189,7 @@ async fn main() -> Result<()> { "/prpc", ra_rpc::prpc_routes!(KmsState, RpcHandler, trim: "KMS."), ) + .mount("/", rocket::routes![health]) .manage(state.clone()); if metrics_enabled {