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

Question

我今天一直在研究 Goroutines、Channels 和 WaitGroup，在阅读了一段时间之后，我终于开始理解这个概念了。我的问题是，我不确定在这样工作时如何处理错误，主要是因为我使用了 WaitGroup。使用 WaitGroup 时，我首先添加将要执行的 goroutine 数量，但如果其中之一发生错误怎么办？package mainimport (    "errors"    "sync")var waitGroup sync.WaitGroupfunc 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 main

import (

    "errors"

    "log"

    "sync"

    "golang.org/x/sync/errgroup"

)

func main() {

    c := make(chan int, 10)

    var g errgroup.Group

    g.Go(func() error {

        return doSomething(c)

    })

    // g.Wait waits for all goroutines to complete

    // and returns the first non-nil error returned

    // by one of the goroutines.

    if err := g.Wait(); err != nil {

        log.Fatal(err)

    }

}

func doSomething(c chan int) error {

    defer close(c)

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

        n, err := someFunctionThatCanError()

        if err != nil {

            return err

        }

        c <- n

    }

    return nil

}

func someFunctionThatCanError() (int, error) {

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

}

在操场上运行它。