Go中使用WithCancel取消任务详解

使用context.WithCancel可主动取消协程任务，调用返回的cancel函数通知所有相关协程停止，通过select监听ctx.Done()判断取消信号，defer cancel确保资源释放。

在Golang中，context.WithCancel 是一种控制协程生命周期的重要方式。当你需要主动取消一个正在运行的任务时，可以使用它来通知所有相关协程停止工作，避免资源浪费或程序卡死。

创建可取消的Context

调用 context.WithCancel 会返回一个新的 context 和一个 cancel 函数。这个 cancel 函数用于触发取消信号，而 context 可以传递给其他函数或协程，用来监听是否被取消。

ctx, cancel := context.WithCancel(context.Background())
defer cancel() // 确保在函数退出时调用cancel

这里的 context.Background() 是根context，通常作为起始点。defer cancel() 能确保资源及时释放。

在协程中监听取消信号

启动一个协程执行长时间任务时，可以通过 select 监听 ctx.Done() 来判断是否收到取消请求。

go func(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("任务被取消:", ctx.Err())
            return
        default:
            fmt.Println("任务运行中...")
            time.Sleep(500 * time.Millisecond)
        }
    }
}(ctx)

当 cancel() 被调用后，ctx.Done() 通道会被关闭，select 就能立即感知并退出循环。

主动触发取消

在适当的时候调用 cancel() 函数即可中断任务。比如用户按下 Ctrl+C、超时发生或某个条件满足时。

// 模拟一段时间后取消
time.Sleep(2 * time.Second)
cancel()

一旦 cancel 执行，所有监听该 context 的协程都会收到通知，并应尽快清理资源和退出。

注意事项与最佳实践

• 总是调用 defer cancel()，防止 context 泄漏
• 将 context 作为第一个参数传入函数，符合Go惯例
• 不要把 cancel 函数传给下游，除非明确需要由下游控制取消
• 检查 ctx.Err() 可知具体取消原因（如 canceled 或 deadline exceeded）

基本上就这些。合理使用 context.WithCancel 能让你的程序更健壮、可控。关键是传递 context 并在每个协程中响应取消信号。不复杂但容易忽略细节。