configure prometheus handler to limit maximum requests in flight and …

…maximum request execution time
hipages · Jan 25, 2024 · 57e431b · 57e431b
1 parent 8303654
commit 57e431b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/cmd/server.go b/cmd/server.go
@@ -71,7 +71,15 @@ to quickly create a Cobra application.`,
 			IdleTimeout:  time.Second * 60,
 		}
 
-		http.Handle(metricsEndpoint, promhttp.Handler())
+		opts := promhttp.HandlerOpts{
+			Timeout:             5 * time.Second,
+			MaxRequestsInFlight: 2,
+		}
+		handler := promhttp.InstrumentMetricHandler(
+			prometheus.DefaultRegisterer,
+			promhttp.HandlerFor(prometheus.DefaultGatherer, opts),
+		)
+		http.Handle(metricsEndpoint, handler)
 		http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 			_, err := w.Write([]byte(`<html>
 			 <head><title>php-fpm_exporter</title></head>