Inject Pyroscope middleware into the call chain

grafana · Aug 2, 2024 · e42ddb0 · e42ddb0
1 parent 2d2a80e
commit e42ddb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/http/http.go b/pkg/http/http.go
@@ -145,7 +145,7 @@ func (s *Server) WithPrometheus() *Server {
 
 // WithProfiling adds a middleware that extracts k6 labels from the baggage and adds them to the context.
 func (s *Server) WithProfiling() *Server {
- s.router.Use(k6.LabelsFromBaggageHandler)
+ s.router = s.router.With(k6.LabelsFromBaggageHandler)
 
  return s
 }