pgx事务超时未生效问题咨询：与pq差异及解决方法

之前使用pq库，现在想要切换为pgx库，需求是当数据库中运行的事务耗时过久时将其终止。pq能够在context过期时立即终止事务，但pgx会等待查询完成，仅在后续查询或提交时返回错误。请问这是正常现象，还是操作有误？这种情况下该如何终止事务？

代码示例 #

import (
    "context"
    "database/sql"
    "fmt"
    _ "github.com/jackc/pgx/v5/stdlib"
    _ "github.com/lib/pq"
    "log"
    "time"
)

func main() {
    fmt.Println("================= PQ ==============")
    pqWork()
    fmt.Println("================= ==============")

    fmt.Println("================= PGX ==============")
    pqxWork()
    fmt.Println("================= ==============")
}

func pqWork() {
    connStr := "host=localhost port=5432 user=... password=... dbname=... "
    db, err := sql.Open("postgres", connStr)
    if err != nil {
        panic(err)
    }
    defer db.Close()

    exec(db)
}

func pqxWork() {
    connStr := "host=localhost port=5432 user=... password=... dbname=... "
    db, err := sql.Open("pgx", connStr)
    if err != nil {
        panic(err)
    }
    defer db.Close()

    exec(db)
}

func exec(db *sql.DB) {
    ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
    defer cancel()
    begin := time.Now()
    tx, err := db.BeginTx(ctx, nil)
    if err != nil {
        panic(err)
    }

    _, err = tx.Exec("select pg_sleep(10)")
    if err != nil {
        fmt.Println("err. time:", time.Since(begin).Milliseconds(), "ms")
        tx.Rollback()
        return
    }
    fmt.Println("success. time:", time.Since(begin).Milliseconds(), "ms")
    err = tx.Commit()
    if err != nil {
        panic(err)
    }
}

运行结果 #

================= PQ ==============
err. time:  524 ms
================= ==============
================= PGX ==============
success. time:  10032 ms
panic: sql: transaction has already been committed or rolled back

这是pgx stdlib包装的正常行为，并非操作错误。两者的核心差异在于：

• pq库在context取消时，会主动向PostgreSQL服务器发送CANCEL请求，立即终止正在执行的查询/事务。
• pgx的stdlib包装（即你使用的github.com/jackc/pgx/v5/stdlib）默认不会在context取消时主动发送终止信号，而是等待当前查询执行完成后，再检查context状态并返回错误，这就导致了你看到的延迟现象。

要实现context过期时立即终止事务，有两种可行方案：

方案1：使用pgx原生接口（推荐） #

pgx原生API支持在context取消时自动发送CANCEL请求到数据库，无需依赖database/sql层。示例代码如下：

import (
    "context"
    "fmt"
    "github.com/jackc/pgx/v5"
    "time"
)

func pgxNativeWork() {
    connStr := "host=localhost port=5432 user=... password=... dbname=..."
    conn, err := pgx.Connect(context.Background(), connStr)
    if err != nil {
        panic(err)
    }
    defer conn.Close(context.Background())

    ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
    defer cancel()
    begin := time.Now()

    tx, err := conn.Begin(ctx)
    if err != nil {
        panic(err)
    }
    defer tx.Rollback(ctx)

    _, err = tx.Exec(ctx, "select pg_sleep(10)")
    if err != nil {
        fmt.Println("err. time:", time.Since(begin).Milliseconds(), "ms")
        return
    }

    fmt.Println("success. time:", time.Since(begin).Milliseconds(), "ms")
    err = tx.Commit(ctx)
    if err != nil {
        panic(err)
    }
}

这段代码会在context超时后立即终止pg_sleep(10)查询，表现和pq一致。

方案2：配置pgx stdlib启用自动取消 #

如果必须继续使用database/sql接口，可以通过pgx的连接配置启用CancelOnTimeout选项，让它在context取消时主动发送终止信号。修改你的pqxWork函数如下：

import "github.com/jackc/pgx/v5/pgxstdlib"

func pqxWork() {
    // 解析连接字符串并添加配置
    config, err := pgx.ParseConfig("host=localhost port=5432 user=... password=... dbname=...")
    if err != nil {
        panic(err)
    }
    config.CancelOnTimeout = true // 关键配置：超时后自动取消查询

    // 用pgxstdlib的OpenDB方法创建sql.DB
    db := pgxstdlib.OpenDB(*config)
    defer db.Close()

    exec(db)
}

注意需要导入github.com/jackc/pgx/v5/pgxstdlib替代原有的stdlib导入，确保配置生效。

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