go - 并行获取多个字段的模式
问题描述
我需要从外部服务中为我的系统并行获取多个字段(在此示例中,由 Name()、Age() 和 CanDrive() 方法模拟)。
fetchUser() 方法可以满足我的要求,但是如果您认为我可以拥有 10 多个字段,它似乎太冗长了。有没有更好的方法可以实现这一点?
操场: https: //play.golang.org/p/90sNq1GmrD8
代码(与操场相同):
package main
import (
"fmt"
"sync"
)
type User struct {
Name string
Age int
CanDrive *bool
}
func Name() (string, error) {
return "foobar", nil
}
func Age() (int, error) {
return 25, nil
}
func CanDrive() (bool, error) {
return true, nil
}
func fetchUser() (*User, error) {
var wg sync.WaitGroup
errs := make(chan error)
user := &User{}
wg.Add(1)
go func() {
var err error
defer wg.Done()
user.Name, err = Name()
errs <- err
}()
wg.Add(1)
go func() {
var err error
defer wg.Done()
user.Age, err = Age()
errs <- err
}()
wg.Add(1)
go func() {
defer wg.Done()
canDrive, err := CanDrive()
if err == nil {
user.CanDrive = &canDrive
}
errs <- err
}()
// wait until all go-routines are completed successfully
// if that's the case, close the errs channel
go func() {
wg.Wait()
close(errs)
}()
// keep waiting for errors (or for the error channel to be closed
// if all calls succeed)
for err := range errs {
if err != nil {
return nil, err
}
}
return user, nil
}
func main() {
user, _ := fetchUser()
fmt.Println(user)
}
解决方案
在不了解您的场景的更多细节的情况下,我唯一的建议是将 Go 例程错误处理分离到另一个包中。
幸运的是,已经有一个包可以做同样的事情,名为errgroup
. 以下是使用该errgroup
包的原始代码的实现:
package main
import (
"context"
"fmt"
"golang.org/x/sync/errgroup"
)
type User struct {
Name string
Age int
CanDrive *bool
}
func Name() (string, error) {
return "foobar", nil
}
func Age() (int, error) {
return 25, nil
}
func CanDrive() (bool, error) {
return true, nil
}
func fetchUser(ctx context.Context) (*User, error) {
group, ctx := errgroup.WithContext(ctx)
user := &User{}
group.Go(func() (err error) {
user.Name, err = Name()
return
})
group.Go(func() (err error) {
user.Age, err = Age()
return
})
group.Go(func() error {
canDrive, err := CanDrive()
if err == nil {
user.CanDrive = &canDrive
}
return err
})
if err := group.Wait(); err != nil {
return nil, err
}
return user, nil
}
func main() {
user, err := fetchUser(context.Background())
fmt.Println(user, err)
}
推荐阅读
- javascript - VUE:视图变化时的过渡元素
- sql-server - SQL Server 不存在或访问被拒绝,但仅适用于某些用户
- swiftui - 如何在 SwiftUI 中使 UIViewRepresentable 在 tvOS 上具有焦点?
- azure - Java 中用于 Spring 启动的 Azure Blob Storage Starter API 出错
- python - 一个集合是否转换为 numpy 数组?
- c - Bison Flex 减少/减少悬挂 else 与中间动作的冲突
- ios - UIWebView 处理链接
- r - 有没有办法使用 r 中的 map 或 lapply 命令传递时间序列数据的子集?
- r - 有没有办法在 R 的 tis 包 v1.38 中更新 FederalHoliday 函数
- itext7 - 如何获取用 IText 7 编写的行数