Go语言如何向YAML文件动态添加结构体格式新条目

需要在运行时将结构体值config.Sif["snk_prod"]动态添加到YAML文件的sif节点下，生成指定格式的YAML输出。编写Go代码时填充结构体出现错误，需解决思路，要求以结构体形式添加内容。

原代码运行时会触发空指针panic，核心原因是CustomConfig的Sif字段未初始化就直接赋值。

• 初始化Map字段：使用结构体前必须先初始化Sif这个map类型字段，避免空指针异常
• 直接操作解析后的YAML对象：无需先marshal结构体再unmarshal成YamlObject，可以直接将结构体值转换为兼容的map类型，或者直接把结构体存入目标节点
• 简化流程：解析原始YAML后，直接找到sif节点，将结构体值添加进去，减少不必要的序列化/反序列化步骤

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 ecfg struct {
	Type   string   `yaml:"type,omitempty"`
	Inputs []string `yaml:"inputs,omitempty"`
	Ep     string   `yaml:"ep,omitempty"`
	Dken   string   `yaml:"dken,omitempty"`
}

func main() {
	// 解析原始YAML文档
	doc := make(YamlObject)
	if err := yaml.Unmarshal([]byte(document), &doc); err != nil {
		panic(err)
	}

	// 找到sif节点
	sifNode := findChild(doc, "spec", "mec", "customConfig", "sif")
	if sifNode == nil {
		panic("sif节点不存在")
	}

	// 构造要添加的结构体
	snkProd := ecfg{
		Type:   "sk_hc_ls",
		Inputs: []string{"tesslt"},
		Ep:     "${NT}",
	}

	// 将结构体转换为YAML兼容的map，直接添加到sif节点
	var snkProdMap YamlObject
	snkProdBytes, _ := yaml.Marshal(snkProd)
	if err := yaml.Unmarshal(snkProdBytes, &snkProdMap); err != nil {
		panic(err)
	}
	(*sifNode)["snk_prod"] = snkProdMap

	// 生成最终YAML
	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:]...)
}

• 去掉冗余的CustomConfig结构体，直接操作解析后的YAML对象
• 初始化要添加的结构体后，通过临时序列化/反序列化将其转为YamlObject类型，确保能正确合并到目标节点
• 避免了原代码中未初始化Sif map导致的空指针panic

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}

内容的提问来源于stack exchange，提问作者JME