context通过ctx.Done()通道通知协程主动退出,实现协作式取消;WithTimeout设超时,WithCancel支持手动取消,均需调用cancel()防泄漏。
在 Go 中,context 是控制协程生命周期、传递取消信号和超时的关键机制。它不直接“杀死”协程,而是通过通道(ctx.Done())通知协程主动退出,配合 select 实现协作式取消 —— 这是 Go 推荐的并发安全做法。
当你需要限制一个操作最多执行多久,context.WithTimeout 是最常用的方式。它返回一个带截止时间的 Context 和一个用于释放资源的 cancel 函数。
示例:启动一个可能耗时的协程,1.5 秒后自动取消:
ctx, cancel := context.WithTimeout(context.Background(), 1500*time.Millisecond) defer cancel() // 防止资源泄漏
go func(ctx context.Context) { select { case <-time.After(2 * time.Second): // 模拟慢操作 fmt.Println("操作完成") case <-ctx.Done(): // 超时或被取消 fmt.Println("操作被取消:", ctx.Err()) // context deadline exceeded } }(ctx)
注意:cancel() 必须调用(哪怕提前返回),否则底层定时器不会释放,可能造成内存泄漏。
适合用户点击“停止”、服务收到 shutdown 信号等场景。你手动调用 cancel() 向所有监听该 ctx 的协程广播终止信号。
常见用法:
ctx, cancel := context.WithCancel(parentCtx)
case
cancel() 即刻触发所有监听者退出ctx,实现统一控制真实项目中,context 要贯穿整个调用链。标准库已深度集成:
http.Request.Context() 提供请求生命周期绑定的上下文,可直接用于下游调用database/sql)支持 QueryContext、ExecContext 等方法,超时/取消会自动中断查询ctx context.Context 作为第一个参数,并在阻塞操作前检查 ctx.Err() 或使用 select
例如调用带超时的 HTTP 客户端:
req, _ := http.NewRequestWithContext(ctx, "GET", "https://api.example.com", nil)
resp, err := http.DefaultClient.Do(req)
if err != nil {
if errors.Is(err, context.DeadlineExceeded) {
log.Println("请求超时")
}
return
}
初学者容易踩坑的地方:
cancel():尤其在 error early return 时,建议用 defer cancel() 并确保它在正确作用域context.Background() 或 context.TODO():前者用于顶层无父上下文场景(如 main),后者仅作占位,不应出现在生产代码context.WithValue 传值(非取消逻辑),而非新建取消上下文