使用 golang otelmux 检测 OTEL 跟踪，获取 trace_id 的方法

从现在开始，努力学习吧！本文《使用 golang otelmux 检测 OTEL 跟踪，获取 trace_id 的方法》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

在我的 golang 应用程序中，我使用 otelmux 中间件和 ddotel 跟踪器提供程序来检测 OTEL 跟踪中间件：

package tracing

import (
    "github.com/gorilla/mux"
    "go.opentelemetry.io/contrib/instrumentation/github.com/gorilla/mux/otelmux"
    ddotel "gopkg.in/DataDog/dd-trace-go.v1/ddtrace/opentelemetry"
    "gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
)

// Middleware returns a tracing middleware that will add a span for each request.
func Middleware(t *TraceProvider) mux.MiddlewareFunc {
    sampler := tracer.NewRateSampler(0.1)
    provider := ddotel.NewTracerProvider(
        tracer.WithSampler(sampler), 
        tracer.WithRuntimeMetrics(),
    )
    return otelmux.Middleware("my-service", otelmux.WithTracerProvider(provider))
}

每当跟踪附加到请求时（考虑采样率），我想从根跨度中提取跟踪 ID，这样我就可以将其注入到日志记录 API 中，以便关联来自该请求的日志具有来自同一请求的跟踪。

到目前为止，我还没有找到直接从请求上下文获取跟踪 ID 的方法。

如何提取跟踪 ID？

正确答案

感谢 @hamed-n 评论的最佳解决方案：

trace包中有一个从span上下文中提取trace ID的函数：“go.opentelemetry.io/otel/trace”：

func GetTraceID(ctx context.Context) string {
    spanCtx := trace.SpanContextFromContext(ctx)
    if spanCtx.HasTraceID() {
        traceID := spanCtx.TraceID()
        return traceID.String()
    }
    return ""
}

就我而言，由于我使用的提供程序是 DataDog，因此我必须将跟踪 ID 转换为 uint64 big endian：

func GetTraceID(ctx context.Context) string {
    spanCtx := trace.SpanContextFromContext(ctx)
    if spanCtx.HasTraceID() {
        // Since datadog trace provider (ddtrace) uses big endian uint64 for the trace ID, we must first to first convert it back to uint64.
        traceID := spanCtx.TraceID()
        traceIDRaw := [16]byte(traceID)
        traceIDUint64 := byteArrToUint64(traceIDRaw[8:])
        traceIDStr := strconv.FormatUint(traceIDUint64, 10)
        return traceIDStr
    }
    return ""
}

func byteArrToUint64(buf []byte) uint64 {
    var x uint64
    for i, b := range buf {
        x = x<<8 + uint64(b)
        if i == 7 {
            return x
        }
    }
    return x
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。