快捷導航

golang中channel+error來做異步錯誤處理有多香

更新時間：2023年01月19日 10:22:49 作者：文大俠666

官方推薦golang中錯誤處理當做值處理，既然是值那就可以在channel中傳輸，這篇文章主要介紹了golang 錯誤處理channel+error真的香,需要的朋友可以參考下

場景

如下，一次任務(wù)起多個協(xié)程異步處理任務(wù)，比如同時做服務(wù)/redis/mysql/kafka初始化，當某一個協(xié)程出現(xiàn)錯誤(初始化失敗)時，程序是停止還是繼續(xù)呢？如何記錄錯誤？如何控制優(yōu)雅的退出全部工作協(xié)程呢？

為了解決類似的問題，常見如下三種解決方案：

1.中斷退出并記錄日志

如下，最簡單粗暴的方式就是，一旦協(xié)程中發(fā)生錯誤，記錄日志立即退出，外層如果想取到錯誤可以通過共享全局變量，單個協(xié)程無法控制所有協(xié)程動作。

// 出錯中斷協(xié)程，打印日志，退出

func TestSimpleExit(t *testing.T) {
	wg := sync.WaitGroup{}

	wg.Add(1)
	go func() {
		defer wg.Done()

		//do something
		if err := doSomething(); err != nil {
			t.Logf("Error when call doSomething:%v\n", err)
			return
		}
	}()

	wg.Wait()
}

2.監(jiān)控error，可選記錄日志或退出

前面講過，既然error是值，那就可以在channel中傳輸，可以單獨開一個channel，所有協(xié)程錯誤都發(fā)送到這個通道。

	// 數(shù)據(jù)處理流程
	dataFunc := func(ctx context.Context, dataChan chan int, errChan chan error) {
		defer wg.Done()

		for {
			select {
			case v, ok := <-dataChan:
				if !ok {
					log.Println("Receive data channel close msg!")
					return
				}

				if err := doSomething2(v); err != nil {
					errChan <- err
					continue
				}

				// do ...

			case <-ctx.Done():
				log.Println("Receive exit msg!")
				return
			}
		}
	}

	wg.Add(1)
	go dataFunc(ctx, dataChan, errChan)

	wg.Add(1)
	go dataFunc(ctx, dataChan, errChan)

監(jiān)控錯誤error通道，統(tǒng)一記錄和退出，一旦檢測到錯誤可以通過ctx通知所有協(xié)程退出，這里可以靈活控制監(jiān)控到錯誤時的錯誤處理策略(是否記錄日志/是否退出等)，error通道可以同步或異步處理。
整體流程如下

異步監(jiān)控error

	// 錯誤處理流程，error處理通道異步等待
	wg.Add(1)
	go func(errChan chan error) {
		defer wg.Done()

		for {
			select {
			case v, ok := <-errChan:
				if !ok {
					log.Println("Receice err channel close msg!")
					return
				}

				// 收到錯誤時，可選擇記錄日志或退出
				if v != nil {
					t.Logf("Error when call doSomething:%v\n", v)
					cancel() // 通知全部退出
					return
				}

			case <-ctx.Done():
				log.Println("Receive exit msg!")
				return
			}
		}

	}(errChan)

	dataChan <- 1
	wg.Wait()

同步監(jiān)控error

	// 錯誤處理流程，error處理通道同步等待
	for {
		select {
		case v, ok := <-errChan:
			if !ok {
				log.Println("Receice err channel close msg!")
				goto EXIT
			}

			// 收到錯誤時，可選擇記錄日志或退出
			if v != nil {
				t.Logf("Error when call doSomething:%v\n", v)
				cancel()
				goto EXIT
			}

		case <-ctx.Done():
			log.Println("Receive exit msg!")
			goto EXIT
		}
	}

EXIT:
	wg.Wait()

3.官方庫errgroup

考慮到error輸出到通道后統(tǒng)一處理是golang常用手段，官方也針對封裝了一個error處理包，errgroup顧名思義，多個協(xié)程的error被當做一個組，一旦某個協(xié)程出錯所有協(xié)程都退出，只輸出第一個error。

func TestSimpleChannel5(t *testing.T) {
	eg, ctx := errgroup.WithContext(context.Background())

	dataChan := make(chan int)
	defer close(dataChan)

	// 數(shù)據(jù)處理流程
	dataFunc := func() error {
		for {
			select {
			case v, ok := <-dataChan:
				if !ok {
					log.Println("Receive data channel close msg!")
					return nil
				}

				if err := doSomething2(v); err != nil {
					return err
				}

				// do ...

			// 增加ctx通知完成
			case <-ctx.Done():
				log.Println("Receive exit msg!")
				return nil
			}
		}
	}

	eg.Go(dataFunc)
	eg.Go(dataFunc)
	eg.Go(dataFunc)

	dataChan <- 1

	// 錯誤處理流程，任何一個協(xié)程出現(xiàn)error，則會調(diào)用ctx對應(yīng)cancel函數(shù)，所有相關(guān)協(xié)程都會退出
	if err := eg.Wait(); err != nil {
		fmt.Printf("Something is wrong->%v\n", err)
	}
}

類似上一小節(jié)，可以看到errgroup就是結(jié)合waitgroup cancel和channel通道封裝的。

4.監(jiān)控error，全部日志合并后輸出

同樣是上述場景，有時候我們的需求是返回所有的錯誤(不是第一個錯誤)。

比如在服務(wù)啟動時，對 redis、kafka、mysql 等各種資源初始化場景，可以把所有相關(guān)資源初始化的錯誤都返回，展示給用戶統(tǒng)一排查。
另一種場景就是在 web 請求中，校驗請求參數(shù)時，返回所有參數(shù)的校驗錯誤給客戶端的場景。

這種需求，一般考慮使用多錯誤管理(hashicorp/go-multierror庫)，如下一個簡答同步任務(wù)演示多錯誤管理，所有返回的錯誤可以通過Append歸并成一個錯誤，實際上是通過error wrap的方式合并起來的，因此也可以使用Is/As判斷嵌套error。

// 多路協(xié)程error合并，用于多路check場景
func TestSimpleChannel3(t *testing.T) {

	// 同步執(zhí)行多個任務(wù)，返回error合并
	var err = func() error {
		var result error

		if err := doSomething(); err != nil {
			result = multierror.Append(result, err)
		}

		if err := doSomething2(nil); err != nil {
			result = multierror.Append(result, err)
		}

		return result
	}()

	// 打印輸出
	if err != nil {
		fmt.Printf("%v\n", err)
	}

	// 獲取錯誤列表
	if err != nil {
		if merr, ok := err.(*multierror.Error); ok {
			fmt.Printf("%v\n", merr.Errors)
		}
	}

	// 判斷是否為某種類型
	if err != nil && errors.Is(err, Error1) {
		fmt.Println("Errors contain error 1")
	}

	// 判斷是否其中一個error能夠轉(zhuǎn)換成指定error
	var e MyError
	if err != nil && errors.As(err, &e) {
		fmt.Println("One Error can be convert to nyerror")
	}
}

那么，在起多個異步任務(wù)時，就可以如下處理，返回的多個error通過channel消費合并展示。

func TestSimpleChannel4(t *testing.T) {
	wg := sync.WaitGroup{}

	taskNum := 10
	errChan := make(chan error, taskNum)

	// 異步執(zhí)行多個任務(wù)
	step := func(stepNum int, errChan chan error) {
		defer wg.Done()
		errChan <- fmt.Errorf("step %d error", stepNum)
	}

	for i := 0; i < taskNum; i++ {
		wg.Add(1)
		go step(i, errChan)
	}

	// 等待任務(wù)完成
	go func() {
		wg.Wait()
		close(errChan)
	}()

	// err通道阻塞等待，可能的所有錯誤合并
	var result *multierror.Error
	for err := range errChan {
		result = multierror.Append(result, err)
	}

	// 出現(xiàn)一個錯誤時，選擇記錄日志或退出
	if len(result.Errors) != 0 {
		log.Println(result.Errors)
	}
}