http_proxy: allow specifying middleware prometheus opts

saucelabs · Jun 11, 2024 · 8efb92f · 8efb92f
1 parent 9ccab01
commit 8efb92f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/http_proxy.go b/http_proxy.go
@@ -93,6 +93,7 @@ type HTTPProxyConfig struct {
 	ConnectTimeout    time.Duration
 	ReadLimit         SizeSuffix
 	WriteLimit        SizeSuffix
+	PromHTTPOpts      []middleware.PrometheusOpt
 
 	// TestingHTTPHandler uses Martian's [http.Handler] implementation
 	// over [http.Server] instead of the default TCP server.
@@ -366,7 +367,7 @@ func (hp *HTTPProxy) middlewareStack() (martian.RequestResponseModifier, *martia
 	}
 
 	if hp.config.PromRegistry != nil {
-		p := middleware.NewPrometheus(hp.config.PromRegistry, hp.config.PromNamespace)
+		p := middleware.NewPrometheus(hp.config.PromRegistry, hp.config.PromNamespace, hp.config.PromHTTPOpts...)
 		stack.AddRequestModifier(p)
 		stack.AddResponseModifier(p)