package tracing import ( "context" "net/http" "github.com/containous/alice" "github.com/containous/traefik/v2/pkg/log" "github.com/containous/traefik/v2/pkg/middlewares" "github.com/containous/traefik/v2/pkg/tracing" "github.com/opentracing/opentracing-go" "github.com/opentracing/opentracing-go/ext" ) const ( entryPointTypeName = "TracingEntryPoint" ) // NewEntryPoint creates a new middleware that the incoming request. func NewEntryPoint(ctx context.Context, t *tracing.Tracing, entryPointName string, next http.Handler) http.Handler { log.FromContext(middlewares.GetLoggerCtx(ctx, "tracing", entryPointTypeName)).Debug("Creating middleware") return &entryPointMiddleware{ entryPoint: entryPointName, Tracing: t, next: next, } } type entryPointMiddleware struct { *tracing.Tracing entryPoint string next http.Handler } func (e *entryPointMiddleware) ServeHTTP(rw http.ResponseWriter, req *http.Request) { spanCtx, _ := e.Extract(opentracing.HTTPHeaders, tracing.HTTPHeadersCarrier(req.Header)) span, req, finish := e.StartSpanf(req, ext.SpanKindRPCServerEnum, "EntryPoint", []string{e.entryPoint, req.Host}, " ", ext.RPCServerOption(spanCtx)) defer finish() ext.Component.Set(span, e.ServiceName) tracing.LogRequest(span, req) req = req.WithContext(tracing.WithTracing(req.Context(), e.Tracing)) recorder := newStatusCodeRecoder(rw, http.StatusOK) e.next.ServeHTTP(recorder, req) tracing.LogResponseCode(span, recorder.Status()) } // WrapEntryPointHandler Wraps tracing to alice.Constructor. func WrapEntryPointHandler(ctx context.Context, tracer *tracing.Tracing, entryPointName string) alice.Constructor { return func(next http.Handler) (http.Handler, error) { return NewEntryPoint(ctx, tracer, entryPointName, next), nil } }