diff --git a/net/gtrace/gtrace.go b/net/gtrace/gtrace.go index 6051e33ab..e5b55c819 100644 --- a/net/gtrace/gtrace.go +++ b/net/gtrace/gtrace.go @@ -144,14 +144,14 @@ func GetBaggageVar(ctx context.Context, key string) *gvar.Var { return NewBaggage(ctx).GetVar(key) } -// SetTraceID injects custom trace id into context to propagate. -func SetTraceID(ctx context.Context, traceID trace.TraceID) context.Context { +// WithTraceID injects custom trace id into context to propagate. +func WithTraceID(ctx context.Context, traceID trace.TraceID) context.Context { sc := trace.SpanContextFromContext(ctx) if !sc.HasTraceID() { var ( span trace.Span ) - ctx, span = NewSpan(ctx, "gtrace.SetTraceID") + ctx, span = NewSpan(ctx, "gtrace.WithTraceID") defer span.End() sc = trace.SpanContextFromContext(ctx) } diff --git a/net/gtrace/gtrace_z_unit_feature_http_test.go b/net/gtrace/gtrace_z_unit_feature_http_test.go index 9daf88b85..0ccc7e53e 100644 --- a/net/gtrace/gtrace_z_unit_feature_http_test.go +++ b/net/gtrace/gtrace_z_unit_feature_http_test.go @@ -46,7 +46,7 @@ func Test_Client_Server_Tracing(t *testing.T) { }) } -func Test_SetTraceID(t *testing.T) { +func Test_WithTraceID(t *testing.T) { gtest.C(t, func(t *gtest.T) { p := 8889 s := g.Server(p) @@ -61,7 +61,7 @@ func Test_SetTraceID(t *testing.T) { time.Sleep(100 * time.Millisecond) - ctx := gtrace.SetTraceID(context.TODO(), traceID) + ctx := gtrace.WithTraceID(context.TODO(), traceID) prefix := fmt.Sprintf("http://127.0.0.1:%d", p) client := g.Client()