import (
"go.opentelemetry.io/otel/trace"
// "github.com/zeromicro/go-zero/core/limit"
"github.com/zeromicro/go-zero/core/logx"
...
)
func New Logic(ctx context.Context, svcCtx *svc.ServiceContext) *StockOrderLogic {
// 为每条消息生成独立的 trace ID
// UUID 格式: "550e8400-e29b-41d4-a716-446655440000"
// OpenTelemetry TraceID 需要 32 个十六进制字符(去掉连字符)
// OpenTelemetry SpanID 需要 16 个十六进制字符
traceUUID := uuid.New()
spanUUID := uuid.New()
// 将 UUID 转换为十六进制字符串(去掉连字符)
traceIDHex := strings.ReplaceAll(traceUUID.String(), "-", "")
spanIDHex := strings.ReplaceAll(spanUUID.String(), "-", "")
// SpanID 只需要前 16 个字符
if len(spanIDHex) > 16 {
spanIDHex = spanIDHex[:16]
} else if len(spanIDHex) < 16 {
// 如果不足 16 个字符,用 0 补齐
spanIDHex = spanIDHex + strings.Repeat("0", 16-len(spanIDHex))
}
// 创建 OpenTelemetry trace ID 和 span ID
traceIDBytes, err := trace.TraceIDFromHex(traceIDHex)
if err != nil {
// 如果解析失败,使用空值(但这种情况不应该发生)
traceIDBytes = trace.TraceID{}
}
spanIDBytes, err := trace.SpanIDFromHex(spanIDHex)
if err != nil {
// 如果解析失败,使用空值
spanIDBytes = trace.SpanID{}
}
// 创建 span context
spanCtx := trace.NewSpanContext(trace.SpanContextConfig{
TraceID: traceIDBytes,
SpanID: spanIDBytes,
Remote: false,
})
// 将 span context 注入到 context 中
ctx = trace.ContextWithSpanContext(ctx, spanCtx)
// 同时设置自定义 key(备用,确保兼容性)
ctx = context.WithValue(ctx, traceIDKey, traceIDHex)
ctx = context.WithValue(ctx, spanIDKey, spanIDHex)
return &BussLogic{
Logger: logx.WithContext(ctx),
ctx: ctx,
svcCtx: svcCtx,
}
}