Improve health probe

tiberiuv · Feb 24, 2025 · b46e196 · b46e196
1 parent 6011732
commit b46e196
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/api.rs b/src/api.rs
@@ -173,7 +173,19 @@ async fn check_ip(
     }
 }
 
-async fn health() -> impl IntoResponse {
+async fn health(
+    State(app): State<App>,
+    ExtractRealIp(real_client_ip): ExtractRealIp,
+    request: Request,
+) -> impl IntoResponse {
+    let result = app
+        .appsec_client
+        .appsec_request(request, real_client_ip, app.config.proxy_headers)
+        .await;
+    if let Err(err) = result {
+        tracing::error!(?err);
+        return StatusCode::INTERNAL_SERVER_ERROR.into_response();
+    };
     StatusCode::OK.into_response()
 }