当前位置： > > > > 将配置结构加载到 go 中后，访问嵌套属性时出现问题

将配置结构加载到 go 中后，访问嵌套属性时出现问题

来源：stackoverflow
2024-04-19 20:18:34
0浏览
收藏

本篇文章给大家分享《将配置结构加载到 go 中后，访问嵌套属性时出现问题》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

问题内容

所以我想将 config.yaml 文件加载到我的程序中。

defaulttimeout: 0.25
defaultworkspace: ibm
workspaces:
  ibm:
    client: 47943938
    defaultgoal: 27.5
    projects:
      - 159929907
      - 160046030
      - 160064503
  pp:
    defaultgoal: 5.5
    project: 160064503

因此我创建以下结构：

type config struct {
    defaulttimeout   float64 `yaml:"defaulttimeout"`
    defaultworkspace string  `yaml:"defaultworkspace"`
    workspaces       struct {
        ibm struct {
            client      int     `yaml:client`
            defaultgoal float64 `yaml:"defaultgoal"`
            projects    []int   `yaml:"projects"`
        }
        pp  struct {
            defaultgoal float64 `yaml:"defaultgoal"`
            project     int     `yaml:project`
        }
    }
}

基本上，我列出了不同的工作区，其中包含不同的属性和一个共同属性：defaultgoal。现在我想以一种干净有效的方式访问这个属性。

现在我有了参数 *workspacename 来决定我要选择哪个工作空间。所以问题是我如何使用它作为我想要选择的 workspace 的键，然后进一步获取相应条目的 defaultgoal ？

我尝试了以下操作（config 是从 yaml 源加载的结构）

wsConf := reflect.Indirect(reflect.ValueOf(config.Workspaces)).FieldByName(*workspaceName)

但是我有 reflect.value ，我必须将其转换回一个结构体来评估 defaultgoal 。

这里有什么好的方法吗？如有任何帮助，我们将不胜感激！

解决方案

与使用反射（reflect 包）相比，使用 map[string]interface{} 作为字段类型（特别是 workspaces 字段），然后将此地图中的项目视为特定工作区可能会更容易。

type config struct {
    defaulttimeout   float64 `yaml:"defaulttimeout"`
    defaultworkspace string  `yaml:"defaultworkspace"`
    workspaces       map[string]interface{} `yaml: workspaces`
}

type ibm struct {
            client      int     `yaml:client`
            defaultgoal float64 `yaml:"defaultgoal"`
            projects    []int   `yaml:"projects"`
}
type pp struct {
            defaultgoal int `yaml:"defaultgoal"`
            project     int `yaml:project`
}

func main() {
    var cfg config
    // init cfg somehow
    for space, value := range cfg.workspaces {
        switch space {
        case "ibm": // convert `value` to ibm struct
        case "pp": // convert `value` to pp struct
    }
}

如果您想根据 workspace 名称解组为特定结构，可以尝试此方法。 （使用json时一般不需要额外的marshal步骤。更多）。工作示例

package main

import (
    "log"

    "gopkg.in/yaml.v2"
)

type Config struct {
    DefaultTimeout   float64 `yaml:"defaultTimeout"`
    DefaultWorkspace string  `yaml:"defaultWorkspace"`
    Workspaces       map[string]interface{}
}

type PP struct {
    DefaultGoal int `yaml:"defaultGoal"`
    Project     int `yaml:"project"`
}

type IBM struct {
    Client      int     `yaml:"client"`
    DefaultGoal float64 `yaml:"defaultGoal"`
    Projects    []int   `yaml:"projects"`
}

func main() {
    yamlData := `defaultTimeout: 0.25
defaultWorkspace: ibm
workspaces:
ibm:
    client: 47943938
    defaultGoal: 27.5
    projects:
    - 159929907
    - 160046030
    - 160064503
pp:
    defaultGoal: 5.5
    project: 160064503`

    var c Config
    if err := yaml.Unmarshal([]byte(yamlData), &c); err != nil {
        panic(err)
    }

    for space, value := range c.Workspaces {
        bs, err := yaml.Marshal(value)
        if err != nil {
            panic(err)
        }

        switch space {
        case "ibm":
            var i IBM
            if err := yaml.Unmarshal(bs, &i); err != nil {
                panic(err)
            }

            log.Printf("IBM value: %v", i)

        case "pp":
            var p PP
            if err := yaml.Unmarshal(bs, &p); err != nil {
                panic(err)
            }

            log.Printf("PP value: %v", p)
        }
    }
}

到这里，我们也就讲完了《将配置结构加载到 go 中后，访问嵌套属性时出现问题》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注米云公众号，带你了解更多关于的知识点！