Add a limit to the number of in-flight requests (#1166)

In order to avoid stuck collectors using up all system resources, add a limit to the number of parallel in-flight scrape requests. This will return a 503 error. Default to 40 requests, this seems like a reasonable number based on: * Two Prometheus servers scraping every 15 seconds. * Failing scrapes after 5 minutes of stuckness. Signed-off-by: Ben Kochie <superq@gmail.com>
6 years ago · ffefc8e74d
parent bcec99e0aa
commit ffefc8e74d
1 changed files with 11 additions and 4 deletions
--- a/node_exporter.go
+++ b/node_exporter.go
@ -36,12 +36,14 @@ type handler struct {
 	// the exporter itself.
 	exporterMetricsRegistry *prometheus.Registry
 	includeExporterMetrics  bool
+	maxRequests             int
 }

-func newHandler(includeExporterMetrics bool) *handler {
+func newHandler(includeExporterMetrics bool, maxRequests int) *handler {
 	h := &handler{
 		exporterMetricsRegistry: prometheus.NewRegistry(),
 		includeExporterMetrics:  includeExporterMetrics,
+		maxRequests:             maxRequests,
 	}
 	if h.includeExporterMetrics {
 		h.exporterMetricsRegistry.MustRegister(
@ -113,6 +115,7 @@ func (h *handler) innerHandler(filters ...string) (http.Handler, error) {
 		promhttp.HandlerOpts{
 			ErrorLog:            log.NewErrorLogger(),
 			ErrorHandling:       promhttp.ContinueOnError,
+			MaxRequestsInFlight: h.maxRequests,
 		},
 	)
 	if h.includeExporterMetrics {
@ -139,6 +142,10 @@ func main() {
 			"web.disable-exporter-metrics",
 			"Exclude metrics about the exporter itself (promhttp_*, process_*, go_*).",
 		).Bool()
+		maxRequests = kingpin.Flag(
+			"web.max-requests",
+			"Maximum number of parallel scrape requests. Use 0 to disable.",
+		).Default("40").Int()
 	)

 	log.AddFlags(kingpin.CommandLine)
@ -149,7 +156,7 @@ func main() {
 	log.Infoln("Starting node_exporter", version.Info())
 	log.Infoln("Build context", version.BuildContext())

-	http.Handle(*metricsPath, newHandler(!*disableExporterMetrics))
+	http.Handle(*metricsPath, newHandler(!*disableExporterMetrics, *maxRequests))
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte(`<html>
 			<head><title>Node Exporter</title></head>