golang如何優雅的編寫事務代碼

時間 2020-05-26

標籤 golang 如何優雅編寫事務代碼欄目 Go 简体版

原文原文鏈接

前言

新手程序員大概有以下特色程序員

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

重構套路

1、提早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
}

先解決嵌套設計

2、goto+label提取重複代碼

代碼3d

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
}

3、封裝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