Merge pull request #18 from blockpane/develop

Prometheus should use it's own servemux/server
blockpane · Jul 2, 2022 · e4984d1 · e4984d1
2 parents e695c35 + 6bf5dc7
commit e4984d1
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/td2/prometheus.go b/td2/prometheus.go
@@ -9,6 +9,7 @@ import (
 	"log"
 	"net/http"
 	"sync"
+	"time"
 )
 
 var (
@@ -156,7 +157,17 @@ func prometheusExporter(ctx context.Context, updates chan *promUpdate) {
 		}
 	}()
 
+	promMux := http.NewServeMux()
+
 	l("serving prometheus metrics at 0.0.0.0:%d/metrics", td.PrometheusListenPort)
-	http.Handle("/metrics", promhttp.Handler())
-	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", td.PrometheusListenPort), nil))
+	promMux.Handle("/metrics", promhttp.Handler())
+	promSrv := &http.Server{
+		Addr:              fmt.Sprintf(":%d", td.PrometheusListenPort),
+		Handler:           promMux,
+		ReadTimeout:       20 * time.Second,
+		WriteTimeout:      20 * time.Second,
+		IdleTimeout:       120 * time.Second,
+		ReadHeaderTimeout: 20 * time.Second,
+	}
+	log.Fatal(promSrv.ListenAndServe())
 }