使用context.WithTimeout可创建自动取消的上下文,通过select监听ctx.Done()实现goroutine超时控制,避免资源浪费。
在Go语言中,使用
context
控制并发任务的超时是一种标准且推荐的做法。它能帮助你在指定时间内取消或终止正在运行的 goroutine,避免资源浪费和程序阻塞。
创建带超时的 Context
通过
context.WithTimeout
可以创建一个会在指定时间后自动取消的 context:
这段代码创建了一个最多存活3秒的 context。时间一到,context 会自动触发取消信号,无需手动调用 cancel,但显式调用可提前终止。
在 Goroutine 中监听 Context 取消
你可以在并发任务中定期检查 context 是否已被取消:
立即学习
“
go语言免费学习笔记(深入)
”;
go语言参考手册 中文CHM版
Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译
下载
当 context 超时后,
ctx.Done()
通道会被关闭,select 会执行对应分支,从而退出 goroutine。
等待任务完成或超时
主协程通常需要等待任务结束,可以使用 channel 配合 context 实现安全等待:
这样无论任务完成还是 context 超时,都能正确处理结果或错误。
基本上就这些。只要合理使用 context.WithTimeout 和 select 监听 ctx.Done(),就能有效控制并发任务的生命周期,防止无限等待。
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel() // 确保释放资源
go func(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("任务被取消:", ctx.Err())
return
default:
// 执行一些工作
time.Sleep(500 * time.Millisecond)
fmt.Println("工作中...")
}
}
}(ctx)
resultCh := make(chan string, 1)
go func() {
result := doWork() // 模拟耗时操作
resultCh <- result
}()
select {
case res := <-resultCh:
fmt.Println("任务成功:", res)
case <-ctx.Done():
fmt.Println("任务超时:", ctx.Err())
}
