fix(api): http pprof labels (#3441)

ovh · Oct 11, 2018 · 13a2388 · 13a2388
1 parent cc8b7fb
commit 13a2388
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/engine/api/router.go b/engine/api/router.go
@@ -83,9 +83,10 @@ type HandlerConfigFunc func(service.Handler, ...HandlerConfigParam) *service.Han
 func (r *Router) pprofLabel(fn http.HandlerFunc) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		labels := pprof.Labels("http-path", r.URL.Path)
-		pprof.Do(r.Context(), labels, func(ctx context.Context) {
-			fn.ServeHTTP(w, r)
-		})
+		ctx := pprof.WithLabels(r.Context(), labels)
+		pprof.SetGoroutineLabels(ctx)
+		r = r.WithContext(ctx)
+		fn(w, r)
 	}
 }