快捷導(dǎo)航

go goroutine 怎樣進(jìn)行錯誤處理

更新時間：2021年07月16日 10:15:40 作者：mb60e703e6a8897

在 Go 語言程序開發(fā)中，goroutine 的使用是比較頻繁的，因此在日常編碼的時候 goroutine 里的錯誤處理，怎么做會比較好呢，本文就來詳細(xì)介紹一下

前言

在 Go 語言程序開發(fā)中，goroutine 的使用是比較頻繁的，因此在日常編碼的時候 goroutine 里的錯誤處理，怎么做會比較好呢？

一般我們的業(yè)務(wù)代碼如下：

func main() {
 var wg sync.WaitGroup
 wg.Add(2)
 go func() {
  //... 業(yè)務(wù)邏輯
  wg.Done()
 }()
 go func() {
  //... 業(yè)務(wù)邏輯
  wg.Done()
 }()
 wg.Wait()
}

在上面的代碼中，我們運行了多個 goroutine，每個協(xié)程又是單獨行動的，想要拋出 error 錯誤信息，也不怎么明智。

通過錯誤日志記錄

常用的第一種方法：通過把錯誤記錄寫入到日志文件中，再結(jié)合相關(guān)的 logtail 進(jìn)行采集和梳理。

但這又會引入新的問題，那就是調(diào)用錯誤日志的方法寫的到處都是，代碼結(jié)構(gòu)也比較亂、不直觀。

最重要的是無法針對 error 做特定的邏輯處理和流轉(zhuǎn)。

利用 channel 傳輸

大家可能會想到 Go 的經(jīng)典哲學(xué)：不要通過共享內(nèi)存來通信，而是通過通信來實現(xiàn)內(nèi)存共享（Do not communicate by sharing memory; instead, share memory by communicating）。

第二種方法：利用 channel 來傳輸多個 goroutine 中的 errors：

func main() {
 cherrors := make(chan error)
 wgDone := make(chan bool)

 var wg sync.WaitGroup
 wg.Add(2)
 go func() {
  //... 業(yè)務(wù)邏輯
  wg.Done()
 }()
 go func() {
  //... 業(yè)務(wù)邏輯
  err := returnErr()
  if err != nil {
   cherrors <- err
  }
  wg.Done()
 }()
 go func() {
  wg.Wait()
  close(wgDone)
 }()

 select {
 case <-wgDone:
  break
 case err := <-cherrors:
  close(cherrors)
  fmt.Println(err)
 }

 time.Sleep(time.Second)
}

func returnErr() error {
 return errors.New("出錯啦。。我是錯誤信息")
}

雖然使用 channel 后已經(jīng)方便了不少，但編寫 channel 還要關(guān)心一些非業(yè)務(wù)向的邏輯。

使用 sync/errgroup

第三種方法，就是使用官方提供的 golang.org/x/sync/errgroup 標(biāo)準(zhǔn)庫：

type Group
    func WithContext(ctx context.Context) (*Group, context.Context)
    func (g *Group) Go(f func() error)
    func (g *Group) Wait() error

Go：啟動一個協(xié)程，在新的 goroutine 中調(diào)用給定的函數(shù)。
Wait：等待協(xié)程結(jié)束，直到 Go 方法中的所有函數(shù)調(diào)用都返回，然后返回其中第一個非零錯誤（如果有錯誤的話）。

結(jié)合其特性能夠非常便捷的針對多 goroutine 進(jìn)行錯誤處理：

func main() {
 group := new(errgroup.Group)

 nums := []int{-1, 0, 1}
 for _, num := range nums {
  num := num
  group.Go(func() error {
   res, err := output(num)
   fmt.Println(res)
   return err
  })
 }

 if err := group.Wait(); err != nil {
  fmt.Println("Get errors: ", err)
 } else {
  fmt.Println("Get all num successfully!")
 }
}

func output(num int) (int, error) {
 if num < 0 {
  return 0, errors.New("math: square root error!")
 }
 return num, nil
}

每啟動一個新的 goroutine 都直接使用 Group.Go 方法，在等待和錯誤處理上使用 Group.Wait 方法。

這種方法進(jìn)行錯誤處理的好處是不需要關(guān)注非業(yè)務(wù)邏輯的控制代碼，比較簡潔明了。