如何从基于otelmux的Go OTEL追踪中提取Trace ID？

要从请求上下文中获取Trace ID关联日志，你可以通过OpenTelemetry的trace包直接提取当前Span的上下文信息，具体实现步骤如下：

1. 导入必要依赖 #

确保代码中引入OTEL的trace核心包：

import "go.opentelemetry.io/otel/trace"

2. 在请求Handler中提取Trace ID #

在处理请求的handler函数内，从请求上下文获取Span上下文，校验有效性后提取Trace ID：

func MyRequestHandler(w http.ResponseWriter, r *http.Request) {
    // 从请求上下文解析Span上下文
    spanCtx := trace.SpanContextFromContext(r.Context())
    
    // 仅处理被采样的有效追踪
    if spanCtx.IsValid() && spanCtx.IsSampled() {
        // 将Trace ID转为十六进制字符串（兼容DataDog格式）
        traceID := spanCtx.TraceID().String()
        // 注入到日志API，示例：
        // logger.Info("处理用户请求", "trace_id", traceID)
    }
    
    // 业务逻辑处理...
}

3. 封装工具函数复用（可选） #

如果多个handler需要提取Trace ID，可以封装成工具函数：

package tracing

import (
    "context"
    "go.opentelemetry.io/otel/trace"
)

// GetTraceID 从上下文提取有效Trace ID，无有效追踪时返回空串
func GetTraceID(ctx context.Context) string {
    spanCtx := trace.SpanContextFromContext(ctx)
    if spanCtx.IsValid() && spanCtx.IsSampled() {
        return spanCtx.TraceID().String()
    }
    return ""
}

调用示例：

traceID := tracing.GetTraceID(r.Context())
if traceID != "" {
    // 日志注入逻辑
}

关键说明 #

• 只有被采样的请求（IsSampled()返回true）才会生成有效Span上下文，未采样请求无法获取有效Trace ID。
• DataDog的OTEL实现完全兼容标准OTEL的Trace ID格式，转换后的字符串可直接在DataDog平台关联追踪数据与日志。

内容的提问来源于stack exchange，提问作者Alechko