如何在连接关闭时终止正在运行的查询

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《如何在连接关闭时终止正在运行的查询》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

我们在后端使用 https://github.com/go-gorm/gorm/ ORM 和脚本来连接到我们的 PostgreSQL 数据库。

有时，当脚本正在进行时，我们会通过在本地按 Ctrl + C 或在生产中终止 pod/进程来手动终止作业。我们在所有脚本中确实有一个 defer DB.Close()，我还添加了处理 SIGINT / SIGTERM 信号以在终止时执行 DB.Close() 。

问题是，即使关闭连接后，任何已经运行的现有查询也不会被终止并继续占用数据库资源。有没有办法在直接从 gorm 或通过其他一些 hack 退出之前终止此连接池启动的任何查询。

考虑使用 pg_backend_pid() 并使用 pg_stat_activity 终止查询，但是我们在运行新的 pg_backend_pid() 时获得的 pid 将与正在运行的不一样。

版本：jinzhu/gorm v1.9.2

解决方案

您可以使用 use begintx(ctx context.context, opts *sql.txoptions) ，它将上下文作为参数。

这是一个小例子：

import (
    "context"
    "fmt"
    "os"
    "os/signal"

    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/postgres"
)

func main() {
    db, err := gorm.Open("postgres", "host=localhost port=5432 user=gorm dbname=gorm password=mypassword sslmode=disable")
    if err != nil {
        panic(err)
    }
    defer db.Close()
    ctx := context.Background()

    ctx, cancel := context.WithCancel(ctx)
    c := make(chan os.Signal, 1)
    signal.Notify(c, os.Interrupt)
    defer func() {
        signal.Stop(c)
        cancel()
    }()
    go func() {
        select {
        case <-c:
            cancel()
        case <-ctx.Done():
        }
    }()

    transaction, err := db.DB().BeginTx(ctx, nil)
    _, err = transaction.Exec("SELECT pg_sleep(100)")
    if err != nil {
        fmt.Println(err.Error())
    }
}

本篇关于《如何在连接关闭时终止正在运行的查询》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！