Merge pull request #963 from Dstack-TEE/codex/feat-kms-startup-health

feat(kms): expose a startup health endpoint

Author: kvinwang

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<KmsState>) -> RawText<String> {
     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 {