Golang postgresql查询与交易和松鼠

piok6c0g  于 2023-06-22  发布在  PostgreSQL
关注(0)|答案(1)|浏览(146)

我知道如何分别使用松鼠和事务,但我不知道如何一起使用它们。我应该什么时候回滚或提交?我的尝试是正确的还是错误的?如果不是,我错在哪里……

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中执行查询后回滚或提交

vm0i2vca

vm0i2vca1#

你的努力是伟大的。但是....RunWith(db.repo.GetDatabase())在这种情况下是不正确的。因为您应该传递事务连接tx。指示Squirrel使用事务对象作为查询的数据库连接。
如果使用DB连接而不是事务连接,则Squirrel查询将不是事务的一部分。每个查询将单独执行并立即提交到数据库。
我们还可以用defer语句更新RollBackCommit,这将确保交易在函数退出之前得到正确处理和完成。
这里是更新的代码。

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

希望这能帮上忙。
也见

相关问题