为什么我只接收到一部分启动的 goroutine 中的错误，而不是所有的错误？

有志者，事竟成！如果你在学习Golang，那么本文《为什么我只接收到一部分启动的 goroutine 中的错误，而不是所有的错误？》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我定义了一个cycle类来处理并发任务。我想要的是运行两个函数，每个函数都在一个 goroutine 中，等待它们完成并将它们的输出错误合并在一起。但我只收到一个错误。每个方法的职责如下：

run-在goroutine中运行函数，并收集其错误

waitalldone-将所有函数错误合并在一起并等待所有函数完成

do1、do2 - 测试函数

import (
    "fmt"
    "go.uber.org/multierr"
    "sync"
    "testing"
)

type Cycle struct {
    errChan chan error
    wg sync.WaitGroup
}

func NewCycle() *Cycle {
    return &Cycle{
        errChan: make(chan error),
        wg:      sync.WaitGroup{},
    }
}

// run fn and collect its error into error channel
func (c *Cycle) Run(fn func() error) {
    c.wg.Add(1)
    go func() {
        defer c.wg.Done()
        if err := fn(); err != nil {
            c.errChan <- err
        }
    }()
}

// wait all fn finish and combine their error together
func (c *Cycle) WaitAllDone() error {
    var err error
    go func() {
        for {
            if tmpErr, ok := <-c.errChan; ok {
                err = multierr.Append(err, tmpErr)
            } else{
                break
            }
        }
    }()
    c.wg.Wait()
    close(c.errChan)
    return err
}

func Do1() error {
    return fmt.Errorf("ERR1")
}

func Do2() error {
    return fmt.Errorf("ERR2")
}

func Test41(t *testing.T) {
    c := NewCycle()
    c.Run(Do1)
    c.Run(Do2)
    if err := c.WaitAllDone(); err != nil {
        t.Log(err)
    }
}

最终t.log(err)输出err1或err2，但我希望它输出err1 err2。为什么它会漏掉一个错误。

正确答案

这是因为 (*cycle).waitalldone 不会等待收集错误的 goroutine 完成。如果您使用 -race 标志运行代码，有时它可能会报告几个 data race 错误。这是其中之一：

$ go test -race .
==================
warning: data race
write at 0x00c0000a0610 by goroutine 10:
  m.(*cycle).waitalldone.func1()
      /home/zeke/src/temp/76370962/main_test.go:40 +0xb6

previous read at 0x00c0000a0610 by goroutine 7:
  m.(*cycle).waitalldone()
      /home/zeke/src/temp/76370962/main_test.go:48 +0x14e
  m.test41()
      /home/zeke/src/temp/76370962/main_test.go:63 +0xa4
  testing.trunner()
      /snap/go/current/src/testing/testing.go:1576 +0x216
  testing.(*t).run.func1()
      /snap/go/current/src/testing/testing.go:1629 +0x47

此更改将解决该问题：

func (c *cycle) waitalldone() error {
    var err error
+   done := make(chan int)
    go func() {
        for {
            if tmperr, ok := <-c.errchan; ok {
                err = multierr.append(err, tmperr)
            } else {
                break
            }
        }
+       close(done)
    }()
    c.wg.wait()
    close(c.errchan)
+   <-done
    return err
  }

并且可以使用 range 子句来简化 for 循环：

func (c *Cycle) WaitAllDone() error {
    var err error
    done := make(chan int)
    go func() {
        for tmpErr := range c.errChan {
            err = multierr.Append(err, tmpErr)
        }
        close(done)
    }()
    c.wg.Wait()
    close(c.errChan)
    <-done
    return err
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。