Go Context 生产实践：超时控制、Goroutine 泄漏排查与避坑指南

// 错误示范：如果 doSomeRPC 提前返回或 panic，定时器没有被清理，会造成短期的资源滞留
func fetchDataBad(ctx context.Context) error {
    ctx, _ = context.WithTimeout(ctx, 2*time.Second)
    return doSomeRPC(ctx)
}

// 正确姿势：哪怕超时了，也要立刻显式调用 cancel()
func fetchDataGood(ctx context.Context) error {
    ctx, cancel := context.WithTimeout(ctx, 2*time.Second)
    defer cancel() // 归还底层 timer 资源，防止 Goroutine 滞留
    return doSomeRPC(ctx)
}

// 错误示范：全局用 string 做 key，随时可能被第三方包或下游覆盖
ctx = context.WithValue(ctx, "trace_id", "xyz-123")

// 正确姿势：使用非导出的结构体类型作为 key
type traceKey struct{}

func WithTraceID(ctx context.Context, traceID string) context.Context {
    return context.WithValue(ctx, traceKey{}, traceID)
}

func FromTraceID(ctx context.Context) (string, bool) {
    val, ok := ctx.Value(traceKey{}).(string)
    return val, ok
}

func BenchmarkContextChain(b *testing.B) {
    baseCtx := context.Background()
    b.Run("WithCancel", func(b *testing.B) {
        b.ReportAllocs()
        for i := 0; i < b.N; i++ {
            ctx, cancel := context.WithCancel(baseCtx)
            cancel()
            _ = ctx
        }
    })

    b.Run("WithValue-10-Layers", func(b *testing.B) {
        b.ReportAllocs()
        for i := 0; i < b.N; i++ {
            ctx := baseCtx
            for j := 0; j < 10; j++ {
                ctx = context.WithValue(ctx, j, j)
            }
            _ = ctx.Value(0)
        }
    })
}

goos: darwin
goarch: arm64
BenchmarkContextChain/WithCancel-8              17441695        68.45 ns/op       64 B/op        1 allocs/op
BenchmarkContextChain/WithValue-10-Layers-8      2147312       558.20 ns/op      480 B/op       10 allocs/op

func handleRequest(ctx context.Context) {
    // 继承所有上下文数据（TraceID 等），但剥离超时和取消信号
    asyncCtx := context.WithoutCancel(ctx)

    go func() {
        // 请求返回后，这里依然可以正常执行，不会被外层的超时干掉
        writeAuditLog(asyncCtx)
    }()
}

stop := context.AfterFunc(ctx, func() {
    log.Println("Context canceled, cleaning up resources...")
    conn.Close()
})
// 如果提前正常处理完了，取消掉这个回调即可
defer stop()