如何在很短的时间间隔内/同时进行多个查询

Question

嘿，我收到一条错误消息：来自 pgx 的 conn busy

我不知道如何解决这个问题。这是我的功能：

func (r *proverbRepo) SelectPendingProverbs(table string) (proverbs []domain.Proverb, err error) {
    query := fmt.Sprintf("SELECT id, proverb literal FROM %s", table)
    rows, err := r.Db.Query(context.Background(), query)
    defer rows.Close()

    if err != nil {
        return
    }

    for rows.Next() {
        var prov domain.Proverb
        if err = rows.Scan(&prov.ID, &prov.Literal); err != nil {
            return
        }
        proverbs = append(proverbs, prov)
    }
    return
}

r.Db 是pgx.Connect(context.Background(), os.Getenv("PSQL_URL"))

我在很短的时间间隔内从两个单独的前端请求中获取两个不同的表。

第一个请求通过，另一个返回conn busy错误信息。

我真的不知道要找什么，有人可以帮我吗？

Answer 1

pgx.Connect() 返回一个不能同时使用的pgx.Conn。这就是这种类型的godocs 状态：

Conn 是一个 PostgreSQL 连接句柄。并发使用是不安全的。使用连接池来管理来自多个 goroutine 的多个数据库连接的访问​​。

因此，如果您将 pgx.Connect() 替换为来自 github.com/jackc/pgx/pgxpool 的 pgxpool.Connect() 应该没问题。

Answer 2

pgx.Connect(context.Background(), os.Getenv("PSQL_URL")) 返回的r.Db 如果属于*pgx.Conn 类型，则表示不是并发安全的单个连接。通常，您希望使用连接池为您处理并发并允许重用打开的连接。 要使用连接池，请将 import github.com/jackc/pgx/v4 替换为 github.com/jackc/pgx/v4/pgxpool 并使用 pgxpool.Connect() 而不是 pgx.Connect() 进行连接，并且 api 将是相同的：

r.Pool := pgxpool.Connect(context.Background(), os.Getenv("PSQL_URL"))
r.Pool.Query(context.Background(), query)
...

如果在任何给定点您需要使用单个连接来访问某些较低级别的功能，您可以安全地获取并发安全连接，如下所示：

conn, err := r.Pool.Acquire(context.Background())
if err != nil {
    fmt.Fprintln(os.Stderr, "Error acquiring connection:", err)
    os.Exit(1)
}
defer conn.Release()
...