golang 编写事务 golang怎样优雅的编写事务代码示例
雪山飞猪 人气:0前言
新手程序员大概有如下特点
- if嵌套经常超过3层、经常出现重复代码、单个函数代码特别长。
- 只会crud,对语言特性和语言的边界不了解。
- 不懂面向对象原则和设计模式,以为copy代码就算学会了,常见的是代码职责不明确或者写出万能类
- 不知道数据结构和算法的重要性,以为靠硬件能解决运行慢的问题
- 架构不懂,搭建框架不会,搭建环境不会,使用的软件底层原理一问三不知
其实吧,很多人干了很多年,看似是老手,平时工作看似很忙,其实做的都是最简单的活。
这就像去锻炼,有的人每天练的很积极,准时打卡,频繁发朋友圈,貌似是正能量,结果是几年下来体型还是那样,该减的肥肉没少,要增的肌肉没加,为什么会这样?因为从来都是挑最简单最轻松的练
貌似吐槽多了,下面演示一下如何将一坨烂事务代码重构得优雅
需求
执行一个事务,需要调用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
上面的代码其实还有一点问题
- defer里有rollback的代码
- goto虽然好,但是不建议使用
我们可以对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
加载全部内容