use context from middleware in proxy

Signed-off-by: Jörn Friedrich Dreyer <jfd@butonic.de>
This commit is contained in:
Jörn Friedrich Dreyer
2021-07-05 15:43:50 +02:00
parent 2583c6a1a7
commit 5c024b4543

View File

@@ -1,7 +1,6 @@
package proxy
import (
"context"
"crypto/tls"
"net"
"net/http"
@@ -213,12 +212,12 @@ func (p *MultiHostReverseProxy) AddHost(policy string, target *url.URL, rt confi
}
func (p *MultiHostReverseProxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
ctx := context.Background()
ctx := r.Context()
var span *trace.Span
// Start root span.
if p.config.Tracing.Enabled {
ctx, span = trace.StartSpan(context.Background(), r.URL.String())
ctx, span = trace.StartSpan(ctx, r.URL.String())
defer span.End()
p.propagator.SpanContextToRequest(span.SpanContext(), r)
}