跳转到主内容
websoft网络软件专家 - 深耕网络技术,打造实用软件!

Golang如何使用context控制并发任务超时

使用context.WithTimeout可创建自动取消的上下文,通过select监听ctx.Done()实现goroutine超时控制,避免资源浪费。 在Go语言中,使用 context 控制并发任务的超时是一种标准且推荐的做法。它能帮助你在指定时间内取消或终止正在运行的 goroutine,避免资源浪费和程序阻塞。 创建带超时的 Context 通过 context.WithTimeout 可以创建一个会在指定时间后自动取消的 context:
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() // 确保释放资源
这段代码创建了一个最多存活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数组安全、编译 下载
go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("任务被取消:", ctx.Err()) return default: // 执行一些工作 time.Sleep(500 * time.Millisecond) fmt.Println("工作中...") } } }(ctx)
当 context 超时后, ctx.Done() 通道会被关闭,select 会执行对应分支,从而退出 goroutine。 等待任务完成或超时 主协程通常需要等待任务结束,可以使用 channel 配合 context 实现安全等待:
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()) }

这样无论任务完成还是 context 超时,都能正确处理结果或错误。 基本上就这些。只要合理使用 context.WithTimeout 和 select 监听 ctx.Done(),就能有效控制并发任务的生命周期,防止无限等待。

相关文章