卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34869本站已运行393

为什么我只收到部分错误,而不是我启动的 goroutine 中的所有错误?

为什么我只收到部分错误,而不是我启动的 goroutine 中的所有错误?

php小编苹果为你解答:在Go语言中,当一个goroutine发生错误时,并不会自动传播给主协程。相反,它会被静默地忽略掉,这可能会导致你只收到部分错误而不是启动的所有goroutine中的错误。这是因为Go语言设计的初衷是让程序保持稳定和高效,即使在出现错误的情况下也不会立即停止整个程序。如果你想要捕获所有的错误,你可以使用通道或者其他机制来显示地传递错误信息。这样你就可以确保所有的错误都被正确处理。

问题内容

我定义了一个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)输出err1err2,但我希望它输出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
}
卓越飞翔博客
上一篇: 在 Go 中解组 sql.NullTime 结构
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏