Go database/sql.Conn.Raw 里修改驱动连接安全吗：回调期间并发与连接生命周期

项目需要调用数据库驱动的专有能力时，最容易写出一段“看起来能跑”的代码：在 Conn.Raw 里拿到底层连接，随后把它交给异步 goroutine，等回调结束后再慢慢使用。真正的边界恰恰相反，底层连接只在回调期间借用，不能带出回调。

Conn.Raw 适合一次性的驱动扩展操作，不是把池里的连接永久取出来。把驱动连接限制在回调内，并让错误返回值决定连接是否还能继续复用。

先确认 Conn.Raw 借用的到底是什么

sql.Conn 代表池中的一条固定数据库连接，而不是整个池。调用 Raw 时，标准库把底层驱动连接传给回调；官方文档明确要求这个 driverConn 不能在回调外使用。也就是说，下面的异步保存方式不是延长生命周期，而是制造悬空借用。

var saved any
err := conn.Raw(func(driverConn any) error {
    saved = driverConn // 回调返回后不要再读取 saved
    return nil
})
_ = err
// 错误示例：这里继续把 saved 当成驱动连接使用

安全写法是把专有操作完整放在回调里，回调只返回结果或错误；不要把接口值、驱动连接指针或依赖它的异步任务带出去。调用链可以概括为 Conn.Raw 暴露 driverConn，回调完成后进入 回调返回，外层再负责 Conn.Close。

回调里怎样区分可恢复错误与坏连接

不要只根据“回调返回了 error”就关闭 Conn。database/sql 对 driver.ErrBadConn 有专门语义：它表示驱动连接已经不适合继续使用，标准库可以据此放弃或重新建立底层连接。业务错误、参数错误或驱动返回的普通错误，则不自动等价于连接损坏。

err := conn.Raw(func(driverConn any) error {
    dc, ok := driverConn.(interface{ Ping(context.Context) error })
    if !ok {
        return errors.New("driver does not expose Ping")
    }
    return dc.Ping(ctx)
})
if errors.Is(err, driver.ErrBadConn) {
    // 让上层按坏连接路径处理，不把它当成普通业务失败
    log.Printf("驱动连接不可复用: %v", err)
}

错误处理的重点是保留错误链，而不是手动猜测底层连接状态。若决定结束这条 sql.Conn，调用 Conn.Close 让连接回到池的管理范围；关闭后再次操作会得到 ErrConnDone，这就是可观察的终态。

并发使用时，锁住的是 Conn 还是 driverConn

sql.Conn 的方法由标准库管理，Conn.Close 可以与运行中的查询并发调用，并会等待其他操作结束。但这条保证不能扩展到底层驱动对象：Raw 回调内部的 driverConn 仍要遵守驱动接口自己的并发约定。最稳妥的方式是让回调内操作短、同步、可完成，并把结果复制成普通 Go 值。

如果驱动扩展操作必须跨多个步骤保持会话状态，优先把步骤放在同一个 Raw 回调里；如果必须跨 goroutine 协作，就传递数据和结果通道，不传递 driverConn 本身。这样关闭连接时，生命周期仍由 database/sql 统一收口。

常见坑：Raw 不是连接池逃生舱

• 把 driverConn 存入结构体：回调结束后继续使用，违反 Raw 的生命周期契约。
• 在 Raw 回调里启动 goroutine：回调返回时 goroutine 可能还在访问底层对象。
• 把所有错误都改成 driver.ErrBadConn：会误导连接池丢弃本来可复用的连接。
• 忘记 Conn.Close：固定连接长期占用池资源，后续获取连接可能阻塞。

相关问题

Raw 回调返回后，Conn 还能继续查询吗？

通常可以，只要回调返回的错误不是 driver.ErrBadConn，且 Conn 没有被关闭。

Conn.Close 会立刻打断正在执行的查询吗？

它会等待其他操作结束；查询自身仍应使用可取消的上下文控制等待时间。

为什么关闭后会出现 ErrConnDone？

这是 Conn.Close 后的明确终态，表示这条固定连接已经归还池，不能再继续操作。

最后做一次边界核对

写 Raw 扩展前只问三件事：底层对象是否只在回调内使用，错误是否保留了 driver.ErrBadConn 语义，外层是否最终调用了 Conn.Close。三项都能回答清楚，连接池才不会替一段异步借用代码背锅。