我正在尝试将以下 yaml 解组为以下结构，如何将 List 字段转换为 map 字段并使用结构访问它？

Question

YAML 文件：

namespaces:
  - namespace: default
    aliasname: k8s
    components:
      - component: comp1
        replicas: 1
        port: 8080
      - component: comp2
        replicas: 1
        port: 9999
  - namespace: ns2
    components:
      - component: comp1
        replicas: 1

从上面的 YAML 文件中，我想创建如下结构：

type Namespaces struct {
    NamespaceName    string       `yaml:"namespace"`
    Aliasname     string       `yaml:"aliasname,omitempty"`
    ListOfComponents []Components `yaml:"components"`
    ComponentMap     map[string]Components

}

type Components struct {
    ComponentName string `yaml:"component"`
    NumReplicas   int    `yaml:"replicas"`
    Port          int    `yaml:"port"`
}

type Config struct {
    ListOfNamespaces []Namespaces `yaml:"namespaces"`
    NamespaceMap map[string]Namespaces
}

当分别访问config 和namespace 对象时，应该能够检索到字段Namespacemap 和Componentmap。我创建了一个将命名空间和组件列表转换为映射的方法，但是当我调用config.Namespacemap 或Namespace.ComponentMap 时，它返回一个空映射。

基本上我想知道：我们如何为类型结构添加额外的字段？我想从config 结构访问新变量，例如地图。

更新： 感谢 blami 指导我，但是当我尝试为组件编写相同的内容时，它并没有给我包含组件映射的整个命名空间映射：

type Components struct {
    ComponentName string `yaml:"component"`
    NumReplicas   int    `yaml:"replicas"`
    Port          int    `yaml:"port"`

}

type Namespaces struct {
    NamespaceName string                `yaml:"namespace"`
    Aliasname  string                `yaml:"aliasname"`
    ComponentMap  map[string]Components `yaml:"components"`
}

func (n *Namespaces) UnmarshalYAML(unmarshal func(interface{}) error) error {

    type origNamespace struct {
        ListOfComponents []Components `yaml:"components"`
    }

    var on origNamespace
    err1 := unmarshal(&on)
    if err1 != nil {
        return err1
    }
    n.ComponentMap = make(map[string]Components)
    for _, i := range on.ListOfComponents {
        n.ComponentMap[i.ComponentName] = i
    }

    return nil
}

当我运行 config.NamespaceMap 时，它会给出以下信息

map[:{NamespaceName: K8sNamespace: ComponentMap:map[comp1:{ComponentName:comp1 NumShards:0 NumReplicas:1 Port:0 EpochLength:0}]}]

Answer 1

如果你想进行这样的转换，你需要在 Config 和 Namespace 类型上编写一个自定义的 UnmarshalYAML() 接收器。这是这样做的基本工作示例（仅适用于“命名空间”）：

type Config struct {
    Namespaces map[string]Namespace `yaml:"namespaces"`
}

// Replace go-yaml built-in unmarshaller with custom that will transform list to map.
// Note this code is meant only as demonstration!
func (c *Config) UnmarshalYAML(unmarshal func(interface{}) error) error {
    type origConfig struct {
        Namespaces []Namespace `yaml:"namespaces"`
    }
    var o origConfig
    err := unmarshal(&o)
    if err != nil {
        return err
    }

    // Assign namespaces in list to map; namespaces with same name will be overwritten
    c.Namespaces = make(map[string]Namespace)
    for _, n := range o.Namespaces {
        c.Namespaces[n.Namespace] = n
    }
    return nil
}

// You can use "Config" type as usual
func main() {
    var config Config
    err := yaml.Unmarshal(<your_yaml>, &config)
    if err != nil {
        log.Panic(err)
    }
    fmt.Println(config.Namespaces)
    // map[default:{default k8s} ns2:{ns2 }]
    fmt.Printf("%v\n", config.Namespaces["default"])
    // {default k8s}
}

如代码示例中所述，这将导致一些问题（例如，如果命名空间名称相同怎么办？）。

游乐场链接：https://go.dev/play/p/IKg8kmRnknq