feat: add some traceparent utils

telemetry/tracev2/memory/trace.go

@@ -33,8 +33,14 @@ func (t *memoryTrace) Start(ctx context.Context, name string) context.Context {
 	parentCtxCfg := trace.SpanContextConfig{}
 
 	if t.span == nil {
+		if traceparent, ok := tracev2.TraceParentFromContext(ctx); ok {
+			parentCtxCfg.TraceID = traceparent
+		}
+
 		newCtx, span := t.start(ctx, name, parentCtxCfg)
+
 		t.span = span
+
 		return newCtx
 	}
 

telemetry/tracev2/utils.go

@@ -0,0 +1,28 @@
+package tracev2
+
+import (
+	"context"
+
+	"github.com/w-h-a/pkg/utils/metadatautils"
+)
+
+const (
+	traceParentKey = "traceparent"
+)
+
+func ContextWithTraceParent(ctx context.Context, traceparent [16]byte) (context.Context, error) {
+	return metadatautils.MergeContext(ctx, map[string]string{
+		traceParentKey: string(traceparent[:]),
+	}, true), nil
+}
+
+func TraceParentFromContext(ctx context.Context) (traceparent [16]byte, found bool) {
+	traceId, ok := metadatautils.GetContext(ctx, traceParentKey)
+	if !ok {
+		return
+	}
+
+	copy(traceparent[:], traceId)
+
+	return
+}