errgroup 是 Go 语言的一个第三方库，通常用于处理并发操作，并允许在多个 goroutine 中跟踪错误。它的设计目的是让你能够并行运行多个任务，并在一个任务失败时提前取消其他任务，同时收集和返回第一个错误。
errgroup 是 golang.org/x/sync/errgroup 包中的一个工具，它使得错误处理变得更加简洁，并且可以让你在并发操作中处理错误时不必手动管理 goroutines 和同步。
首先，你需要安装 errgroup 库：
Life is like a journey
go get golang.org/x/sync/errgroup
errgroup 会收集所有 goroutine 中返回的错误。errgroup 会停止等待其他 goroutines 完成，并且可以提前返回该错误。errgroup 主要功能errgroup 会返回该错误，其他 goroutines 将会被取消（通过上下文取消）。errgroup 会收集每个任务的错误，并在遇到第一个错误时就停止等待。errgroup 使用方式errgroup 的核心是 errgroup.Group 类型，提供了 Go 方法和 Wait 方法。
errgroup 添加一个 goroutine，执行一个函数。package main
import (
"fmt"
"golang.org/x/sync/errgroup"
"net/http"
"time"
)
func fetch(url string) error {
// 模拟 HTTP 请求
time.Sleep(2 * time.Second)
if url == "http://example.com" {
return fmt.Errorf("error fetching %s", url)
}
fmt.Println("Fetched:", url)
return nil
}
func main() {
var g errgroup.Group
// 添加多个 goroutine 执行 fetch 函数
urls := []string{"http://google.com", "http://example.com", "http://golang.org"}
for _, url := range urls {
url := url // 避免闭包问题
g.Go(func() error {
return fetch(url)
})
}
// 等待所有 goroutine 完成
if err := g.Wait(); err != nil {
fmt.Println("Error:", err)
} else {
fmt.Println("All URLs fetched successfully.")
}
}
在这个例子中，我们启动了多个 goroutines 来模拟 HTTP 请求，每个请求可能返回一个错误。g.Go() 向 errgroup 添加一个新的任务，g.Wait() 等待所有任务完成并检查是否有错误。
Fetched: http://google.com
Fetched: http://golang.org
Error: error fetching http://example.com
fetch 函数中的 url 是 http://example.com，它会返回一个错误，errgroup 会立即停止等待其他 goroutines 完成，直接返回该错误。errgroup 还支持通过上下文取消机制来优雅地终止 goroutines。通过传递一个带有取消功能的 context，可以在发生错误时中止其他任务的执行。
package main
import (
"context"
"fmt"
"golang.org/x/sync/errgroup"
"net/http"
"time"
)
func fetch(ctx context.Context, url string) error {
// 模拟 HTTP 请求
select {
case <-time.After(2 * time.Second):
// 模拟网络延迟
if url == "http://example.com" {
return fmt.Errorf("error fetching %s", url)
}
fmt.Println("Fetched:", url)
return nil
case <-ctx.Done():
// 如果上下文被取消，提前返回
return ctx.Err()
}
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
var g errgroup.Group
// 添加多个 goroutine 执行 fetch 函数
urls := []string{"http://google.com", "http://example.com", "http://golang.org"}
for _, url := range urls {
url := url // 避免闭包问题
g.Go(func() error {
return fetch(ctx, url)
})
}
// 等待所有 goroutine 完成
if err := g.Wait(); err != nil {
fmt.Println("Error:", err)
} else {
fmt.Println("All URLs fetched successfully.")
}
}
Fetched: http://google.com
Fetched: http://golang.org
Error: error fetching http://example.com
在这个例子中，一旦遇到错误，context.Cancel() 会被调用，取消所有其他正在运行的 goroutines。
g.Go 启动任务g.Go 方法是 errgroup 的关键方法之一。它允许你将一个函数添加到 errgroup 中。每个函数都将在独立的 goroutine 中运行。如果该函数返回错误，errgroup 会立即停止其他任务并返回该错误。
g.Go(func() error {
// Task code that might return an error
})
g.Wait() 等待所有任务完成g.Wait() 会等待所有添加到 errgroup 中的 goroutines 执行完毕，并返回第一个错误。如果没有错误，返回 nil。
if err := g.Wait(); err != nil {
fmt.Println("Error:", err)
}
errgroup 会在第一个 goroutine 返回错误时立即停止其他 goroutines。Wait 方法会返回第一个发生的错误。如果没有错误，它会返回 nil。
errgroup 是 Go 中一个处理并发任务错误的工具，简化了错误的传播和 goroutine 的管理。Go 和 Wait 方法，可以用来并发执行多个任务，并收集第一个错误。errgroup 会立即停止执行其他任务，并返回该错误。context 结合使用，支持上下文取消机制，优雅地终止并发任务。errgroup 在处理并发任务时，尤其是需要捕获和传播错误时非常有用，特别是在需要等待多个任务执行完毕并检查错误时，它可以帮助简化代码的实现。