您现在的位置是：网站首页> 编程资料编程资料

golang db事务的统一封装的实现_Golang_

2023-05-26 477人已围观

简介 golang db事务的统一封装的实现_Golang_

简单封装

偷懒第一步

采用defer处理Rollback

 defer tx.Rollback()

无论成功与否，均进行Rollback操作，只是有点影响，如果成功还调用Rollback的话，将会报错。虽然可以忽略，但作为程序员，有必要进一步调整。

偷懒第二步

根据执行结果来选择执行Rollback，避免无效使用。

 defer func() { //根据执行结果选择执行Rollback if err != nil && tx != nil { log.Println("ExecSqlWithTransaction defer err :", err) tx.Rollback() } }()

如此，我们就可以根据事务的执行结果决定是否Rollback了。

偷懒第三步

封装，以上代码本身就具有极大的普适性，因此，我们抽出通用的参数，将此过程封装成一个func，以后就可以直接调用了。

 func ExecSqlWithTransaction(database *sql.DB, query string, args ...interface{}) (err error) { tx, err := database.Begin() if err != nil { return err } defer func() { if err != nil && tx != nil { tx.Rollback() } }() stmt, err := tx.Prepare(query) if err != nil { return err } defer stmt.Close() _, err = stmt.Exec(args...) if err != nil { return err } return tx.Commit() }

封装后我们可以如下使用：

 if err := ExecSqlWithTransaction(database,sqlQuery,paras...);err != nil{ //错误处理 }

封装后是不是很简洁啊？

进一步封装

在一个事务中可能会出现多个SELECT、UPDATE等操作，以上封装仅处理了一种操作，还不能满足我们的实际需求，因此需要更进一步封装。

 func ExecSqlWithTransaction(db *sql.DB, handle func(tx *sql.Tx) error) (err error) { tx, err := db.Begin() if err != nil { return err } defer func() { if err != nil { tx.Rollback() } }() if err = handle(tx); err != nil { return err } return tx.Commit() }

在handle func内可以直接使用事务tx进行增删改查。

到此这篇关于golang db事务的统一封装的实现的文章就介绍到这了,更多相关golang db事务封装内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！

您可能感兴趣的文章: