fix: placement of context

kitabisa · Aug 31, 2024 · 8187018 · 8187018
1 parent 16d6d4a
commit 8187018
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/instrumentation/github.com/gorilla/mux/otelmux/mux.go b/instrumentation/github.com/gorilla/mux/otelmux/mux.go
@@ -171,12 +171,12 @@ func (tw traceware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		w.Header().Add(tw.traceResponseHeaderKey, span.SpanContext().TraceID().String())
 	}
 
-	r2 := r.WithContext(ctx)
 	rrw := getRRW(w)
 	defer putRRW(rrw)
 
 	ctx = context.WithValue(ctx, "trace-id", span.SpanContext().TraceID().String())
-
+	r2 := r.WithContext(ctx)
+
 	tw.handler.ServeHTTP(rrw.writer, r2)
 	if rrw.status > 0 {
 		span.SetAttributes(semconv.HTTPStatusCode(rrw.status))