多字段同时获取的方式

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《多字段同时获取的方式》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

我需要从外部服务为我的系统并行获取多个字段（在本例中，通过 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)
}

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