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事务封装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Golang Recover处理错误原理解析
Golang 中的 recover 是一个鲜为人知但非常有趣和强大的功能,让我们看看它是如何工作的,以及在 Outreach.io 中如何利用它来处理 Kubernetes 中的错误
2023-12-12
Golang中的panic之避免和处理程序中的异常情况
Golang中的panic是一种异常处理机制，可以在程序出现异常情况时终止程序并打印错误信息。为了避免panic对程序的影响，开发者可以采用一系列技巧，如defer+recover、编写可重入的代码、使用错误返回值等。这些技巧可以帮助开发者优雅地处理程序中的异常情况
2023-04-04
golang敏感词过滤的实现
本文主要介绍了golang敏感词过滤的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-01-01
Golang中channel使用的一些小技巧
这篇文章主要介绍了Golang中channel使用的一些小技巧,本文讲解了关闭2次、读取的时候channel提前关闭了、向已经关闭的channel写数据等技巧及这实例代码,需要的朋友可以参考下
2015-07-07
浅析Go语言容器之数组和切片的使用
在 Java 的核心库中，集合框架可谓鼎鼎大名：Array 、List、Set等等，随便拎一个出来都值得开发者好好学习如何使用甚至是背后的设计源码。虽然Go语言没有如此丰富的容器类型，但也有一些基本的容器供开发者使用，接下来让我们认识一下这些容器类型吧
2022-11-11
Golang搭建开发环境的图文教程
这篇文章主要介绍了Golang搭建开发环境，本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2020-11-11
利用Go语言实现二叉搜索树
二叉树是一种常见并且非常重要的数据结构，在很多项目中都能看到二叉树的身影，当然它也有很多变种，本文要介绍的是二叉搜索树的实现，希望对大家有所帮助
2023-07-07
GO语言开发终端命令行小工具改进更新
这篇文章主要为大家介绍了GO语言开发终端命令行小工具的改进更新,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01
Go语言学习之接口使用的示例详解
Go语言并没有类的定义，接口可以说Go语言最接近于类的实现方式，但是更轻量。本文将通过一些简单的示例和大家介绍下Go语言中接口的使用，感兴趣的可以学习一下
2022-11-11
Go 编程复杂数据类型 Map
这篇文章主要介绍了Go编程复杂数据类型Map，Go中的Map是一组无需的K-V类型的数据，与Python中的字典Dict和Java中的HashMap结构类似。未被初始化的Map为nil
2022-08-08