Specific endpoint for prometheus exporter health checking (metrics-rs…

…#435)
mnpw · Jan 14, 2024 · a83f591 · a83f591
1 parent 27a5ff3
commit a83f591
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/metrics-exporter-prometheus/src/builder.rs b/metrics-exporter-prometheus/src/builder.rs
@@ -435,13 +435,15 @@ impl PrometheusBuilder {
                         let handle = handle.clone();
 
                         async move {
-                            Ok::<_, hyper::Error>(service_fn(move |_| {
+                            Ok::<_, hyper::Error>(service_fn(move |req| {
                                 let handle = handle.clone();
 
                                 async move {
                                     if is_allowed {
-                                        let output = handle.render();
-                                        Ok::<_, hyper::Error>(Response::new(Body::from(output)))
+                                        Ok::<_, hyper::Error>(match req.uri().path() {
+                                            "/health" => Response::new(Body::from("OK")),
+                                            _ => Response::new(Body::from(handle.render())),
+                                        })
                                     } else {
                                         Ok::<_, hyper::Error>(
                                             Response::builder()