使用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 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(),就能有效控制并发任务的生命周期,防止无限等待。










