小编给大家分享一下golang中如何实现db事务的统一封装,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!
事务处理的流程示例
database := db.DB
tx, err := database.Begin()
if err != nil {
return err
}
stmt, err := tx.Prepare(sqlQuery)
if err != nil {
tx.Rollback()
return err
}
_, err = stmt.Exec(paras...)
if err != nil {
tx.Rollback()
return err
}
err = tx.Commit()
if err != nil {
tx.Rollback()
return err
}
以上是我们使用事务时的一般操作,如果每做一次事务的操作均要进行重新写一遍代码岂不是很麻烦,尤其是出错时,Rollback需要多次在不同错误的地方的进行调用处理。
采用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事务的统一封装”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。