使用 errgroup 来结束被阻塞的 goroutine

珍惜时间，勤奋学习！今天给大家带来《使用 errgroup 来结束被阻塞的 goroutine》，正文内容主要涉及到等等，如果你正在学习Golang，或者是对Golang有疑问，欢迎大家关注我！后面我会持续更新相关内容的，希望都能帮到正在学习的大家！

我有两个任务正在 go 例程中运行。我正在使用 errgroup。我不确定如何正确使用 errgroup.withcontext。

在下面的代码中，任务1返回错误，我想在发生这种情况时终止任务2（长时间运行）。请注意，在此示例中添加 time.sleep 只是为了模拟我的问题。实际上，任务 1 和任务 2 正在做实际工作，并且没有任何睡眠调用。

package main

import (
    "context"
    "fmt"
    "golang.org/x/sync/errgroup"
    "time"
)

func task1(ctx context.Context) error {
    time.Sleep(5 * time.Second)
    fmt.Println("first finished, pretend error happened")
    return ctx.Err()
}

func task2(ctx context.Context) error {
    select {
    case <-ctx.Done():
        fmt.Println("task 1 is finished with error")
        return ctx.Err()
    default:
        fmt.Println("second started")
        time.Sleep(50 * time.Second)
        fmt.Println("second finished")
    }
    return nil
}

func test() (err error) {
    ctx := context.Background()
    g, gctx := errgroup.WithContext(ctx)

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

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

    err = g.Wait()
    if err != nil {
        fmt.Println("wait done")
    }

    return err
}

func main() {
    fmt.Println("main")
    err := test()
    if err != nil {
        fmt.Println("main err")
        fmt.Println(err.Error())
    }
}

正确答案

正确处理 context 取消取决于您的任务，而不是选择内的 time.sleep。

如 errgroup 文档中所述：

withcontext 返回一个新的 group 和一个从 ctx 派生的关联上下文。

当传递给 go 的函数第一次返回非零错误或第一次 wait 返回时（以先发生者为准），派生的 context 将被取消。

您正在正确使用错误组，但您的上下文处理需要重构。

这是任务 2 的重构：

func task2(ctx context.Context) error {
    errCh := make(chan bool)

    go func() {
        time.Sleep(50 * time.Second)
        errCh <- true
    }()

    select {
    case <-ctx.Done():
        return fmt.Errorf("context done: %w", ctx.Err())
    case <-errCh:
        return errors.New("task 2 failed")
    }
}

通过这样的选择，您可以等待第一个通道发出。在这种情况下，就是上下文过期，除非你将 time sleep 修改得更低。示例playground。

理论要掌握，实操不能落！以上关于《使用 errgroup 来结束被阻塞的 goroutine》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！