go的并发sync.WaitGroup和errgroup

sync.WaitGroup

WaitGroup顾名思义,就是用来等待一组操作完成的。WaitGroup内部实现了一个计数器,用来记录未完成的操作个数,它提供了三个方法,Add()用来添加计数。Done()用来在操作结束时调用,使计数减一。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/errgroup
errgroup包在sync.WaitGroup功能的基础上,增加了错误传递,以及在发生不可恢复的错误时取消整个goroutine集合,或者等待超时。
包含的三个函数:

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!")
	}
}

你可能感兴趣的:(并发)