go errgroup 用法示例

Go团队在实验仓库中添加了一个名为sync.errgroup的新软件包。 sync.ErrGroup再sync.WaitGroup功能的基础上，增加了错误传递，以及在发生不可恢复的错误时取消整个goroutine集合，或者等待超时

其中包含的函数如下：

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

 

Go()方法不仅允许你传一个匿名的函数，而且还能捕获错误信息，你只要像这样返回一个错误 return err，这使开发者使用goroutines时开发效率显著提高
func (g *Group) Go(f func() error)

func (g *Group) Wait() error

示例1（代码中已有详细的注释）：

package main
import (
	"fmt"
	"golang.org/x/sync/errgroup"
	"errors"
)
func main() {
	group := new(errgroup.Group)
	nums := []int{-1, 0, 1}

	for _, num := range nums {
		tempNum := num // 子协程中若直接访问num，则可能是同一个变量，所以要用临时变量

		// 子协程
		group.Go(func() error {
			if tempNum < 0 {
				return  errors.New("tempNum < 0 !!!")
			}
			fmt.Println("tempNum:",tempNum)
			return nil
		})
	}

	// 捕获err
	if err := group.Wait(); err != nil {
		fmt.Println("Get errors: ", err)
	}else {
		fmt.Println("Get all num successfully!")
	}
}

执行结果如下图（正常捕获到了err）：

示例2（代码中已有详细的注释）：

package main
import (
	"fmt"
	"time"
	xContext "golang.org/x/net/context"
	"golang.org/x/sync/errgroup"
)
func main() {
	ctx, cancel := xContext.WithCancel(xContext.Background())
	group, errCtx := errgroup.WithContext(ctx)

	for index := 0; index < 3; index++ {
		indexTemp := index // 子协程中若直接访问index，则可能是同一个变量，所以要用临时变量

		// 新建子协程
		group.Go(func() error {
			fmt.Printf("indexTemp=%d \n", indexTemp)
			if indexTemp == 0 {
				fmt.Println("indexTemp == 0 start ")
				fmt.Println("indexTemp == 0 end")
			} else if indexTemp == 1 {
				fmt.Println("indexTemp == 1 start")
				//这里一般都是某个协程发生异常之后，调用cancel()
				//这样别的协程就可以通过errCtx获取到err信息，以便决定是否需要取消后续操作
				cancel()
				fmt.Println("indexTemp == 1 err ")
			} else if indexTemp == 2 {
				fmt.Println("indexTemp == 2 begin")

				// 休眠1秒，用于捕获子协程2的出错
				time.Sleep(1 * time.Second)

				//检查 其他协程已经发生错误，如果已经发生异常，则不再执行下面的代码
				err := CheckGoroutineErr(errCtx)
				if err != nil {
					return err
				}
				fmt.Println("indexTemp == 2 end ")
			}
			return nil
		})
	}

	// 捕获err
	err := group.Wait()
	if err == nil {
		fmt.Println("都完成了")
	} else {
		fmt.Printf("get error:%v", err)
	}
}

//校验是否有协程已发生错误
func CheckGoroutineErr(errContext xContext.Context) error {
	select {
	case <-errContext.Done():
		return errContext.Err()
	default:
		return nil
	}
}

执行结果如下图（捕获到了cancel）：