什么情况下使用 ErrGroup VS waitGroup?

Go Official Blog

共 3919字,需浏览 8分钟

 ·

2024-05-12 21:46

Goroutine 是编写 Go 语言并发程序的强大工具。然而管理协程,特别是在处理协程的错误时,可能会变得繁琐。这时,x/sync 包中的 errgroup 就派上用场了。它提供了一种简化的方法来处理并发任务及其错误。

Example for ErrGroup

下面是一个使用 errorGroup 的例子:

package main

import (
 "context"
 "errors"
 "fmt"
 "time"

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

func task1(ctx context.Context) error {
 fmt.Println("Task 1 started successfully")
 select {
 case <-time.After(1 * time.Second):
  fmt.Println("Task 1 completed successfully")
  return nil
 case <-ctx.Done():
  fmt.Println("Task 1 canceled")
  return ctx.Err()
 }
}

func task2(ctx context.Context) error {
 fmt.Println("Task 2 started successfully")
 select {
 case <-time.After(2 * time.Second):
  fmt.Println("Task 2 processed failed")
  return errors.New("Task 2 processed failed due to error")
 case <-ctx.Done():
  fmt.Println("Task 2 canceled")
  return ctx.Err()
 }
}

func task3(ctx context.Context) error {
 fmt.Println("Task 3 started successfully")
 select {
 case <-time.After(3 * time.Second):
  fmt.Println("Task 3 completed successfully")
  return nil
 case <-ctx.Done():
  fmt.Println("Task 3 canceled")
  return ctx.Err()
 }
}

func main() {
 ctx, cancel := context.WithCancel(context.Background())
 defer cancel() // Ensure cancellation happens when main() exits

 g, ctx := errgroup.WithContext(ctx)

 g.Go(func() error {
  return task1(ctx)
 })

 g.Go(func() error {
  return task2(ctx)
 })

 g.Go(func() error {
  return task3(ctx)
 })

 if err := g.Wait(); err != nil {
  fmt.Println("Encountered error:", err)
  cancel()
 } else {
  fmt.Println("All tasks completed successfully")
 }
}

上面程序的输出是:

Task 1 started successfully
Task 2 started successfully
Task 3 started successfully
Task 1 completed successfully
Task 2 processed failed
Encountered error: Task 2 processed failed due to error

在这个例子中:

  • 我们创建了一个带有可取消上下文的 errgroup
  • 定义了3个任务,task1 和 task3 在一定时间后模拟成功完成,而 task2 在更长的时间后通过返回错误来模拟失败。
  • 使用 g.Go 在单独的 goroutine 中启动每个任务。
  • 调用 g.Wait 等待所有任务完成。如果任何任务遇到错误,g.Wait() 会立即返回该错误。在主执行之后,task1 成功完成,task2 遇到了处理失败,而 task3 由于 task2 中的上述失败而被取消。

ErrGroup vs WaitGroup

ErrGroup:

  • 使用 ErrGroup 来管理并发任务中的错误。它聚合了所有协程中的错误,并返回遇到的第一个错误。
  • 需要管理多个可能产生错误的并发任务。
  • 想要利用上下文取消功能来优雅地关闭程序。
  • 不想手动检查多个 WaitGroup 调用的错误
  • 它与 Go 的上下文包无缝集成。任何来自协程的错误都会取消上下文,自动停止其他正在运行的任务。

WaitGroup

  • 使用 WaitGroup 进行基本同步。它简单地等待指定数量的 goroutine 完成后再继续。
  • ]当你只关心任务完成而不预期错误时,它是理想的选择。
  • 它不直接处理错误。你需要在每个 goroutine 中检查错误并单独处理它们。

总结

  • 使用 WaitGroup 进行简单的同步
  • 使用 ErrGroup 进行错误管理,并希望在优雅关闭时有上下文感知的取消。
  • 也可以同时使用它们,WaitGroup 用于基本同步,ErrGroup 用于一组任务中的详细错误处理。

相关的 issue[1]

参考资料
[1]

x/sync/errgroup: get all errors: https://github.com/golang/go/issues/23595

浏览 76
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报