Golang 小技巧：使用 context.WithTimeout 控制任务超时

在 Go 中调用数据库、HTTP 接口或者执行耗时任务时，最好给操作设置一个超时时间，避免任务一直阻塞。

可以使用 context.WithTimeout：

ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()

然后把 ctx 传给后续任务：

select {
case <-time.After(3 * time.Second):
	fmt.Println("任务完成")
case <-ctx.Done():
	fmt.Println("任务超时:", ctx.Err())
}

因为超时时间设置为 2 秒，而任务需要 3 秒，所以最终会进入：

任务超时: context deadline exceeded

为什么要 defer cancel()

即使任务提前完成，也建议调用 cancel()，这样可以及时释放相关资源。

常见写法就是：

ctx, cancel := context.WithTimeout(parentCtx, 2*time.Second)
defer cancel()

小结

context.WithTimeout 很适合用在：

• HTTP 请求
• 数据库查询
• RPC 调用
• 耗时任务控制

给外部调用加上超时限制，可以避免某个任务长时间卡住整个业务流程。