本篇文章给大家分享《在不同的 DAO 中使用相同的 DB 事务》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。
问题内容我的一个 golang 应用程序中有一个用例，用于更新 api 中的多个表，如果其中一个更新失败，我希望回滚所有以前的更新（java 中的 @transactional 就是这样做的）。 我尝试按以下方式进行操作：
func (d *dao) method1(opt string) error {
tx, err := d.db.begintx(context.background(), nil)
if err != nil {
return errors.errorf("unable to start transaction")
}
err := d.dao1.update1(opt)
if err != nil {
log.infof("error : %s. rolling back transaction", err)
_ = tx.rollback()
log.infof("transaction rolled back while update 1")
return err
}
err = d.dao2.update2(opt)
if err != nil {
log.errorf("error in making update 2 "+
"error %v ", err)
_ = tx.rollback()
log.infof("transaction rolled back while update 2")
return err
}
}
}
func(d *dao1) update1 error {
var query = "update t1 set c1 = ? where id = ?"
_, err := d.db.exec(query, "v1", "v2")
return err
}
func(d *dao2) update2 error {
var query = "update t2 set c2 = ? where id = ?"
_, err := d.db.exec(query, "v1", "v2")
return err
}但这不起作用，因为我们正在 update1 和 update2 方法中创建新事务。 所以我尝试在方法 update1 和 update2 的方法参数中传递事务，如下所示：
func Update1(tx sql.Tx) error {
var query = "update T1 set C1 = ? where id = ?"
_, err := tx.Exec(query, "v1", "v2")
return err
}但问题是在某些用例中我只想调用方法 update1 。那么，在 update1 方法中传递的交易值应该是多少？
或者有更好的方法吗？有人可以帮忙吗？
正确答案
如果我理解正确，在某些情况下，您希望调用 update1 但不调用 update2，而在其他情况下，您希望两个更新都发生。如果是这样的话，那么我建议这更多是一个设计问题，而不是 tx 的使用问题。我认为设计两个函数，每个函数都描述您期望的行为，并将调用哪个函数的逻辑委托给调用者，可能会起作用。类似于：
func Caller() error {
if condition {
return UpdateOne()
}
return UpdateBoth()
}
func UpdateOne() error { … }
func UpdateBoth() error { … }
到这里，我们也就讲完了《在不同的 DAO 中使用相同的 DB 事务》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注编程网公众号，带你了解更多关于的知识点！
