如何在yaml文件中新增条目

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《如何在yaml文件中新增条目》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

我需要从结构值 config.sif["snk_prod"] 添加条目到 yaml 文件（我需要在运行时填充它）并尝试以下操作，我尝试了以下操作，但在填充时出现错误结构，有什么想法吗？

package main

import "gopkg.in/yaml.v3"

const document = `
spec:
  mec:
    customconfig:
      sif:
        prom_exporter:
          type: prometheus_exporter
        snk_dev:   
          type: sk_hc_logs
          inputs:
            - tesslt
          ep: ${nt}
          dken: ${sn}
`

type yamlobject map[string]any

type customconfig struct {
    sif map[string]interface{} `yaml:"sif,omitempty"`
}

type ecfg struct {
    type   string   `yaml:"type,omitempty"`
    inputs []string `yaml:"inputs,omitempty"`
    ep     string   `yaml:"ep,omitempty"`
    dken   string   `yaml:"dken,omitempty"`
}

func main() {
    t := customconfig{}
    config := &t
   // -------i need to add it as struct values as i got the values on runtime dynamically  
    config.sif["snk_prod"] = ecfg{
        type:   "sk_hc_ls",
        inputs: []string{"tesslt"},
        ep:     "${nt}",
    }
    yamlbytes, err := yaml.marshal(t)

    doc := make(yamlobject)
    if err := yaml.unmarshal([]byte(document), &doc); err != nil {
        panic(err)
    }
    addon := make(yamlobject)
    if err := yaml.unmarshal(yamlbytes, &addon); err != nil {
        panic(err)
    }
    node := findchild(doc, "spec", "mec", "customconfig", "sif")
    if node == nil {
        panic("must not happen")
    }

    for key, val := range addon {
        (*node)[key] = val
    }
    outdoc, err := yaml.marshal(doc)
    if err != nil {
        panic(err)
    }
    println(string(outdoc))
}

func findchild(obj yamlobject, path ...string) *yamlobject {
    if len(path) == 0 {
        return &obj
    }
    key := path[0]
    child, ok := obj[key]
    if !ok {
        return nil
    }
    obj, ok = child.(yamlobject)
    if !ok {
        return nil
    }
    return findchild(obj, path[1:]...)
}

https://go.dev/play/p/6chsojpxqpw

搜索后我找到了这个答案https://stackoverflow.com/a/74089724/6340176 这与我的非常相似，变化是我需要将其添加为结构值

最后我需要在 sif 下添加新条目

最后输出应如下所示

spec:
  mec:
    customConfig:
      sif:
        prom_exporter:
          type: prometheus_exporter
        snk_dev:   
          type: sk_hc_logs
          inputs:
            - tesslt
          ep: ${NT}
          dken: ${SN}
        snk_prod:   
          type: sk_hc_ls
          inputs:
            - tesslt
          ep: ${NT}

正确答案

按如下方式替换 yamlbytes 的创建：

t := make(map[string]interface{})
    t["snk_prod"] = ecfg{
        type:   "sk_hc_ls",
        inputs: []string{"tesslt"},
        ep:     "${nt}",
    }
    yamlbytes, err := yaml.marshal(t)

然后你就会得到预期的结果。

顺便说一句：触发恐慌是因为您尝试将值插入到未初始化的映射中（config.sif 是 nil）。例如，您可以在分配值之前使用以下代码行简单地创建一个空地图：

config.sif = make(map[string]interface{})

但是代码中会有一个额外的不需要的 sif 节点，例如像这样的东西：

...
    sif:
        prom_exporter:
            type: prometheus_exporter
        sif:
            snk_prod:
    ...

因此，应如开头所示生成要动态添加的 yaml 片段。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~