4. WithCancel：手动取消的 Context 实现原理与实战

context.WithCancel 是 Go 标准库中最基础、最常用的手动取消机制。它返回一个可取消的 Context 副本，以及一个 cancel 函数，调用该函数会向该 Context 及其所有衍生子 Context 广播取消信号。理解 WithCancel 的实现原理，是掌握 Context 超时控制与链路取消的基石。

4.1 核心概念：取消树与信号传播

WithCancel 的核心思想是构建一棵 取消传播树：

• 父 Context：调用 WithCancel 时传入的 Context（可以是 Background、TODO 或其他 Context）。
• 子 Context：WithCancel 返回的新 Context，内部持有一个 取消通道（done channel）。
• cancel 函数：调用时关闭子 Context 的 done 通道，并递归取消所有基于该子 Context 衍生的后代 Context。

4.2 内部数据结构

Go 标准库中，WithCancel 返回的 Context 底层是一个 cancelCtx 结构体：

type cancelCtx struct {
    Context

    mu       sync.Mutex            // 保护以下字段
    done     chan struct{}         // 关闭时表示取消
    children map[canceler]struct{} // 存储所有子 canceler
    err      error                 // 取消原因
}

4.3 取消流程：三步走

1. 加锁：获取 cancelCtx.mu，保证并发安全。
2. 关闭 done 通道：通过 close(c.done) 广播取消信号。所有通过 ctx.Done() 监听的 goroutine 都会收到通知。
3. 递归取消子 Context：遍历 children map，对每个子 canceler 调用其 cancel() 方法，形成链式传播。

4.4 实战模式：goroutine 生命周期管理

WithCancel 最常见的实战场景是 启动多个 goroutine 并统一取消：

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel() // 确保 main 退出时取消所有子 goroutine

    // 启动工作 goroutine
    go worker(ctx, "worker-1")
    go worker(ctx, "worker-2")

    // 模拟运行 2 秒后手动取消
    time.Sleep(2 * time.Second)
    cancel() // 手动触发取消

    // 等待 goroutine 退出（实际场景建议用 WaitGroup）
    time.Sleep(500 * time.Millisecond)
    fmt.Println("main exit")
}

func worker(ctx context.Context, name string) {
    for {
        select {
        case <-ctx.Done():
            fmt.Printf("%s: received cancel signal, exiting\n", name)
            return
        default:
            // 模拟工作
            fmt.Printf("%s: working...\n", name)
            time.Sleep(500 * time.Millisecond)
        }
    }
}

4.5 WithCancel 与取消传播的 SVG 架构图

4.6 性能与最佳实践

4.7 与 WithTimeout / WithDeadline 的关系

WithTimeout 和 WithDeadline 内部实际上也是基于 WithCancel 实现的：

• 它们创建一个 cancelCtx，并启动一个后台 goroutine 在超时后自动调用 cancel 函数。
• 因此，理解 WithCancel 的取消传播机制，就理解了整个 Context 取消体系的核心。