使用事务和 squirrel 进行 Golang postgresql 查询

Question

我了解如何单独使用松鼠和交易，但不了解如何一起使用它们。我什么时候应该回滚或提交？ 我的尝试正确与否？如果不是的话，我哪里错了...

tx, err := db.repo.GetDatabase().Begin()
if err != nil {
    return nil, err
}

sb := squirrel.StatementBuilder.
    Insert("dependencies").
    Columns("correlation_id", "name", "age").
    PlaceholderFormat(squirrel.Dollar).
    RunWith(db.repo.GetDatabase())

for _, human:= range humans{
    sb = sb.Values(
        human.CorrelationID,
        human.Name,
        human.Age,
    )
}

_, err = sb.Exec()
if err != nil {
    if err := tx.Rollback(); err != nil {
        return nil, err
    }
}

if err := tx.Commit(); err != nil {
    return nil, err
}

据我了解，我正在尝试在 postgresql 中执行查询后回滚或提交

Answer 1

你的努力是伟大的。但在这种情况下

....RunWith(db.repo.GetDatabase())

是不正确的。因为您应该传递交易连接

tx

。指示 Squirrel 使用事务对象作为查询的数据库连接。

如果您使用数据库连接而不是事务连接，Squirrel 查询将不会成为事务的一部分。每个查询将单独执行并立即提交到数据库。

我们还可以使用

RollBack

语句更新

Commit

和

defer

，这将确保事务在函数退出之前得到正确处理和完成。

这是更新后的代码..

tx, err := db.repo.GetDatabase().Begin()
if err != nil {
    return nil, err
}

// added defer rollback and commit
defer func() {
    if err != nil {
        fmt.Println("An error happened while executing the queries - ", err)
        tx.Rollback()
        return
    }
    err = tx.Commit()
}()

response := make([]storage.URLStorage, 0, len(urls))

sb := squirrel.StatementBuilder.
    Insert("dependencies").
    Columns("correlation_id", "name", "age").
    PlaceholderFormat(squirrel.Dollar).
    RunWith(tx)

for _, human := range humans {
    sb = sb.Values(
        human.CorrelationID,
        human.Name,
        human.Age,
    )
}

// the error will be handled by the defer
_, err = sb.Exec()

// you can execute multiple queries with the transaction
for _, human := range someOtheSlice {
    sb = sb.Values(
        human.CorrelationID,
        human.Name,
        human.Age,
    )
}

_, err = sb.Exec()

// If any error happened this query executions, all will be roll backed with the defer

希望这有帮助。

另请参阅

• 数据库/sql Tx - 检测提交或回滚
• 延迟的使用

Answer 2

@Pratheesh PC 提到的解决方案将进行多少次数据库调用