Tracing: Enable traces to profiles (#88896)

pkg/infra/tracing/tracing.go

@@ -272,6 +272,8 @@ func (ots *TracingService) initOpentelemetryTracer() error {
 		}
 	}
 
+	tp = NewProfilingTracerProvider(tp)
+
 	// Register our TracerProvider as the global so any imported
 	// instrumentation in the future will default to using it
 	// only if tracing is enabled

pkg/infra/tracing/tracing_profiling.go

@@ -0,0 +1,27 @@
+package tracing
+
+import (
+	"context"
+
+	otelpyroscope "github.com/grafana/otel-profiling-go"
+	trace "go.opentelemetry.io/otel/trace"
+)
+
+type profilingTracerProvider struct {
+	trace.TracerProvider
+	wrappedTp tracerProvider
+}
+
+// NewProfilingTracerProvider creates a new tracer provider that annotates pprof
+// samples with span_id label. This allows to establish a relationship
+// between pprof profiles and reported tracing spans.
+func NewProfilingTracerProvider(tp tracerProvider) tracerProvider {
+	return &profilingTracerProvider{
+		TracerProvider: otelpyroscope.NewTracerProvider(tp),
+		wrappedTp:      tp,
+	}
+}
+
+func (tp *profilingTracerProvider) Shutdown(ctx context.Context) error {
+	return tp.wrappedTp.Shutdown(ctx)
+}