大家可能会想到 Go 的经典哲学:不要通过共享内存来通信,而是通过通信来实现内存共享(Do not communicate by sharing memory; instead, share memory by communicating)。
第二种方法:利用 channel 来传输多个 goroutine 中的 errors:
func main() {
cherrors := make(chan error)
wgDone := make(chan bool)
var wg sync.WaitGroup
wg.Add(2)
go func() {
//... 业务逻辑
wg.Done()
}()
go func() {
//... 业务逻辑
err := returnErr()
if err != nil {
cherrors <- err
}
wg.Done()
}()
go func() {
wg.Wait()
close(wgDone)
}()
select {
case <-wgDone:
break
case err := <-cherrors:
close(cherrors)
fmt.Println(err)
}
time.Sleep(time.Second)
}
func returnErr() error {
return errors.New("出错啦。。我是错误信息")
}
虽然使用 channel 后已经方便了不少,但编写 channel 还要关心一些非业务向的逻辑。
使用 sync/errgroup
第三种方法,就是使用官方提供的 golang.org/x/sync/errgroup 标准库:
type Group
func WithContext(ctx context.Context) (*Group, context.Context)
func (g *Group) Go(f func() error)
func (g *Group) Wait() error
Go:启动一个协程,在新的 goroutine 中调用给定的函数。
Wait:等待协程结束,直到 Go 方法中的所有函数调用都返回,然后返回其中第一个非零错误(如果有错误的话)。
结合其特性能够非常便捷的针对多 goroutine 进行错误处理:
func main() {
group := new(errgroup.Group)
nums := []int{-1, 0, 1}
for _, num := range nums {
num := num
group.Go(func() error {
res, err := output(num)
fmt.Println(res)
return err
})
}
if err := group.Wait(); err != nil {
fmt.Println("Get errors: ", err)
} else {
fmt.Println("Get all num successfully!")
}
}
func output(num int) (int, error) {
if num < 0 {
return 0, errors.New("math: square root error!")
}
return num, nil
}