如何使用 context.WithTimeout 控制数据库并发查询超时？

在 Go 语言开发中，使用 context.WithTimeout 生成带超时的上下文对象，并将其传入数据库驱动的 QueryContext 或 ExecContext 接口，能强制单个查询在设定时间内终止。该方案适用于防止慢查询耗尽连接池，但需注意它仅控制单次请求时长，不直接限制并发请求数量。

先说结论：使用 context.WithTimeout 是控制数据库查询超时的标准做法，能有效避免单条慢查询阻塞整个服务。

适合：需要防止个别慢查询占用数据库连接池资源的场景。
先看：确认使用的数据库驱动支持 Context 参数（如 database/sql 标准库）。
建议：务必在 defer 中调用 cancel 函数，避免内存泄漏。

命令速用版

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
rows, err := db.QueryContext(ctx, "SELECT * FROM users")

为什么会这样

context.WithTimeout 创建一个会在指定时间后自动取消的上下文。当数据库驱动支持 Context 时，底层会在上下文取消时中断查询操作。这能防止网络抖动或复杂 SQL 导致请求无限期挂起。

分步处理

1. 导入 context 和 time 包。

2. 在查询函数内部创建 context，设定合理超时时长。

3. 立即使用 defer 注册 cancel 函数。

4. 将 ctx 传递给 QueryContext、ExecContext 或 PrepareContext。

5. 捕获错误并判断是否为 context.DeadlineExceeded。

怎么验证是否生效

构造一个执行时间超过设定阈值的慢查询（如使用 SLEEP 函数）。观察程序是否在设定时间后返回错误，且错误信息包含 context deadline exceeded。检查数据库连接池状态，确认连接未被永久占用。

常见坑

1. 忘记调用 cancel()：会导致 goroutine 和内存泄漏。

2. 误以为能限制并发数：它只控制时间，不控制同时发出的请求数量，需配合信号量使用。

3. 超时时间设置过短：可能导致正常复杂查询被误杀，需根据业务 SLA 设定。

常见问题

context.WithTimeout 能限制数据库并发连接数吗？

不能。它只控制单个查询的超时时间，限制并发连接数需使用 db.SetMaxOpenConns 或信号量。

超时后数据库端的查询会自动停止吗？

取决于数据库驱动实现。大多数驱动会尝试取消发送请求，但数据库服务端可能仍在执行，需结合数据库端配置。

参考来源

1. Go 官方文档 - context 包：https://pkg.go.dev/context

2. Go 官方文档 - database/sql 包：https://pkg.go.dev/database/sql