如何使用 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

我已經盡力提供一個例子來說明我在說什么。 一個循環將在doSomething()中運行 10 次，它會在每次迭代時調用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")
}

在操場上運行它。

Answer 2

我已經修改了您的示例，我不知道它是否是一個干凈的解決方案，但它避免了死鎖等。

func doSomething(c chan int) {
    defer close(c) // ensure channel is closed
    for i := 0; i < 10; i++ {
        n, err := someFunctionThatCanError(i)

        if err != nil {
            // calculate remaining waits and quickly release them
            for j := i; j < 10; j++ {
                waitGroup.Done()

            }
            return
        }

        c <- n
        waitGroup.Done()
    }

}

func someFunctionThatCanError(i int) (int, error) {
    fmt.Println("running", i)
    if i > 3 {
        return 1, errors.New("an error")
    }
    return 1, nil
}

這里是游樂場