在不同的 DAO 中使用相同的 DB 事务

本篇文章给大家分享《在不同的 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 事务》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！