前言
新手程序员大概有如下特点
其实吧，很多人干了很多年，看似是老手，平时工作看似很忙，其实做的都是最简单的活。
这就像去锻炼，有的人每天练的很积极，准时打卡，频繁发朋友圈，貌似是正能量，结果是几年下来体型还是那样，该减的肥肉没少，要增的肌肉没加，为什么会这样？因为从来都是挑最简单最轻松的练
貌似吐槽多了，下面演示一下如何将一坨烂事务代码重构得优雅
需求
执行一个事务，需要调用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
