从支付超时踩坑说起：Go 生产环境下的 Context 治理与分布式链路追踪落地

// 典型反例：链路断裂 + 无法感知取消
func (s *OrderService) PaySuccess(ctx context.Context, orderID string) {
    // 异步执行后续逻辑
    go func() {
        // 坑点 1：使用 Background 导致 TraceID、SpanID 全部丢失
        // 坑点 2：如果下游任务堆积，无法实现级联超时控制
        err := s.notifyService.Send(context.Background(), orderID)
        if err != nil {
            log.Printf("notify failed: %v", err)
        }
    }()
}

// 正确姿势：保留 Trace 上下文，但不受父 Context 超时取消影响
func (s *OrderService) PaySuccess(ctx context.Context, orderID string) {
    // 剥离取消信号，保留 Tracing Baggage 和 TraceID
    asyncCtx := context.WithoutCancel(ctx)
    
    // 给异步任务单独附加合理的超时时间
    asyncCtx, cancel := context.WithTimeout(asyncCtx, 5*time.Second)
    
    go func() {
        defer cancel()
        // 此时 notifyService 依然能在日志和 APM 中打印出同一条 TraceID
        if err := s.notifyService.Send(asyncCtx, orderID); err != nil {
            s.logger.ErrorContext(asyncCtx, "notify failed", "order_id", orderID, "err", err)
        }
    }()
}

type contextKey struct{}

var traceKey = contextKey{} // 外部包无法覆盖这个 key

import (
    "net/http"
    "go.opentelemetry.io/otel"
    "go.opentelemetry.io/otel/propagation"
    "go.opentelemetry.io/otel/trace"
)

// 客户端发起调用：Inject（注入到 HTTP Header）
func DoHTTPRequest(ctx context.Context, url string) (*http.Response, error) {
    req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
    if err != nil {
        return nil, err
    }

    // 将 ctx 中的 Trace 上下文注入到 req.Header 中
    otel.GetTextMapPropagator().Inject(ctx, propagation.HeaderCarrier(req.Header))

    return http.DefaultClient.Do(req)
}

// 服务端接收请求：Extract（从 HTTP Header 提取）
func HTTPMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 从请求头提取 Trace 上下文
        ctx := otel.GetTextMapPropagator().Extract(r.Context(), propagation.HeaderCarrier(r.Header))
        
        tr := otel.Tracer("http-server")
        ctx, span := tr.Start(ctx, r.URL.Path, trace.WithSpanKind(trace.SpanKindServer))
        defer span.End()

        // 传递提取后的 Context
        next.ServeHTTP(w, r.WithContext(ctx))
    })
}