如何在应用停止时终止长时间运行的数据库查询？

你的核心需求是在应用停止时优雅中断独立goroutine里的数据库查询，同时避免跨goroutine直接调用context的cancelFunc这种不良实践。下面是符合context设计规范的解决方案：

正确实现方案 #

利用context的层级传递特性，在main函数中创建顶层可取消context，将其传入goroutine后与超时context组合，这样无论是应用主动停止还是查询超时，都能正确触发取消：

package main

import (
    "context"
    "database/sql"
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time"
)

func main() {
    // 创建顶层可取消context，用于统一触发应用级别的取消操作
    mainCtx, mainCancel := context.WithCancel(context.Background())
    defer mainCancel() // 确保main退出时清理context

    // 启动执行长查询的goroutine，传入顶层context
    go func(ctx context.Context) {
        // 组合顶层context与超时context，同时满足超时自动取消和应用停止主动取消
        queryCtx, cancel := context.WithTimeout(ctx, 3*time.Minute)
        defer cancel() // 及时释放context资源

        rows, err := db.QueryContext(queryCtx, "very long query")
        if err != nil {
            fmt.Println("查询执行失败:", err)
            return
        }
        defer rows.Close()

        // 处理查询结果逻辑...
    }(mainCtx)

    // 监听应用停止信号（比如Ctrl+C）
    sigChan := make(chan os.Signal, 1)
    signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
    <-sigChan

    // 触发顶层context取消，所有基于该context的子context都会被取消
    mainCancel()
    fmt.Println("应用正在停止，已取消数据库查询")
}

方案优势 #

• 符合context设计原则：通过父context的取消传递触发子context终止，避免跨goroutine直接调用子context的cancelFunc
• 同时支持两种取消场景：查询超时自动取消、应用停止主动取消
• 代码结构清晰，无需额外的stop通道和嵌套goroutine，减少并发复杂度

原方案的问题 #

你之前的实现中，在独立goroutine里直接调用了子context的cancelFunc，这违反了context的使用规范：cancelFunc应该由创建该context的goroutine调用，或者通过父context的取消间接触发。跨goroutine直接调用cancelFunc可能引发潜在的并发问题，也不利于context资源的统一管理。

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