如何结合Squirrel与PostgreSQL事务？代码是否正确？

问题分析 #

1. 事务绑定错误
   你开启了PostgreSQL事务tx，但Squirrel的语句构建器用RunWith(db.repo.GetDatabase())绑定的是原始数据库连接，而非当前事务。这导致插入操作完全脱离了事务控制，后续的tx.Rollback()和tx.Commit()对这次插入没有任何影响——不管事务怎么操作，插入都会直接提交到数据库。

2. 错误处理逻辑缺陷
   当sb.Exec()执行失败时，你直接返回回滚的错误，丢失了原始的执行错误。这会导致排查问题时无法知道真正的失败原因（比如插入违反约束、连接异常等）。

修正后的代码 #

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

// 关键改动：将Squirrel绑定到开启的事务tx上，而非原始DB连接
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,
    )
}

_, err = sb.Exec()
if err != nil {
    // 先回滚，回滚的错误可以日志记录，优先返回原始执行错误
    rollbackErr := tx.Rollback()
    if rollbackErr != nil {
        // 这里可以用日志库记录rollbackErr，比如log.Printf("回滚失败: %v", rollbackErr)
    }
    return nil, err
}

// 执行成功后提交事务
if err := tx.Commit(); err != nil {
    return nil, err
}

核心要点总结 #

• Squirrel与事务结合的关键：必须将语句构建器通过RunWith(tx)绑定到开启的事务对象上，所有通过该构建器执行的操作才会纳入事务管理。
• 回滚/提交时机：
  • 当任何事务内的操作执行失败时，必须调用tx.Rollback()撤销所有已执行的操作；
  • 当所有操作都执行成功后，调用tx.Commit()将所有变更持久化到数据库。
• 错误处理优先级：优先保留原始业务操作的错误，回滚的错误可以作为辅助信息记录，避免掩盖真正的问题。

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