了解为什么在提交时返回连接繁忙的 PGX 原因是什么？

一分耕耘，一分收获！既然打开了这篇文章《了解为什么在提交时返回连接繁忙的 PGX 原因是什么？》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我有一个函数可以将值批量插入到表中。 tx.commit() 返回 conn busy。正如我从阅读代码中得到的那样，conn.begin() 实际上使它变得忙碌。

所以问题是如何正确地做到这一点？我应该将事务与批量查询一起使用吗？或者交易是在幕后创建的？

// InsertItems adds items to the table
func (r *Repository) InsertItems(ctx context.Context, values []service.Transaction) error {

    conn, err := r.pool.Acquire(ctx)
    if err != nil {
        return fmt.Errorf("acquire connection: %w", err)
    }
    defer conn.Release()

    tx, err := conn.Begin(ctx)
    if err != nil {
        return fmt.Errorf("starting pgx transaction: %w", err)
    }
    defer func() { _ = tx.Rollback(ctx) }()

    batch := pgx.Batch{}

    for _, v := range values {

        query := fmt.Sprintf(`INSERT INTO %v (id, date, amount) VALUES ($1, $2, $3)`, r.tableName)

        batch.Queue(query, v.ID, v.Date, v.Amount)
    }

    batchRes := tx.SendBatch(ctx, &batch)
    defer func() {
        if err := batchRes.Close(); err != nil {
            logger.Errorf("closing batch result: %v", err)
        }
    }()

    cmdTag, err := batchRes.Exec()
    if err != nil {
        return fmt.Errorf("batch res exec: %w", err)
    }

    logger.Debugf("inserted rows: %d", cmdTag.RowsAffected())

    if err := tx.Commit(ctx); err != nil {
        return fmt.Errorf("commiting pgx transaction: %w", err)
    }

    return nil
}

正确答案

您在对批处理结果调用 close 之前调用 commit。您需要先关闭批处理结果然后才能再次使用底层连接。

要强制执行延迟操作的正确顺序，您可以执行以下操作：

// InsertItems adds items to the table
func (r *Repository) InsertItems(ctx context.Context, values []service.Transaction) (err error) {
    conn, err := r.pool.Acquire(ctx)
    if err != nil {
        return fmt.Errorf("acquire connection: %w", err)
    }
    defer conn.Release()

    batch := new(pgx.Batch)
    for _, v := range values {
        query := fmt.Sprintf(`INSERT INTO %v (id, date, amount) VALUES ($1, $2, $3)`, r.tableName)
        _ = batch.Queue(query, v.ID, v.Date, v.Amount)
    }

    tx, err := conn.Begin(ctx)
    if err != nil {
        return fmt.Errorf("starting pgx transaction: %w", err)
    }   
    result := tx.SendBatch(ctx, batch)
    defer func() {
        if e := result.Close(); e != nil {
            logger.Errorf("closing batch result: %v", e)
            err = e
        }
        
        if err != nil {
            _ = tx.Rollback(ctx)
        } else {
            if e := tx.Commit(ctx); e != nil {
                err = e
            }
        }
    }()
    tag, err := result.Exec()
    if err != nil {
        return fmt.Errorf("batch res exec: %w", err)
    }
    
    logger.Debugf("inserted rows: %d", tag.RowsAffected())
    return nil
}

好了，本文到此结束，带大家了解了《了解为什么在提交时返回连接繁忙的 PGX 原因是什么？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！