使用context.WithCancel创建可取消的context,调用cancel函数通知子协程退出;2. 通过ctx.Done()监听取消信号,子协程定期检查ctx状态以实现协程取消和超时控制。

在Go语言中,context 包是处理请求生命周期、超时控制和协程取消的核心工具。它常用于Web服务、数据库调用、API请求等需要控制执行时间或主动中断任务的场景。下面直接讲清楚怎么用 context 实现超时控制和协程取消。
1. Context 的基本结构与传递
Context 是一个接口,包含截止时间、键值对和取消信号。它不能被修改,只能通过派生生成新的 context。推荐始终将 context 作为函数的第一个参数,命名为 ctx。
示例:func doWork(ctx context.Context) {
// 使用 ctx 控制逻辑
}
最常用的根 context 是 context.Background(),通常用于主函数或请求入口。
2. 实现协程取消(Cancel)
使用 context.WithCancel 可以创建一个可取消的 context。当调用 cancel 函数时,所有派生的 context 都会被通知取消。
立即学习“go语言免费学习笔记(深入)”;
代码示例:ctx, cancel := context.WithCancel(context.Background())go func() { defer cancel() // 任务完成时调用 cancel time.Sleep(2 * time.Second) fmt.Println("任务完成") }()
select { case <-ctx.Done(): fmt.Println("收到取消信号:", ctx.Err()) }
这里 ctx.Done() 返回一个 channel,当 cancel 被调用或超时,channel 会被关闭,可用于通知子协程退出。
子协程中应定期检查 ctx 是否被取消:
for {
select {
case <-ctx.Done():
fmt.Println("协程退出")
return
default:
// 执行任务
}
}
3. 超时控制(Timeout)
更常见的需求是设置超时时间,使用 context.WithTimeout 或 context.WithDeadline。
WithTimeout 示例:ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel()go func() { time.Sleep(4 * time.Second) cancel() // 超时前手动完成也可调用 }()
<-ctx.Done() fmt.Println("超时或取消:", ctx.Err())
如果 3 秒内未完成,ctx.Done() 会触发,ctx.Err() 返回 context.DeadlineExceeded。
这种机制非常适合控制 HTTP 请求或数据库查询的最长执行时间。
4. 实际应用:HTTP 请求超时控制
结合 net/http,可以将 context 传给请求,实现精确控制。
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second) defer cancel()req, _ := http.NewRequest("GET", "https://www.php.cn/link/c19fa3728a347ac2a373dbb5c44ba1c2", nil) req = req.WithContext(ctx) // Go 1.13 后推荐使用 req = req.Clone(ctx)
client := &http.Client{} resp, err := client.Do(req) if err != nil { fmt.Println("请求失败:", err) return } defer resp.Body.Close() fmt.Println("状态码:", resp.StatusCode)
这个请求最多等待 2 秒,即使服务器要 3 秒才响应,也会被中断。
基本上就这些。context 的关键是“传递”和“监听 Done”。只要在协程中持续监听 ctx.Done(),就能实现优雅取消。超时不复杂,但容易忽略 defer cancel(),记得释放资源。










