Go_エラーハンドリングと並行処理

GOGOGO/00_32_errorGroup at master · ywsrock/GOGOGO

My Golang Trial Validation Only. Contribute to ywsrock/GOGOGO development by creating an account on GitHub.

https://github.com/ywsrock/GOGOGO/tree/master/00_32_errorGroup

package main

import (
  "errors"
  "fmt"
  "golang.org/x/sync/errgroup"
  "runtime/debug"
)

func main() {
  // エラーグループ起動
  // 実行タスク指定
  result, err := runGoErrGroup(subPro, subPro1)
  if err != nil {
    fmt.Println(err)
  }
  for _, v := range result {
    fmt.Println(v)
  }

}

func runGoErrGroup(fns ...func(int) (int, error)) ([]int, error) {
  eg := errgroup.Group{}
  result := make([]int, len(fns))

  for i, fn := range fns {
    i, fn := i, fn
    eg.Go(func() error {
      defer func() {
        if err := recover(); err != nil {
          debug.PrintStack()
        }
      }()
      // タスク実行
      k, err := fn(i)
      // エラーなければ、結果を設定、エラーの場合、処理終了
      if err == nil {
        result[i] = k
      }
      return err
    })
  }
  // エラー終了待ち
  if err := eg.Wait(); err != nil {
    // エラーを無視して、結果を返す
    // return result, err
    // エラー終了 log.Fatal(err)
    // log.Println(err)
    return nil, err // 結果を無視、エラー返す
  }
  return result, nil
}

func subPro(i int) (int, error) {
  if i == 0 {
    return i, errors.New("0 で割ることはできません")
  }
  return i, nil
}

func subPro1(i int) (int, error) {
  return i, nil
}

main 関数

1. runGoErrGroup 関数を呼び出し、実行するタスクとして subPro と subPro1 関数を指定します。

2. runGoErrGroup 関数の戻り値として、結果とエラーを受け取ります。

3. もしエラーが発生した場合は、エラーメッセージを表示します。

4. 結果を順番に表示します。

runGoErrGroup 関数

1. errgroup.Group を初期化します。

2. 結果を格納するためのスライス result を作成します。

3. 各関数とインデックスをループで処理し、ゴルーチンとして実行します。

4. ゴルーチン内では、タスクの実行結果を result スライスに格納します。

5. もしエラーが発生した場合、エラーを返して処理を終了します。

6. eg.Wait() を呼び出して、すべてのゴルーチンの終了を待ちます。もしエラーが発生した場合は、エラーを返します。

サブ関数