快捷導(dǎo)航

golang如何優(yōu)雅的編寫事務(wù)代碼示例

更新時(shí)間：2020年05月26日 14:28:06 作者：雪山飛豬

這篇文章主要介紹了golang如何優(yōu)雅的編寫事務(wù)代碼示例，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

前言

新手程序員大概有如下特點(diǎn)

if嵌套經(jīng)常超過(guò)3層、經(jīng)常出現(xiàn)重復(fù)代碼、單個(gè)函數(shù)代碼特別長(zhǎng)。
只會(huì)crud，對(duì)語(yǔ)言特性和語(yǔ)言的邊界不了解。
不懂面向?qū)ο笤瓌t和設(shè)計(jì)模式，以為copy代碼就算學(xué)會(huì)了，常見(jiàn)的是代碼職責(zé)不明確或者寫出萬(wàn)能類
不知道數(shù)據(jù)結(jié)構(gòu)和算法的重要性，以為靠硬件能解決運(yùn)行慢的問(wèn)題
架構(gòu)不懂，搭建框架不會(huì)，搭建環(huán)境不會(huì)，使用的軟件底層原理一問(wèn)三不知

其實(shí)吧，很多人干了很多年，看似是老手，平時(shí)工作看似很忙，其實(shí)做的都是最簡(jiǎn)單的活。
這就像去鍛煉，有的人每天練的很積極，準(zhǔn)時(shí)打卡，頻繁發(fā)朋友圈，貌似是正能量，結(jié)果是幾年下來(lái)體型還是那樣，該減的肥肉沒(méi)少，要增的肌肉沒(méi)加，為什么會(huì)這樣？因?yàn)閺膩?lái)都是挑最簡(jiǎn)單最輕松的練

貌似吐槽多了，下面演示一下如何將一坨爛事務(wù)代碼重構(gòu)得優(yōu)雅

需求

執(zhí)行一個(gè)事務(wù)，需要調(diào)用one、two、three、four、five幾個(gè)方法，任意一個(gè)方法失敗，都回滾事務(wù)
下面是這些方法的簡(jiǎn)單模擬，我們用盡可能少的代碼模擬一個(gè)操作

//開(kāi)啟事務(wù)
func beginTransaction() {
 fmt.Println("beginTransaction")
}

//回滾事務(wù)
func rollback() {
 fmt.Println("rollback")
}

//提交事務(wù)
func commit() {
 fmt.Println("commit")
}

//執(zhí)行one操作
func one() (err error) {
 fmt.Println("one ok")
 return nil
}

//執(zhí)行two操作
func two() (err error) {
 fmt.Println("two ok")
 return nil
}

//執(zhí)行three操作
func three() (err error) {
 fmt.Println("two ok")
 return nil
}

//執(zhí)行four操作
func four() (err error) {
 fmt.Println("four ok")
 return nil
}

//執(zhí)行five操作
func five() (err error) {
 err = errors.New("five panic")
 panic("five")
 return err
}

爛代碼示例

下面演示開(kāi)啟一個(gè)事務(wù)，依次執(zhí)行one、two、three、four、five 5個(gè)操作，前四個(gè)成功，第五個(gè)失敗

這是新手程序員常見(jiàn)使用事務(wù)的代碼風(fēng)格，其實(shí)也不光是事務(wù)，所有的代碼都可能長(zhǎng)下邊這樣

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
 }
}

重構(gòu)套路

一、提前return去除if嵌套

通過(guò)提前返回error，來(lái)去掉一些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提取重復(fù)代碼

代碼

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統(tǒng)一捕獲panic

上面的代碼其實(shí)還有一點(diǎn)問(wèn)題

defer里有rollback的代碼
goto雖然好，但是不建議使用

我們可以對(duì)panic和defer進(jìn)行封裝，模擬一下try-catch，實(shí)現(xiàn)如下

可以看到，rollback只調(diào)用了一次，完美的進(jìn)行了事務(wù)代碼重構(gòu)

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
}

這樣，我們就可以用非常少的代碼實(shí)現(xiàn)事務(wù)，并且簡(jiǎn)單清晰好維護(hù)，以上為chenqionghe原創(chuàng)，light weight baby

到此這篇關(guān)于golang如何優(yōu)雅的編寫事務(wù)代碼示例的文章就介紹到這了,更多相關(guān)golang 編寫事務(wù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: