Merge pull request #71 from lyt122/bugfix

bugfix:fix the traceid cannot be shown in the response
CocaineCong · Mar 3, 2024 · 6a125d2 · 6a125d2
2 parents 17afc7e + d6a554d
commit 6a125d2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/tracing/tracing.go b/pkg/tracing/tracing.go
@@ -19,6 +19,7 @@ package tracing
 
 import (
 	"context"
+	"github.com/gin-gonic/gin"
 
 	"go.opentelemetry.io/otel/trace"
 
@@ -68,8 +69,8 @@ func newResource(serviceName string) *resource.Resource {
 	)
 }
 
-func GetTraceID(ctx context.Context) string {
-	spanCtx := trace.SpanContextFromContext(ctx)
+func GetTraceID(ctx *gin.Context) string {
+	spanCtx := trace.SpanContextFromContext(ctx.Request.Context())
 	if spanCtx.HasTraceID() {
 		traceID := spanCtx.TraceID()
 		return traceID.String()