goの同時sync.WaitGroupとerrgroup

sync.WaitGroup

WaitGroupはその名の通り、一連の操作が完了するのを待つために使用されます.WaitGroupの内部には、未完了の操作数を記録するカウンタが実装されており、カウントを追加するための3つの方法が提供されています.Done()は、操作終了時に呼び出し、カウントを1つ減らすために使用されます.Wait()は、すべての操作が終了するのを待つために使用されます.すなわち、カウントが0になり、カウントが0でない場合に待機し、カウントが0の場合にすぐに戻ります.簡単な例を次に示します.

package main

import (
    "fmt"
    "sync"
)

func main() {

    var wg sync.WaitGroup

    wg.Add(2) //  ， 2 
    go func() {
        fmt.Println("Goroutine 1")
        wg.Done() //  ， 
    }()

    go func() {
        fmt.Println("Goroutine 2")
        wg.Done() //  ， 
    }()

    wg.Wait() //  ， 0
}

sync.WaitGroupの欠点はgoroutinesでエラーが発生した場合、戻ることができないため、エラーの原因を知ることはできません.

errgroup

公式サイトを参照してください.https://godoc.org/golang.org/x/sync/errgrouperrgroupパッケージはsync.WaitGroup機能に基づいて、エラー転送を追加したり、リカバリ不可能なエラーが発生した場合にgoroutineセット全体をキャンセルしたり、タイムアウトを待機したりします.次の3つの関数が含まれます.

func WithContext(ctx context.Context) (*Group, context.Context)

func (g *Group) Go(f func() error)

func (g *Group) Wait() error

上記の例を次のように書き換えます.

package main

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

func main() {

    var g errgroup.Group

	g.Go(func() error{
		fmt.Println("Goroutine 1")
		.......
		return err  // 
	})
   g.Go(func() error{
		fmt.Println("Goroutine 2")
		......
		return err  // 
	})
    
	if err := group.Wait(); err != nil {
		fmt.Println("Get errors: ", err)
	}else {
		fmt.Println("successful")
	}

}

次の例では、WithContext関数を使用します.これは、主に各goruntine間でデータを転送するのに便利です.

package main

import (
	"fmt"
	"time"

	"golang.org/x/sync/errgroup"
	"golang.org/x/net/context"
)

func checkGoroutineErr(errCtx context.Context) error {
	select {
	case errCtx.Done():
		return errCtx.Err()
	default:
		return nil
	}
}

func main() {
	ctx , cancel := context.WithCancel(context.Background())
	group, errCtx := errgroup.WithContext(ctx)

	for i := 0; i < 3; i++ {
		index := i
		group.Go(func() error {
			fmt.Println("index=", index)
			if index == 0 {
				fmt.Println("index == 0, end!")
			}else if index == 1 {
				fmt.Println("index == 1, start...")
				cancel()
				fmt.Println("inde == 1, has error!")
			}else if index == 2 {
				fmt.Println("index == 2, start...")
				time.Sleep(time.Second * 3)
				if err := checkGoroutineErr(errCtx); err != nil {
					return err
				}
				fmt.Println("index == 2, has done!")
			}
			return nil
		})
	}

	err := group.Wait()
	if err != nil {
		fmt.Println("Get error: ", err)
	}else {
		fmt.Println("All Done!")
	}
}