github/wonderwall
1
0
Fork 0
mirror of https://github.com/nais/wonderwall.git synced 2026-05-10 18:37:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(middleware): use trace_id as correlation id, if available

Browse Source 
Co-authored-by: sindrerh2 <sindre.rodseth.hansen@nav.no>
This commit is contained in:
Trong Huu Nguyen
2025-01-28 10:56:32 +01:00
parent 1f730a3d68
commit 98cc534806
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/middleware/correlationid.go
View File

@@ -6,6 +6,7 @@ import (
chi_middleware "github.com/go-chi/chi/v5/middleware"
"github.com/google/uuid"
"go.opentelemetry.io/otel/trace"
)
func CorrelationIDHandler(next http.Handler) http.Handler {
@@ -16,6 +17,11 @@ func CorrelationIDHandler(next http.Handler) http.Handler {
}
ctx := r.Context()
span := trace.SpanFromContext(ctx)
if span.SpanContext().HasTraceID() {
id = span.SpanContext().TraceID().String()
}
ctx = context.WithValue(ctx, chi_middleware.RequestIDKey, id)
next.ServeHTTP(w, r.WithContext(ctx))
}