在 go 并发函数中处理错误的方法有:使用 sync.errorgroup:它收集第一个发生的错误并返回。使用 channels:在 goroutine 之间通信错误,允许发送和接收错误消息。使用 recover:捕捉并发函数中的 panic 并返回引发 panic 的值。
如何处理 Go 并发函数中的错误
在并发程序中,处理错误至关重要。Go 提供了多种机制来处理并发函数中发生的错误。
使用 ErrorGroup
立即学习“go语言免费学习笔记(深入)”;
sync.ErrorGroup 是一个用于管理多个并发操作并收集其中任何一个产生的第一个错误的结构。函数 Add(f func()) error 向组中添加一个操作。每次调用该函数时,错误会被存储在组中。调用 Wait() 时,它将返回遇到的第一个错误(如果有)。
import (
"context"
"fmt"
"sync"
"sync/atomic"
)
func main() {
var eg sync.ErrGroup
var count int64
for i := 0; i < 10; i++ {
eg.Add(func() error {
atomic.AddInt64(&count, 1)
fmt.Printf("Count: %dn", count)
if count%2 == 0 {
return fmt.Errorf("even count: %d", count)
}
return nil
})
}
if err := eg.Wait(); err != nil {
fmt.Println(err)
}
}
使用 Channels
Channels 可用于在并发函数之间通信,包括错误。使用 channel,您可以将错误消息发送给一个指定 channel,然后从另一个 goroutine 中接收它们。
import (
"context"
"fmt"
"sync"
)
func main() {
ctx := context.Background()
errCh := make(chan error)
wg := sync.WaitGroup{}
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
if i%2 == 0 {
errCh <- fmt.Errorf("even number: %d", i)
return
}
errCh <- nil
}(i)
}
go func() {
wg.Wait()
close(errCh)
}()
for err := range errCh {
if err != nil {
fmt.Println(err)
}
}
}
使用 recover
使用 recover 语句可以 catch 发生在并发函数中的 panic。recover() 返回引发 panic 的值。
import (
"context"
"fmt"
"sync"
)
func main() {
ctx := context.Background()
errCh := make(chan error)
wg := sync.WaitGroup{}
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
if i%2 == 0 {
panic(fmt.Errorf("even number: %d", i))
}
errCh <- nil
}(i)
}
go func() {
wg.Wait()
close(errCh)
}()
for err := range errCh {
if err != nil {
fmt.Println(err)
}
}
// 在主 goroutine 中 recover 任何未处理的 panic
defer func() {
if r := recover(); r != nil {
fmt.Println("Recovered:", r)
}
}()
}