这篇文章主要讲解了golang编写事务的方法,内容清晰明了,对此有兴趣的小伙伴可以学习一下,相信大家阅读完之后会有帮助。
前言
新手程序员大概有如下特点
其实吧,很多人干了很多年,看似是老手,平时工作看似很忙,其实做的都是最简单的活。
这就像去锻炼,有的人每天练的很积极,准时打卡,频繁发朋友圈,貌似是正能量,结果是几年下来体型还是那样,该减的肥肉没少,要增的肌肉没加,为什么会这样?因为从来都是挑最简单最轻松的练
貌似吐槽多了,下面演示一下如何将一坨烂事务代码重构得优雅
需求
执行一个事务,需要调用one、two、three、four、five几个方法,任意一个方法失败,都回滚事务
下面是这些方法的简单模拟,我们用尽可能少的代码模拟一个操作
//开启事务 func beginTransaction() { fmt.Println("beginTransaction") } //回滚事务 func rollback() { fmt.Println("rollback") } //提交事务 func commit() { fmt.Println("commit") } //执行one操作 func one() (err error) { fmt.Println("one ok") return nil } //执行two操作 func two() (err error) { fmt.Println("two ok") return nil } //执行three操作 func three() (err error) { fmt.Println("two ok") return nil } //执行four操作 func four() (err error) { fmt.Println("four ok") return nil } //执行five操作 func five() (err error) { err = errors.New("five panic") panic("five") return err }
烂代码示例
下面演示开启一个事务,依次执行one、two、three、four、five 5个操作,前四个成功,第五个失败
这是新手程序员常见使用事务的代码风格,其实也不光是事务,所有的代码都可能长下边这样
func demo() (err error) { beginTransaction() defer func() { if e := recover(); e != nil { err = fmt.Errorf("%v", e) fmt.Printf("%v panic\n", e) rollback() } }() if err = one(); err == nil { if err = two(); err == nil { if err = three(); err == nil { if err = four(); err == nil { if err = five(); err == nil { commit() return nil } else { rollback() return err } } else { rollback() return err } } else { rollback() return err } } else { rollback() return err } } else { rollback() return err } }
重构套路
一、提前return去除if嵌套
通过提前返回error,来去掉一些else代码,减少嵌套,如下
代码
func demo() (err error) { beginTransaction() defer func() { if e := recover(); e != nil { err = fmt.Errorf("%v", e) fmt.Printf("%v panic\n", e) rollback() } }() if err = one(); err != nil { rollback() return err } if err = two(); err != nil { rollback() return err } if err = three(); err != nil { rollback() return err } if err = four(); err != nil { rollback() return err } if err = five(); err != nil { rollback() return err } commit() return nil }
先解决嵌套
二、goto+label提取重复代码
代码
func demo() (err error) { beginTransaction() defer func() { if e := recover(); e != nil { err = fmt.Errorf("%v", e) fmt.Printf("%v panic\n", e) rollback() } }() if err = one(); err != nil { goto ROLLBACK } if err = two(); err != nil { goto ROLLBACK } if err = three(); err != nil { goto ROLLBACK } if err = four(); err != nil { goto ROLLBACK } if err = five(); err != nil { goto ROLLBACK } commit() return nil ROLLBACK: rollback() return err }
三、封装try-catch统一捕获panic
上面的代码其实还有一点问题
我们可以对panic和defer进行封装,模拟一下try-catch,实现如下
可以看到,rollback只调用了一次,完美的进行了事务代码重构
try-catch.go
代码
package exception type Block struct { Try func() Catch func(interface{}) Finally func() } func (t Block) Do() { if t.Finally != nil { defer t.Finally() } if t.Catch != nil { defer func() { if r := recover(); r != nil { t.Catch(r) } }() } t.Try() }
使用代码
exception.Block{ Try: func() { beginTransaction() if err = one(); err != nil { panic(err) } if err = two(); err != nil { panic(err) } if err = three(); err != nil { panic(err) } if err = four(); err != nil { panic(err) } if err = five(); err != nil { panic(err) } err = nil commit() }, Catch: func(e interface{}) { rollback() fmt.Printf("%v panic\n", e) err = fmt.Errorf("%v", e) }, }.Do() return err }
这样,我们就可以用非常少的代码实现事务,并且简单清晰好维护,以上为chenqionghe原创,light weight baby
看完上述内容,是不是对golang编写事务的方法有进一步的了解,如果还想学习更多内容,欢迎关注亿速云行业资讯频道。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。