Golang transaction 事务使用的正确姿势

开发 后端
本文中作者展示了 golang 事务的三种写法。

本文中作者展示了 golang 事务的三种写法

第一种写法

这种写法非常朴实,程序流程也非常明确,但是事务处理与程序流程嵌入太深,容易遗漏,造成严重的问题

func DoSomething() (err error) {
    tx, err := db.Begin()
    if err != nil {
        return
    }


    defer func() {
        if p := recover(); p != nil {
            tx.Rollback()
            panic(p)  // re-throw panic after Rollback
        }
    }()


    if _, err = tx.Exec(...); err != nil {
        tx.Rollback()
        return
    }
    if _, err = tx.Exec(...); err != nil {
        tx.Rollback()
        return
    }
    // ...


    err = tx.Commit()
    return
}

第二种写法

下面这种写法把事务处理从程序流程抽离了出来,不容易遗漏,但是作用域是整个函数,程序流程不是很清晰

func DoSomething() (err error) {
    txerr := db.Begin()
    if err != nil {
        return
    }


    defer func() {
        if p := recover(); p != nil {
            tx.Rollback()
            panic(p// re-throw panic after Rollback
        } else if err != nil {
            tx.Rollback()
        } else {
            err = tx.Commit()
        }
    }()


    if _err = tx.Exec(...); err != nil {
        return
    }
    if _err = tx.Exec(...); err != nil {
        return
    }
    // ...
    return
}

第三种写法

写法三是对写法二的进一步封装,写法高级一点,缺点同上

func Transact(db *sql.DBtxFunc func(*sql.Txerror) (err error) {
    txerr := db.Begin()
    if err != nil {
        return
    }


    defer func() {
        if p := recover(); p != nil {
            tx.Rollback()
            panic(p// re-throw panic after Rollback
        } else if err != nil {
            tx.Rollback()
        } else {
            err = tx.Commit()
        }
    }()


    err = txFunc(tx)
    return err
}


func DoSomething() error {
    return Transact(dbfunc (tx *sql.Txerror {
        if _err := tx.Exec(...); err != nil {
            return err
        }
        if _err := tx.Exec(...); err != nil {
            return err
        }
    })
}

我的写法

经过总结和实验,我采用了下面这种写法,defer tx.Rollback() 使得事务回滚始终得到执行。当 tx.Commit() 执行后,tx.Rollback() 起到关闭事务的作用, 当程序因为某个错误中止,tx.Rollback() 起到回滚事务,同事关闭事务的作用。

普通场景

func DoSomething() (err error) {
  tx_ := db.Begin()
  defer tx.Rollback()

  if _err = tx.Exec(...); err != nil {
      return
  }
  if _err = tx.Exec(...); err != nil {
      return
  }
  // ...


  err = tx.Commit()
  return
}

循环场景

(1) 小事务 每次循环提交一次 在循环内部使用这种写法的时候,defer 不能使用,所以要把事务部分抽离到独立的函数当中

func DoSomething() (err error) {
    tx_ := db.Begin()
    defer tx.Rollback()

    if _err = tx.Exec(...); err != nil {
        return
    }
    if _err = tx.Exec(...); err != nil {
        return
    }
    // ...


    err = tx.Commit()
    return
}


for {
    if err := DoSomething(); err != nil{
         // ...
    }
}

(2) 大事务 批量提交 大事务的场景和普通场景是一样的,没有任何区别

func DoSomething() (err error) {
    tx_ := db.Begin()
    defer tx.Rollback()

    for{
        if _err = tx.Exec(...); err != nil {
            return
        }
        if _err = tx.Exec(...); err != nil {
            return
        }
        // ...
    }

    err = tx.Commit()
    return
}


责任编辑:庞桂玉 来源: 马哥Linux运维
相关推荐

2017-07-10 13:09:45

前端Flexbox

2017-02-23 15:37:44

OptionObject容器

2016-01-05 11:28:20

按需付费云计算docker

2021-09-15 16:20:02

Spring BootFilterJava

2021-07-13 07:31:26

Springboot编程事务管理

2018-01-11 15:31:39

命令Linux关机

2017-03-16 11:39:33

Openstack源码姿势

2023-01-30 07:41:43

2016-05-09 10:41:03

算法分析开发

2017-10-12 11:30:34

Spark代码PR

2021-01-08 08:10:34

MySQL表空间回收

2020-08-31 06:54:37

注解脱敏ELK

2021-07-12 11:35:13

Go协程Goroutine

2019-10-30 17:06:50

AWS物联网IoT

2022-10-25 18:00:00

Redis事务生产事故

2021-10-08 08:38:00

Pipelineshell命令Jenkins

2019-01-02 10:49:54

Tomcat内存HotSpot VM

2017-06-19 17:43:04

网络带宽网络设计

2020-02-24 15:26:40

Python短信免费

2022-03-28 11:10:35

ShellBash forLinux
点赞
收藏

51CTO技术栈公众号