2. context包概览：核心上下文创建与取消机制

在 Go 语言中，context 包是并发控制与超时传播的基石。本章将深入剖析 context.Background、context.TODO 的用途，以及 context.WithCancel 创建可取消上下文的完整机制。理解这些基础上下文类型，是构建可靠中间件链路的第一步。

2.1 根上下文：Background 与 TODO

所有上下文树的起点都来自两个根上下文函数。它们本身不可取消、没有超时、不携带值，但承担着不同的语义角色。

2.2 可取消上下文：WithCancel 机制

context.WithCancel 是构建并发取消链的核心函数。它基于父上下文创建一个新的子上下文，并返回一个 CancelFunc 用于手动触发取消。

// 创建可取消上下文
ctx, cancel := context.WithCancel(context.Background())

// 在 goroutine 中监听取消信号
go func() {
    select {
    case <-ctx.Done():
        fmt.Println("收到取消信号，清理资源")
        return
    case result := <-workCh:
        fmt.Println("任务完成:", result)
    }
}()

// 主逻辑完成后主动取消
cancel()

2.3 取消传播链：父取消则子取消

上下文取消具有严格的向下传播特性：当父上下文被取消时，所有从它派生的子上下文都会自动收到取消信号。这种机制天然适合构建中间件链路中的超时与错误传播。

2.4 上下文树结构图

下图展示了从 Background 根节点出发，通过 WithCancel 构建的上下文树结构，以及取消信号的传播路径。

2.5 取消传播的代码验证

以下示例演示了父上下文取消后，所有子上下文自动收到信号的完整流程：

func main() {
    // 创建根上下文
    rootCtx := context.Background()

    // 创建可取消的父上下文
    parentCtx, parentCancel := context.WithCancel(rootCtx)

    // 从父上下文派生两个子上下文
    childCtx1, _ := context.WithCancel(parentCtx)
    childCtx2, _ := context.WithCancel(parentCtx)

    // 启动 goroutine 监听子上下文
    go func() {
        <-childCtx1.Done()
        fmt.Println("childCtx1 已取消，原因:", childCtx1.Err())
    }()
    go func() {
        <-childCtx2.Done()
        fmt.Println("childCtx2 已取消，原因:", childCtx2.Err())
    }()

    // 取消父上下文
    parentCancel()

    // 等待 goroutine 输出
    time.Sleep(100 * time.Millisecond)
    // 输出:
    // childCtx1 已取消，原因: context canceled
    // childCtx2 已取消，原因: context canceled
}

2.6 常见错误与避坑指南

• 忘记调用 cancel： 导致上下文及其派生资源无法释放，造成内存泄漏。始终使用 defer cancel()。
• 在多个 goroutine 中同时调用 cancel： CancelFunc 是并发安全的，多次调用只有第一次生效，后续调用无副作用。
• 将 TODO 用于生产代码： TODO 仅作为临时占位，长期遗留会降低代码可维护性。应在确定上下文策略后替换为 Background 或具体派生上下文。
• 忽略 ctx.Err() 的返回值： 取消后应检查错误类型，区分是主动取消还是超时，以便采取不同的处理策略。