zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore(trace): improve rest tracinghandler (#2783)

Browse Source
This commit is contained in:
cong
2023-01-12 12:50:57 +08:00
committed by GitHub
parent a2dcb0079a
commit 43b8c7f641
1 changed files with 33 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rest/handler/tracinghandler.go
View File

@@ -31,7 +31,7 @@ func TracingHandler(serviceName, path string, opts ...TracingOption) func(http.H
ignorePaths := collection.NewSet()
ignorePaths.AddStr(tracingOpts.traceIgnorePaths...)
traceHandler := func(checkIgnore bool) func(http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
propagator := otel.GetTextMapPropagator()
tracer := otel.GetTracerProvider().Tracer(trace.TraceName)
@@ -42,7 +42,7 @@ func TracingHandler(serviceName, path string, opts ...TracingOption) func(http.H
spanName = r.URL.Path
}
if ignorePaths.Contains(spanName) {
if checkIgnore && ignorePaths.Contains(spanName) {
next.ServeHTTP(w, r)
return
}
@@ -68,10 +68,13 @@ func TracingHandler(serviceName, path string, opts ...TracingOption) func(http.H
})
}
}
checkIgnore := ignorePaths.Count() > 0
return traceHandler(checkIgnore)
}
// WithTraceIgnorePaths specifies the traceIgnorePaths option for TracingHandler.
func WithTraceIgnorePaths(traceIgnorePaths []string) TracingOption {
return func(options *tracingOptions) {
options.traceIgnorePaths = traceIgnorePaths
options.traceIgnorePaths = append(options.traceIgnorePaths, traceIgnorePaths...)
}
}