Golang 函数：用 errgroup 实现批量取消-Golang-PHP中文网

Golang 函数：用 errgroup 实现批量取消

PHPz

发布： 2024-10-06 20:42:02

原创

774人浏览过

go 语言提供 errgroup 函数，用于管理并发函数组，实现批量取消操作。它的优点包括：等待所有 goroutine 完成或出现第一个错误。在收到取消信号时取消所有 goroutine。在 goroutine 遇到错误时取消其他 goroutine。

Golang 函数：用 errgroup 实现批量取消

Go 语言函数：使用 errgroup 实现批量取消

引言
在大型并发应用程序中，协调多个并发操作并优雅地处理错误至关重要。Go 语言中，errgroup 函数提供了一种简单、健壮的方法来管理并发函数组，并确保在出现错误时安全取消所有操作。

errgroup 简介
errgroup 是一个并发模式，它允许你将多个 goroutine 组合成一个组，并等待所有 goroutine 完成或出现第一个错误。它的主要优点在于可以通过以下方式实现批量取消操作：

立即学习“go语言免费学习笔记（深入）”；

图可丽批量抠图

用AI技术提高数据生产力，让美好事物更容易被发现

查看详情

等待所有 goroutine 完成，或遇到第一个错误
在收到取消信号时取消所有 goroutine
在 goroutine 遇到错误时取消其他 goroutine

实战案例

假设我们有一个需要并发处理大量文件的任务。以下是使用 errgroup 实现批量取消的代码示例：

package main

import (
    "context"
    "fmt"
    "sync/errgroup"
)

func processFile(ctx context.Context, file string) error {
    // 处理文件
    fmt.Printf("Processing file %s\n", file)
    return nil // 此处根据需要更改
}

func main() {
    // 创建一个 context 用于传递取消信号
    ctx := context.Background()
    ctx, cancel := context.WithCancel(ctx)

    // 创建 errgroup
    var g errgroup.Group

    // 将文件处理函数添加到 errgroup 中
    for _, file := range []string{"file1", "file2", "file3"} {
        g.Go(func() error {
            return processFile(ctx, file)
        })
    }

    // 等待 errgroup 结束，或收到取消信号
    if err := g.Wait(); err != nil {
        fmt.Println("Error:", err)
        // 发出取消信号，取消所有正在运行的 goroutine
        cancel()
        return
    }

    fmt.Println("All files processed successfully")
}

登录后复制

在这个例子中，我们创建了一个 errgroup 并将三个文件处理函数添加到其中。我们使用 context.WithCancel 创建了一个带有取消功能的 context，并将其传递给 processFile 函数。

如果在处理任何文件时发生错误，errgroup 就会被取消，所有正在运行的 goroutine 都会被终止。或者，如果收到取消信号，例如来自用户输入或超时，我们也会主动取消 errgroup。

以上就是Golang 函数：用 errgroup 实现批量取消的详细内容，更多请关注php中文网其它相关文章！