golang db事务的统一封装的实现
更新时间：2021年12月10日 11:57:17 作者：yuchenfw
这篇文章主要介绍了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事务的统一封装的实现的文章就介绍到这了,更多相关golang db事务封装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Go语言中用 os/exec 执行命令的五种方法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
这篇文章主要介绍了golang双指针快速排序的实现代码,通过实例代码补充介绍了Golang实现快速排序和归并排序以及堆排序算法全注释,需要的朋友可以参考下2024-03-03
在某些场景下，我们需要初始化一些资源。有时会采用延迟初始化的方式，在真正需要资源的时候才进行初始化。在这种情况下，Go语言中的sync.Once提供一个优雅且并发安全的解决方案，本文将对其进行详细介绍2023-04-04
这篇文章主要介绍了Goland和IDEA换行符的设置方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-07-07
相信小伙伴都遇到过以下的循环变量的问题，本文主要介绍了go中for range的坑以及解决方案，具有一定的参考价值，感兴趣的可以了解一下2024-01-01
这篇文章主要给大家介绍了关于golang如何利用原始套接字构造UDP包的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用golang具有一定的参考学习价值，需要的朋友们下面来一起看看吧。2017-10-10
这篇文章主要为大家学习介绍了Golang如何仿ps实现获取Linux进程信息，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起了解一下2023-07-07
这篇文章主要为大家介绍了Go语言包和包管理详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-09-09
本文主要介绍了Go语言中程序是怎么编译的实现,深入探讨Go语言的编译机制和最新的模块管理系统Go Modules的使用,具有一定的参考价值,感兴趣的可以了解一下2024-06-06
这篇文章主要介绍了一步步教你打造高效可靠的Go库的相关资料,需要的朋友可以参考下2023-11-11
最新评论