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 <-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 包装。不复杂但容易忽略。
以上就是Golang如何使用context.WithCancel取消操作的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1418246.html
微信扫一扫
支付宝扫一扫