如何使用 WaitGroup 处理错误并终止 Goroutine

Question

我今天一直在研究 Goroutines、Channels 和 WaitGroup，在阅读了一段时间之后，我终于开始理解这个概念了。

我的问题是，我不确定在这样工作时如何处理错误，主要是因为我使用了 WaitGroup。使用 WaitGroup 时，我首先添加将要执行的 goroutine 数量，但如果其中之一发生错误怎么办？

package main

import (

    "errors"

    "sync"

)

var waitGroup sync.WaitGroup

func main() {

    c := make(chan int, 10)

    waitGroup.Add(10)

    go doSomething(c)

    waitGroup.Wait()

}

func doSomething(c chan int) {

    for i := 0; i < 10; i++ {

        n, err := someFunctionThatCanError()

        if err != nil {

            // How do I end the routines and WaitGroups here?

        }

        c <- n

        waitGroup.Done()

    }

    close(c)    

}

func someFunctionThatCanError() (int, error) {

    return 1, errors.New("an error")

}

游乐场：https://play.golang.org/p/ZLsBSqdMD49

我已尽力提供一个例子来说明我正在谈论的内容。一个循环将运行 10 次，doSomething()并且每次迭代都会调用waitGroup.Done()，但是如果在这一切过程中发生错误怎么办，如 所示someFunctionThatCanError()？

当我现在尝试通过返回和/或取消频道来解决它时，我最终陷入了僵局，所以我有点不确定从这里该去哪里。我也不确定如何处理我认为正在等待更多事情发生的 WaitGroup。

非常感谢任何帮助。

Answer 1

使用golang.org/x/sync/errgroup等待并处理来自 goroutine 的错误。package mainimport (&nbsp; &nbsp; "errors"&nbsp; &nbsp; "log"&nbsp; &nbsp; "sync"&nbsp; &nbsp; "golang.org/x/sync/errgroup")func main() {&nbsp; &nbsp; c := make(chan int, 10)&nbsp; &nbsp; var g errgroup.Group&nbsp; &nbsp; g.Go(func() error {&nbsp; &nbsp; &nbsp; &nbsp; return doSomething(c)&nbsp; &nbsp; })&nbsp; &nbsp; // g.Wait waits for all goroutines to complete&nbsp; &nbsp; // and returns the first non-nil error returned&nbsp; &nbsp; // by one of the goroutines.&nbsp; &nbsp; if err := g.Wait(); err != nil {&nbsp; &nbsp; &nbsp; &nbsp; log.Fatal(err)&nbsp; &nbsp; }}func doSomething(c chan int) error {&nbsp; &nbsp; defer close(c)&nbsp; &nbsp; for i := 0; i < 10; i++ {&nbsp; &nbsp; &nbsp; &nbsp; n, err := someFunctionThatCanError()&nbsp; &nbsp; &nbsp; &nbsp; if err != nil {&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; return err&nbsp; &nbsp; &nbsp; &nbsp; }&nbsp; &nbsp; &nbsp; &nbsp; c <- n&nbsp; &nbsp; }&nbsp; &nbsp; return nil}func someFunctionThatCanError() (int, error) {&nbsp; &nbsp; return 1, errors.New("an error")}在操场上运行它。