在Golang的并发编程场景中，Goroutine的超时控制是非常关键的需求，如果Goroutine长时间阻塞或者执行耗时超过预期，很容易导致资源泄漏、程序响应变慢等问题，因此需要掌握可靠的Goroutine超时控制与取消方法。
使用channel结合select实现超时控制
这种方式是比较基础的超时控制思路，通过额外启动一个Goroutine在指定时间后向超时channel发送信号，再用select监听任务channel和超时channel，哪个先返回就执行对应的逻辑。
package main
import (
"fmt"
"time"
)
func task(ch chan string) {
// 模拟耗时任务，这里休眠3秒
time.Sleep(3 * time.Second)
ch <- "任务执行完成"
}
func main() {
taskCh := make(chan string)
timeoutCh := make(chan struct{})
// 启动任务Goroutine
go task(taskCh)
// 启动超时Goroutine，2秒后发送超时信号
go func() {
time.Sleep(2 * time.Second)
close(timeoutCh)
}()
select {
case res := <-taskCh:
fmt.Println(res)
case <-timeoutCh:
fmt.Println("任务执行超时")
}
}
这种方法的优点是逻辑简单直观，不需要依赖额外的包，但是如果需要实现多层级取消、传递取消信号给多个Goroutine，这种方式就会显得比较繁琐，代码可维护性也会下降。
使用context包实现超时控制与取消
context是Golang官方提供的用于跨Goroutine传递上下文信息的标准库，其中WithTimeout和WithCancel方法可以很方便地实现超时控制和主动取消功能，也是目前推荐使用的方案。
context.WithTimeout实现超时控制
通过context.WithTimeout创建一个带有超时时间的上下文，当超时时间到达时，上下文的Done通道会被关闭，所有监听该上下文的Goroutine都可以感知到超时信号。
package main
import (
"context"
"fmt"
"time"
)
func taskWithContext(ctx context.Context) {
select {
case <-time.After(3 * time.Second):
// 模拟耗时3秒的任务
fmt.Println("任务正常执行完成")
case <-ctx.Done():
// 上下文被取消或者超时，打印对应的错误信息
fmt.Println("任务被取消，原因：", ctx.Err())
}
}
func main() {
// 创建超时时间为2秒的上下文
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel() // 即使超时也会执行cancel，避免资源泄漏
go taskWithContext(ctx)
// 等待足够时间让Goroutine执行完毕
time.Sleep(4 * time.Second)
}
context.WithCancel实现主动取消
除了超时自动取消，有时候我们需要在某些条件下主动取消Goroutine的执行，这时候可以使用context.WithCancel创建可手动取消的上下文。
package main
import (
"context"
"fmt"
"time"
)
func taskWithCancel(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("收到取消信号，任务退出")
return
default:
// 模拟任务执行
fmt.Println("任务执行中...")
time.Sleep(1 * time.Second)
}
}
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
go taskWithCancel(ctx)
// 模拟运行3秒后主动取消任务
time.Sleep(3 * time.Second)
cancel()
// 等待Goroutine处理完取消逻辑
time.Sleep(1 * time.Second)
}
两种方法的对比与选择
我们可以通过下面的表格对比两种方案的差异，方便根据实际场景选择：
| 方案 | 优点 | 缺点 | 适用场景 |
|---|---|---|---|
| channel+select | 逻辑简单，无额外依赖 | 多Goroutine取消、信号传递繁琐，可维护性差 | 简单的单Goroutine超时控制场景 |
| context包 | 支持超时、主动取消、多Goroutine信号传递，官方推荐，可维护性好 | 需要理解context的使用规则，有一定的学习成本 | 复杂的并发场景，多Goroutine联动取消、超时控制 |
在实际开发中，除非是非常简单的临时场景，否则更推荐使用context包实现Goroutine的超时控制与取消，它不仅能满足超时需求，还能很好地处理多Goroutine之间的取消信号传递，避免Goroutine泄漏，让并发代码更健壮。