context通过ctx.Done()通道通知协程主动退出,实现协作式取消;WithTimeout设超时,WithCancel支持手动取消,均需调用cancel()防泄漏。

在 Go 中,context 是控制协程生命周期、传递取消信号和超时的关键机制。它不直接“杀死”协程,而是通过通道(ctx.Done())通知协程主动退出,配合 select 实现协作式取消 —— 这是 Go 推荐的并发安全做法。
用 context.WithTimeout 控制超时
当你需要限制一个操作最多执行多久,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() 必须调用(哪怕提前返回),否则底层定时器不会释放,可能造成内存泄漏。
立即学习“go语言免费学习笔记(深入)”;
用 context.WithCancel 主动触发取消
适合用户点击“停止”、服务收到 shutdown 信号等场景。你手动调用 cancel() 向所有监听该 ctx 的协程广播终止信号。
常见用法:
- 创建可取消的上下文:
ctx, cancel := context.WithCancel(parentCtx) - 在 goroutine 中监听:
case - 外部调用
cancel()即刻触发所有监听者退出 - 多个协程可共享同一个
ctx,实现统一控制
在 HTTP 请求与数据库调用中透传 context
真实项目中,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:应复用同一 context,或用
context.WithValue传值(非取消逻辑),而非新建取消上下文 - 忽略子协程的清理:父协程 cancel 后,子协程需自行响应并退出,不能依赖 GC “回收”正在运行的 goroutine










