通八洲科技

如何在Golang中实现并发任务取消_Golang context取消机制示例

日期:2025-12-31 00:00 / 作者:P粉602998670
context.WithCancel 是最直接的取消触发方式,返回可取消的 Context 和 cancel 函数,调用后者协作式通知监听 goroutine 退出;必须传入 ctx 并用 select + ctx.Done() 检测取消,避免泄漏和误用。

context.WithCancel 是最直接的取消触发方式

当你需要手动控制一个任务的生命周期,比如用户点击“停止”按钮、超时前主动终止爬虫请求,context.WithCancel 是首选。它返回一个可取消的 context.Context 和一个 cancel 函数,调用后者即向所有监听该 context 的 goroutine 发送取消信号。

关键点在于:取消是**协作式**的,不是强制杀掉 goroutine;每个 goroutine 必须主动检查 ctx.Done() 并退出。

ctx, cancel := context.WithCancel(context.Background())
defer cancel() // 防止泄漏,但注意:仅适用于单次使用场景

go func(ctx context.Context) { for i := 0; i < 10; i++ { select { case <-time.After(time.Second): fmt.Println("working...", i) case <-ctx.Done(): fmt.Println("cancelled:", ctx.Err()) // context canceled return } } }(ctx)

time.Sleep(3 * time.Second) cancel()

select + ctx.Done() 是唯一可靠的取消检测模式

在 goroutine 内部,不能靠轮询 ctx.Err() != nil 来判断是否被取消——这会错过取消瞬间,且浪费 CPU。正确做法永远是用 select 等待 ctx.Done() 这个只读 channel。

ctx.Done() 在 context 被取消或超时时自动关闭,此时 select 分支可立即响应。若 context 未取消,该分支阻塞,不消耗资源。

context.WithTimeout 和 context.WithDeadline 本质相同,但语义不同

context.WithTimeout(ctx, 2*time.Second)context.WithDeadline(ctx, time.Now().Add(2*time.Second)) 最终行为一致:都会在约 2 秒后触发取消。区别在于可读性与适用场景:

注意:timeout 是相对当前时间的偏移量,不是从 goroutine 启动开始计时——它从 WithTimeout 调用那一刻起算。

子 context 的取消会自动传播到所有后代

Go 的 context 取消是树状传播的:父 context 被取消,所有通过 WithCancel/WithTimeout/WithDeadline 创建的子 context 也会同时被取消,无需手动管理层级。

但要注意:子 context 的取消**不会反向影响父 context**。也就是说,你可以在一个长生命周期的 context 上派生多个短期任务,各自独立取消,互不干扰。

真正难的不是写对那几行 select,而是确保整个调用链路每一层都尊重 ctx —— 少一个 select,就少一处取消入口。