使用 WaitGroup 优雅地处理 Goroutine 中的错误并终止

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《使用 WaitGroup 优雅地处理 Goroutine 中的错误并终止》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

今天我一直在研究 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

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

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

非常感谢任何帮助。

解决方案

使用 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")
}

Run it on the playground。

今天关于《使用 WaitGroup 优雅地处理 Goroutine 中的错误并终止》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！