文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

golang中如何实现db事务的统一封装

2023-06-22 00:27

关注

小编给大家分享一下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事务的统一封装”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注编程网行业资讯频道!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯