golang 模板 json 转 yaml 格式

Question

您好，我希望使用传递给我们的 go api 的 Json 数据以以下格式创建 Yml。

zookeeper:
  configs:
  
    zookeeper_user: custom-user
    zookeeper_group: custom-group
      
    zookeeper_copy_files:
       - source_path: /path/to/file.txt
         destination_path: /tmp/file.txt

    zookeeper_Some_otherfile:
       - source_path: /path/to/file.txt
         destination_path: /tmp/file.txt
  hosts:
    ip-172-31-34-246.us-east-2.compute.internal:
      zookeeper_id: 2
    ip-172-31-37-15.us-east-2.compute.internal:
      zookeeper_id: 3
    ip-172-31-34-231.us-east-2.compute.internal:
      zookeeper_id: 1


      # zookeeper:
      #   properties:
      #     initLimit: 6
      #     syncLimit: 3
      # zookeeper_server:
      #   properties:
      #     num.io: 15
      # zookeeper_registry:
      #   properties:
      #     key: val
      # Zookeeper_center:
      #   properties:
      #     key: val

使用以下结构解组 json 数据

type Service struct {
    Name string `json:"name"`
    Hosts []Host `json:"hosts"`
    Configs  map[string]interface{} `json:"configs"` 

}

type Host struct {
    HostName         string                 `json:"host_name"`
    CustomHostConfig map[string]interface{} `json:"config"`
}

{{ range $s := $.Services }}
{{ $s.Name }}:
  {{ range $k,$v := $s.Vars }}{{ $k }}: "{{ $v }}"
  {{ end }}
  hosts:
    {{ range $h := $s.Hosts }}
    {{ $h.HostName }}: {{ range $k,$v := $h.CustomHostConfig }}
       {{ $k }}: "{{ $v }}"
      {{ end }}
{{ end }}{{ end }}

我正在使用上面的模板代码，它可以很好地创建以下代码

zookeeper:
  configs:

    zookeeper_user: custom-user
    zookeeper_group: custom-group


  hosts:
    ip-172-31-34-246.us-east-2.compute.internal:
      zookeeper_id: 2
    ip-172-31-37-15.us-east-2.compute.internal:
      zookeeper_id: 3
    ip-172-31-34-231.us-east-2.compute.internal:
      zookeeper_id: 1

我需要帮助将以下配置部分添加到最终的 yml 文件中，以某种动态方式基于使用 Configs map[string]interface{} json:"configs"` 作为键值 json 的一部分传递的用户输入。具有正确的缩进级别。

   zookeeper_copy_files:
       - source_path: /path/to/file.txt
         destination_path: /tmp/file.txt

    zookeeper_Some_otherfile:
       - source_path: /path/to/file.txt
         destination_path: /tmp/file.txt

      # zookeeper:
      #   properties:
      #     initLimit: 6
      #     syncLimit: 3
      # zookeeper_server:
      #   properties:
      #     num.io: 15
      # zookeeper_registry:
      #   properties:
      #     key: val
      # Zookeeper_center:
      #   properties:
      #     key: val
  

Answer 1

不要使用 go 模板编写 YAML，只需使用 YAML 实现，例如：

package main

import (
    "fmt"

    "gopkg.in/yaml.v3"
)

type Service struct {
    Name    string                 `json:"name"`
    Hosts   []Host                 `json:"hosts"`
    Configs map[string]interface{} `json:"configs"`
}

type Host struct {
    HostName         string                 `json:"host_name"`
    CustomHostConfig map[string]interface{} `json:"config"`
}

var services []Service

// create some sample data similar to what you show
func init() {
    var zk Service
    zk.Name = "zookeeper"
    zk.Configs = map[string]interface{}{
        "zookeeper_user":  "custom-user",
        "zookeeper_group": "custom-group",
    }
    zk.Hosts = []Host{
        {"ip-172-31-34-246.us-east-2.compute.internal",
            map[string]interface{}{
                "zookeeper_id": 2,
            }},
        {"ip-172-31-37-15.us-east-2.compute.internal",
            map[string]interface{}{
                "zookeeper_id": 3,
            }},
    }
    services = []Service{zk}
}

func main() {
    out := make(map[string]interface{})
    for _, service := range services {
        var hosts = make(map[string]interface{})
        for _, host := range service.Hosts {
            hosts[host.HostName] = host.CustomHostConfig
        }
        out[service.Name] = map[string]interface{}{
            "hosts":   hosts,
            "configs": service.Configs,
        }
    }
    bytes, err := yaml.Marshal(out)
    if err != nil {
        panic(err)
    }
    fmt.Println(string(bytes))
}

输出（go playground）：

zookeeper:
    configs:
        zookeeper_group: custom-group
        zookeeper_user: custom-user
    hosts:
        ip-172-31-34-246.us-east-2.compute.internal:
            zookeeper_id: 2
        ip-172-31-37-15.us-east-2.compute.internal:
            zookeeper_id: 3

这会立即解决您对深度嵌套数据的任何问题，因为 YAML 只是按原样转储结构。