快捷導(dǎo)航

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

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

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

前言

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

一般我們的業(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()
}

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

通過(guò)錯(cuò)誤日志記錄

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

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

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

利用 channel 傳輸

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

第二種方法：利用 channel 來(lái)傳輸多個(gè) 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("出錯(cuò)啦。。我是錯(cuò)誤信息")
}

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

使用 sync/errgroup

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

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

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

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

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
}

每啟動(dòng)一個(gè)新的 goroutine 都直接使用 Group.Go 方法，在等待和錯(cuò)誤處理上使用 Group.Wait 方法。

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