如何确保上下文取消时终止MySQL查询？Go实现疑问

问题背景 #

我发现Go的MySQL驱动不会在上下文（Context）取消时终止正在运行的查询，这对我的应用造成了麻烦——因为应用会生成长时间运行的查询，进而导致数据库锁定（这是另一个问题，暂时不关注）。我希望确保当请求被中止时，该请求对应的所有运行中查询都能被终止。

我目前用的函数是：

/*
GetConn() returns a connection to the database. It also launches a goroutine
that kills any running queries when the connection is closed.
*/
func GetConn(ctx context.Context) (*sqlx.Conn, error) {
    conn, err := db.Connx(ctx)
    if err != nil {
        return nil, err
    }
    var connId int64
    if err := conn.GetContext(ctx, &connId, `SELECT CONNECTION_ID()`); err != nil {
        return nil, err
    }
    go func() {
        // Wait for the context to be canceled, and kill the connection
        select {
        case <-ctx.Done():
            if err := conn.Close(); err != sql.ErrConnDone {
                db.Exec("KILL ?", connId)
            }
        }
    }()

    return conn, nil
}

这个函数能正常工作，但我担心如果调用者从不取消上下文，对应的goroutine会一直等待下去。我刚接触上下文和通道，是不是遗漏了什么？或者有没有更好的方法确保请求中止时查询被终止？

解决方案与优化建议 #

1. 解决goroutine泄漏问题 #

你担心的goroutine泄漏确实存在——如果调用者始终不取消上下文，这个goroutine会一直阻塞在<-ctx.Done()上，永远不会退出。要解决这个问题，你需要同时监听连接关闭的信号：sqlx.Conn实现了context.Context接口，调用conn.Done()会返回一个通道，当连接被关闭时，这个通道会被关闭，goroutine就能及时退出。

修改后的goroutine逻辑如下：

go func(conn *sqlx.Conn, connId int64) {
    select {
    case <-ctx.Done():
        // 上下文取消，尝试终止连接/查询
        if err := conn.Close(); err != sql.ErrConnDone {
            // 用独立上下文执行KILL，避免原上下文已取消导致命令失败
            _, _ = db.ExecContext(context.Background(), "KILL ?", connId)
        }
    case <-conn.Done():
        // 连接已被调用者关闭，直接退出
        return
    }
}(conn, connId)

2. 更精准的查询终止方式 #

当前逻辑先关闭连接再执行KILL，其实可以根据需求选择更精准的命令：

• KILL QUERY ?：仅终止该连接上当前运行的查询，连接本身可复用（如果需要）
• KILL CONNECTION ?：直接终止整个连接，该连接上所有查询都会被终止

另外，执行KILL时务必用ExecContext并传入独立上下文（比如context.Background()），避免因原上下文已取消导致终止命令无法执行。

3. 驱动原生支持的可能性 #

部分MySQL驱动（如github.com/go-sql-driver/mysql）在v1.6.0及以上版本中，已经支持通过上下文取消来终止查询，但需要确保：

• 驱动版本足够新
• 连接参数开启interrupt=true（默认已开启）

不过对于长时间运行的查询，直接执行KILL或终止连接仍然更可靠——驱动的上下文取消依赖MySQL的COM_QUERY中断机制，某些场景下可能不生效。

4. 连接生命周期管理最佳实践 #

必须要求调用者在使用完连接后调用conn.Close()，这不仅能让连接回到连接池，还能触发goroutine的退出。可以在函数注释中明确说明，或者建议调用者使用defer conn.Close()来确保连接被关闭。

内容的提问来源于stack exchange，提问作者Tyler S