[[396764]]

構(gòu)造 error

在 go 語言中，有一個(gè)預(yù)定義的接口：error，該接口自帶一個(gè) Error() 方法，調(diào)用該方法會(huì)返回一個(gè)字符串。

type error interface { 
  Error() string 
} 

調(diào)用該方法，會(huì)返回當(dāng)前錯(cuò)誤的具體結(jié)果。一般有下面幾種方式生成 error。

• errors.New()
• fmt.Errorf()

errors.New()

調(diào)用 errors.New() 會(huì)返回一個(gè) error 類型的結(jié)構(gòu)體，該結(jié)構(gòu)體內(nèi)部會(huì)實(shí)現(xiàn)一個(gè) Error() 方法， 調(diào)用該方法返回的結(jié)果為調(diào)用 errors.New() 方法時(shí)傳入的內(nèi)容。

import ( 
 "errors" 
 "fmt" 
) 
 
func divide(a, b int) (error, int) { 
 if b == 0 { 
    // 被除數(shù)為0，則構(gòu)造一個(gè) error 結(jié)構(gòu)體 
  return errors.New("被除數(shù)不能為0"), 0 
 } 
 var result = a / b 
 return nil, result 
} 
 
func main() { 
 var err error // error 類型數(shù)據(jù)的初始值為 nil，類似于 js 中的 null 
 var result int 
 
 err, result = divide(1, 0) 
 
  if err == nil { 
    // 如果 err 為 nil，說明運(yùn)行正常 
    fmt.Println("計(jì)算結(jié)果", result) 
  } else { 
    // 如果 err 不為 nil，說明運(yùn)行出錯(cuò) 
    // 調(diào)用 error 結(jié)構(gòu)體的 Error 方法，輸出錯(cuò)誤原因 
    fmt.Println("計(jì)算出錯(cuò)", err.Error()) 
  } 
} 

可以看到，上面的代碼中，由于調(diào)用 divide 除法方法時(shí)，由于傳入的被除數(shù)為 0。經(jīng)過判斷，會(huì)拋出一個(gè)由 errors.New 構(gòu)造的 error 類型的結(jié)構(gòu)體。

我們將調(diào)用 error.Error() 方法返回的結(jié)果輸出到控制臺(tái)，可以發(fā)現(xiàn)其返回的結(jié)果，就是傳入 New 方法的值。

執(zhí)行結(jié)果如下：

fmt.Errorf()

通過 fmt.Errorf() 方法構(gòu)造的 error 結(jié)構(gòu)體，與調(diào)用 errors.New() 方法的結(jié)果類似。不同的是，fmt.Errorf() 方法會(huì)進(jìn)行一次數(shù)據(jù)的格式化。

func divide(a, b int) (error, int) { 
 if b == 0 { 
    // 將參數(shù)進(jìn)行一次格式化，格式化后的字符串放入 error 中 
  return fmt.Errorf("數(shù)據(jù) %d 不合法", b), 0 
 } 
 var result = a / b 
 return nil, result 
} 
 
err, result := divide(1, 0) 
fmt.Println("計(jì)算出錯(cuò)", err.Error()) 

執(zhí)行結(jié)果如下：

panic() 與 recover()

panic()

panic() 相當(dāng)于主動(dòng)停止程序運(yùn)行，調(diào)用時(shí) panic() 時(shí)，需要傳入中斷原因。調(diào)用后，會(huì)在控制臺(tái)輸出中斷原因，以及中斷時(shí)的調(diào)用堆棧。我們可以改造一下之前的代碼：

func divide(a, b int) (error, int) { 
 if b == 0 { 
    // 如果程序出錯(cuò)，直接停止運(yùn)行 
  panic("被除數(shù)不能為0") 
 } 
 var result = a / b 
 return nil, result 
} 
 
func main() { 
  err, result := divide(1, 0) 
  fmt.Println("計(jì)算出錯(cuò)", err.Error()) 
} 

在運(yùn)行到 panic() 處，程序直接中斷，并在控制臺(tái)打印出了中斷原因。

panic() 可以理解為，js 程序中的 throw new Error() 的操作。那么，在 go 中有沒有辦法終止 panic() ，也就是類似于 try-catch 的操作，讓程序回到正常的運(yùn)行邏輯中呢?

recover()

在介紹 recover() 方法之前，還需要介紹一個(gè) go 語言中的另一個(gè)關(guān)鍵字：defer。

defer 后的語句會(huì)在函數(shù)進(jìn)行 return 操作之前調(diào)用，常用于資源釋放、錯(cuò)誤捕獲、日志輸出。

func getData(table, sql) { 
  defer 中斷連接() 
  db := 建立連接(table) 
  data := db.select(sql) 
  return data 
} 

defer 后的語句會(huì)被存儲(chǔ)在一個(gè)類似于棧的數(shù)據(jù)結(jié)構(gòu)內(nèi)，在函數(shù)結(jié)束的時(shí)候，被定義的語句按順序出棧，越后面定義的語句越先被調(diào)用。

func divide(a, b int) int { 
  defer fmt.Println("除數(shù)為", b) 
  defer fmt.Println("被除數(shù)為", a) 
 
  result := a / b 
  fmt.Println("計(jì)算結(jié)果為", result) 
 return result 
} 
 
divide(10, 2) 

上面的代碼中，我們?cè)诤瘮?shù)開始運(yùn)行的時(shí)候，先通過 defer 定義了兩個(gè)輸出語句，先輸出除數(shù)，后輸出被除數(shù)。

實(shí)際的運(yùn)行結(jié)果是：

• 先輸出計(jì)算結(jié)果;
• 然后輸出被除數(shù);
• 最后輸出除數(shù);

這和前面提到的，通過 defer 定義的語句會(huì)在函數(shù)結(jié)束的時(shí)候，按照出棧的方式進(jìn)行執(zhí)行，先定義的后執(zhí)行。defer 除了會(huì)在函數(shù)結(jié)束的時(shí)候執(zhí)行，出現(xiàn)異常的的時(shí)候也會(huì)先走 defer 的邏輯，也就是說，我們?cè)谡{(diào)用了 panic() 方法后，程序中斷過程中，也會(huì)先將 defer 內(nèi)的語句運(yùn)行一遍。

這里我們重新定義之前的 divide 函數(shù)，在執(zhí)行之前加上一個(gè) defer 語句，defer 后面為一個(gè)自執(zhí)行函數(shù)，該函數(shù)內(nèi)會(huì)調(diào)用 recover() 方法。

recover() 方法調(diào)用后，會(huì)捕獲到當(dāng)前的 panic() 拋出的異常，并進(jìn)行返回，如果沒有異常，則返回 nil。

func divide(a, b int) int { 
  // 中斷之前，調(diào)用 defer 后定義的語句 
 defer func() { 
  if err := recover(); err != nil { 
   fmt.Println("捕獲錯(cuò)誤", err) 
  } 
 }() 
 
 if b == 0 { 
    // 函數(shù)運(yùn)行被中斷 
  panic("被除數(shù)不能為0") 
  return 0 
 } 
 
 return a / b 
} 
 
result := divide(1, 0) 
fmt.Println("計(jì)算結(jié)果", result) 

上面的代碼運(yùn)行后，我們發(fā)現(xiàn)之前調(diào)用 panic() 中斷的程序被恢復(fù)了，而且后面的計(jì)算結(jié)果也正常進(jìn)行輸出了。

這就有點(diǎn)類似于 try-catch 的邏輯了，只是 recover 需要放在 defer 關(guān)鍵詞后的語句中，更像是 catch 和 finally 的結(jié)合。

本文轉(zhuǎn)載自微信公眾號(hào)「自然醒的筆記本」，可以通過以下二維碼關(guān)注。轉(zhuǎn)載本文請(qǐng)聯(lián)系自然醒的筆記本公眾號(hào)。