golang如何優雅的編寫事務代碼
前言
新手程序員概有如下特點
- if嵌套特別多、重複代碼多。
- 不懂面向對象原則和設計模式，以為拿到源代碼就算學會了，最常見的是封裝的類或方法亂放，或者寫出萬能util類
- 只會crud，對語言特性和語言的邊界不了解。
- 不知道數據結構和算法的重要性，以為靠硬件能解決運行慢的問題
- 架構不懂，搭建框架不會，搭建環境不會，使用的軟件底層原理一問三不知
其實吧，很多人幹了很多年，看似是老手，平時工作看似很忙，其實做的都是最簡單的活。
這就像去鍛煉，有的人每天練的很積極，準時打卡，頻繁發朋友圈，貌似是正能量，結果是幾年下來體型還是那樣，該減的肥肉沒少，要增的肌肉沒加，為什麼會這樣？因為從來都是挑最簡單最輕鬆的練
貌似吐槽多了，下面演示一下如何將一坨爛事務代碼重構得優雅
需求
執行一個事務，需要調用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
}
}
重構套路
一、提前返回去除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，實現如下
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()
}
使用代碼
func demo() (err error) {
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{}) {
err = fmt.Errorf("%v", e)
fmt.Printf("%v panic\n", e)
},
}.Do()
return err
}
這樣，我們就可以用非常少的代碼實現事務，並且簡單清晰好維護，以上為chenqionghe原創，light weight baby
