mirror of
https://gitee.com/johng/gf.git
synced 2024-12-04 05:07:44 +08:00
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gtrace_test
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gogf/gf/net/gtrace"
|
|
"github.com/gogf/gf/test/gtest"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/oteltest"
|
|
"go.opentelemetry.io/otel/trace"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
traceIDStr = "4bf92f3577b34da6a3ce929d0e0e4736"
|
|
spanIDStr = "00f067aa0ba902b7"
|
|
)
|
|
|
|
var (
|
|
traceID = mustTraceIDFromHex(traceIDStr)
|
|
spanID = mustSpanIDFromHex(spanIDStr)
|
|
)
|
|
|
|
func mustTraceIDFromHex(s string) (t trace.TraceID) {
|
|
var err error
|
|
t, err = trace.TraceIDFromHex(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return
|
|
}
|
|
|
|
func mustSpanIDFromHex(s string) (t trace.SpanID) {
|
|
var err error
|
|
t, err = trace.SpanIDFromHex(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return
|
|
}
|
|
|
|
func TestNewCarrier(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
ctx := trace.ContextWithRemoteSpanContext(context.Background(), trace.NewSpanContext(trace.SpanContextConfig{
|
|
TraceID: traceID,
|
|
SpanID: spanID,
|
|
TraceFlags: trace.FlagsSampled,
|
|
}))
|
|
ctx, _ = oteltest.DefaultTracer().Start(ctx, "inject")
|
|
carrier1 := gtrace.NewCarrier()
|
|
otel.GetTextMapPropagator().Inject(ctx, carrier1)
|
|
t.Assert(carrier1.String(), `{"traceparent":"00-4bf92f3577b34da6a3ce929d0e0e4736-0000000000000002-01","tracestate":""}`)
|
|
|
|
ctx = otel.GetTextMapPropagator().Extract(ctx, carrier1)
|
|
gotSc := trace.RemoteSpanContextFromContext(ctx)
|
|
t.Assert(gotSc.TraceID().String(), traceID.String())
|
|
t.Assert(gotSc.SpanID().String(), "0000000000000002")
|
|
})
|
|
}
|