errgroup包通过结合context实现并发任务的错误管理和协同取消,其核心是WithCancel创建的上下文在任一任务出错时自动取消,使其他任务及时退出,从而高效控制并发生命周期。

Golang的
errgroup
context
errgroup
errgroup.Group
基本用法通常涉及以下步骤:
errgroup.Group
g, ctx := errgroup.WithContext(parentCtx)
Group
parentCtx
context
context
Group
Wait
Go
g.Go(func() error { ... })error
g.Wait()
Go
Wait
WithContext
ctx
Wait
nil
package main
import (
"context"
"errors"
"fmt"
"sync/atomic"
"time"
"golang.org/x/sync/errgroup"
)
func main() {
// 创建一个带有取消功能的上下文
parentCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel() // 确保在main函数结束时取消上下文
g, ctx := errgroup.WithContext(parentCtx)
var successCount atomic.Int32
var failCount atomic.Int32
// 任务1:模拟成功
g.Go(func() error {
select {
case <-time.After(1 * time.Second):
fmt.Println("任务1完成:成功")
successCount.Add(1)
return nil
case <-ctx.Done():
fmt.Println("任务1被取消")
return ctx.Err()
}
})
// 任务2:模拟失败,且失败较快
g.Go(func() error {
select {
case <-time.After(500 * time.Millisecond):
fmt.Println("任务2完成:失败")
failCount.Add(1)
return errors.New("任务2出错了")
case <-ctx.Done():
fmt.Println("任务2被取消")
return ctx.Err()
}
})
// 任务3:模拟一个耗时较长的任务,可能会被取消
g.Go(func() error {
select {
case <-time.After(3 * time.Second):
fmt.Println("任务3完成:成功")
successCount.Add(1)
return nil
case <-ctx.Done():
fmt.Println("任务3被取消")
return ctx.Err()
}
})
// 等待所有任务完成或第一个错误发生
if err := g.Wait(); err != nil {
fmt.Printf("errgroup.Wait() 返回错误: %v\n", err)
} else {
fmt.Println("所有任务成功完成")
}
fmt.Printf("成功任务数: %d, 失败任务数: %d\n", successCount.Load(), failCount.Load())
}在这个例子中,任务2会先于其他任务失败。当任务2返回错误时,
errgroup
ctx
g.Wait()
ctx.Done()
立即学习“go语言免费学习笔记(深入)”;
当我们处理多个并发执行的goroutine时,错误处理和任务取消往往是令人头疼的问题。传统上,我们可能会使用
sync.WaitGroup
chan error
sync.Once
sync.WaitGroup
errgroup
sync.WaitGroup
context
errgroup
errgroup
context
ctx.Done()
errgroup.WithContext()
errgroup
errgroup
当我们调用
g, ctx := errgroup.WithContext(parentCtx)
errgroup
context.Context
ctx
parentCtx
g.Wait()
ctx
cancel
g.Go()
nil
errgroup
ctx
<-ctx.Done()
ctx.Err()
这种机制在实际应用中非常有用。想象一下,你有一个服务需要同时从多个微服务获取数据,然后聚合结果。如果其中一个微服务响应超时或者返回了错误,那么继续等待其他微服务的结果可能就没有意义了。
errgroup.WithContext()
例如,在处理HTTP请求时,你可能需要并行地调用多个下游服务。如果一个下游服务失败了,你可以通过
errgroup.WithContext()
package main
import (
"context"
"errors"
"fmt"
"time"
"golang.org/x/sync/errgroup"
)
func fetchUserData(ctx context.Context, userID string) (string, error) {
select {
case <-time.After(1 * time.Second): // 模拟网络延迟
if userID == "error_user" {
return "", errors.New("用户数据获取失败")
}
return fmt.Sprintf("User data for %s", userID), nil
case <-ctx.Done():
fmt.Printf("fetchUserData for %s cancelled\n", userID)
return "", ctx.Err()
}
}
func fetchOrderHistory(ctx context.Context, userID string) (string, error) {
select {
case <-time.After(2 * time.Second): // 模拟更长的网络延迟
return fmt.Sprintf("Order history for %s", userID), nil
case <-ctx.Done():
fmt.Printf("fetchOrderHistory for %s cancelled\n", userID)
return "", ctx.Err()
}
}
func main() {
parentCtx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
g, ctx := errgroup.WithContext(parentCtx)
var userData string
var orderHistory string
// 任务1:获取用户数据
g.Go(func() error {
data, err := fetchUserData(ctx, "error_user") // 模拟一个会失败的用户
if err != nil {
return err
}
userData = data
return nil
})
// 任务2:获取订单历史
g.Go(func() error {
history, err := fetchOrderHistory(ctx, "some_user")
if err != nil {
return err
}
orderHistory = history
return nil
})
if err := g.Wait(); err != nil {
fmt.Printf("处理请求时发生错误: %v\n", err)
} else {
fmt.Printf("成功获取数据: 用户数据: %s, 订单历史: %s\n", userData, orderHistory)
}
// 预期输出:fetchOrderHistory for some_user cancelled, 处理请求时发生错误: 用户数据获取失败
}在这个例子中,
fetchUserData
error_user
ctx
fetchOrderHistory
虽然
errgroup
常见的陷阱:
g.Wait()
g.Wait()
errgroup
WithContext
ctx
Wait
cancel
g.Wait()
g.Wait()
errgroup
errgroup
errgroup
errgroup
sync.Mutex
errgroup
g.Go()
errgroup
g.Go()
errgroup
g.Wait()
errgroup
context
g.Go()
errgroup
ctx
ctx.Done()
ctx
最佳实践:
g.Wait()
errgroup
errgroup
sync.WaitGroup
ctx.Done()
g.Go()
<-ctx.Done()
g.Go()
fmt.Errorf
%w
g.Wait()
errors.Is
errors.As
g.Go()
g.Go
errgroup
errgroup
g.Go
context
遵循这些实践,
errgroup
以上就是Golang的errgroup包如何帮助管理一组goroutine的错误的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号