Golang 的 ErrGroup 源码解读：协程并发中的优雅错误处理

后端

2023-10-05 22:59:18

前言

在并发编程的世界中，处理错误至关重要。在 Golang 中，ErrGroup 是一个强大的工具，它简化了并发任务中的错误处理，确保了程序的健壮性和可靠性。本文将带你深入探索 ErrGroup 的源码，了解它如何实现其优雅的错误处理机制。

ErrGroup 的内部机制

ErrGroup 本质上是一个 WaitGroup，它跟踪正在运行的协程的数量。此外，它还维护一个错误指针，该指针在任何协程返回非 nil 错误时被更新。

ErrGroup 的关键方法是 Wait()，它会阻塞直到所有协程完成。在 Wait() 过程中，如果检测到错误，则 Wait() 将返回该错误，并在内部调用 Wait 的所有协程也将立即返回该错误。

源码分析

ErrGroup 的源码位于 runtime/error.go 文件中。让我们仔细看看其关键函数的实现：

func (eg *ErrGroup) Wait()：阻塞直到所有协程完成。如果检测到错误，则立即返回该错误。
func (eg *ErrGroup) Go(f func()): 启动一个协程并将其添加到 ErrGroup 中。
func (eg *ErrGroup) Err()：返回 ErrGroup 中发生的第一个错误，如果不存在错误则返回 nil。

使用示例

以下是使用 ErrGroup 处理并发任务错误的一个示例：

import (
    "context"
    "fmt"
    "sync"
)

func main() {
    var eg sync.ErrGroup

    for i := 0; i < 5; i++ {
        eg.Go(func() {
            // 模拟并发任务
            if err := doSomething(i); err != nil {
                eg.Go(func() {
                    // 处理错误
                    fmt.Println(err)
                })
            }
        })
    }

    if err := eg.Wait(); err != nil {
        // 处理所有并发任务中发生的第一个错误
        fmt.Println(err)
    }
}

func doSomething(i int) error {
    if i % 2 == 0 {
        return fmt.Errorf("error for task %d", i)
    }

    return nil
}