Tracing: Helper function to retrieve trace ID from context (#1027)

grafana · Jul 10, 2024 · 7c8ae12 · 7c8ae12
1 parent db9eddd
commit 7c8ae12
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/backend/tracing/tracing.go b/backend/tracing/tracing.go
@@ -1,6 +1,7 @@
 package tracing
 
 import (
+	"context"
 	"sync"
 
 	"go.opentelemetry.io/otel"
@@ -40,3 +41,12 @@ func DefaultTracer() trace.Tracer {
 func InitDefaultTracer(tracer trace.Tracer) {
 	defaultTracer = &contextualTracer{tracer: tracer}
 }
+
+func TraceIDFromContext(ctx context.Context, requireSampled bool) string {
+	spanCtx := trace.SpanContextFromContext(ctx)
+	if !spanCtx.HasTraceID() || !spanCtx.IsValid() || (requireSampled && !spanCtx.IsSampled()) {
+		return ""
+	}
+
+	return spanCtx.TraceID().String()
+}