context.WithCancel用于创建可取消的上下文,通过cancel()通知协程停止;示例中启动协程监听ctx.Done(),主函数1.5秒后调用cancel()发送取消信号,协程收到后退出,最后打印“已调用cancel”并等待结束。

在Go语言中,context.WithCancel 用于创建一个可手动取消的上下文(context),常用于控制协程(goroutine)的生命周期。当你希望主动终止某个正在运行的操作时,比如超时、用户中断或任务完成,可以调用取消函数来通知所有监听该 context 的协程停止工作。
基本用法:创建可取消的 Context
使用 context.WithCancel 会返回一个新的 context 和一个取消函数(cancel function)。调用这个函数即可触发取消信号。
ctx, cancel := context.WithCancel(context.Background()) defer cancel() // 确保函数退出前调用 cancel,避免泄漏
上面代码创建了一个基于 Background 的 context,并准备了取消能力。通常用 defer cancel() 来确保资源释放。
协程中监听取消信号
启动的子协程可以通过监听 来感知取消操作。一旦 cancel 被调用,Done() 返回的 channel 就会被关闭,select 可以立即响应。
立即学习“go语言免费学习笔记(深入)”;
go func(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("收到取消信号")
return
default:
fmt.Println("正在执行...")
time.Sleep(500 * time.Millisecond)
}
}
}(ctx)
这段代码在一个循环中定期打印日志,直到接收到取消通知。这是典型的 context 使用模式。
触发取消并等待结束
在主逻辑中,你可以在合适时机调用 cancel() 函数,通知所有依赖此 context 的协程退出。
time.Sleep(2 * time.Second) cancel() // 触发取消fmt.Println("已发送取消请求")
调用 cancel 后,所有从该 context 派生的协程都会收到信号。注意,cancel 是一次性操作,多次调用无副作用。
完整示例:
package mainimport ( "context" "fmt" "time" )
func main() { ctx, cancel := context.WithCancel(context.Background()) defer cancel()
go func(ctx context.Context) { for { select { case zuojiankuohaophpcn-ctx.Done(): fmt.Println("工作协程退出") return default: fmt.Println("工作中...") time.Sleep(300 * time.Millisecond) } } }(ctx) time.Sleep(1.5 * time.Second) cancel() fmt.Println("已调用 cancel") time.Sleep(500 * time.Millisecond) // 等待协程退出}
基本上就这些。关键是理解:WithCancel 提供一种优雅的跨协程通知机制,配合 select 和 Done() 使用,能有效避免 goroutine 泄漏。记得总是调用 cancel,最好用 defer 包装。不复杂但容易忽略。










