dongwuxie5112 2015-08-21 18:41
浏览 236
已采纳

如何在Golang中使用动态键解析YAML

I am trying to parse a YAML file with Go. The problem is that the keys in the YAML file might not always be the same. This is to do API versioning so the user can define the versions they support. For instance V1, V2, V3 etc. They do not need to be in order and can omit versions they don't support i.e, V0, V2, V5 etc.

package main

import (
  "fmt"
  "gopkg.in/yaml.v2"
)

var data = `
---
development:
  skip-header-validation: true
  V1:
    current: "1.0.0"
    mime_types:
      - application/vnd.company.jk.identity+json;
      - application/vnd.company.jk.user+json;
      - application/vnd.company.jk.role+json;
      - application/vnd.company.jk.scope+json;
      - application/vnd.company.jk.test+json;
    skip-mime-type-validation: true
    skip-version-validation: true
  V2:
    current: "2.0.0"
    mime_types:
      - application/vnd.company.jk.identity+json;
      - application/vnd.company.jk.user+json;
      - application/vnd.company.jk.role+json;
      - application/vnd.company.jk.scope+json;
      - application/vnd.company.jk.test+json;

`

type MajorVersion struct {
  Current                 string    `yaml:"current"`
  MimeTypes               []string  `yaml:"mime_types"`
  SkipVersionValidation   bool      `yaml:"skip-version-validation"`
  SkipMimeTypeValidation  bool      `yaml:"skip-mime-type-validation"`
}

type Environment struct {
  SkipHeaderValidation  bool        `yaml:"skip-header-validation"`
  Version               map[string]MajorVersion
}

func main() {
  e := Environment{}

  yaml.Unmarshal([]byte(data), &e)
  fmt.Println(e)
}

I saw a similar question asked here

This is at the top level and I haven't quite figured out how to do this from inside the struct.

  • 写回答

1条回答 默认 最新

  • douwen3836 2015-08-22 15:37
    关注

    First, you are trying to parse the root as an Environment, but its actual type is map[string]Environment. Second, you are going to need a custom Unmarshaler if you want to keep that type structure. Something like this:

    package main
    
    import (
        "fmt"
    
        "gopkg.in/yaml.v2"
    )
    
    var data = `
    ---
    development:
      skip-header-validation: true
      V1:
        current: "1.0.0"
        mime_types:
          - application/vnd.company.jk.identity+json;
          - application/vnd.company.jk.user+json;
          - application/vnd.company.jk.role+json;
          - application/vnd.company.jk.scope+json;
          - application/vnd.company.jk.test+json;
        skip-mime-type-validation: true
        skip-version-validation: true
      V2:
        current: "2.0.0"
        mime_types:
          - application/vnd.company.jk.identity+json;
          - application/vnd.company.jk.user+json;
          - application/vnd.company.jk.role+json;
          - application/vnd.company.jk.scope+json;
          - application/vnd.company.jk.test+json;
    
    `
    
    type MajorVersion struct {
        Current                string   `yaml:"current"`
        MimeTypes              []string `yaml:"mime_types"`
        SkipVersionValidation  bool     `yaml:"skip-version-validation"`
        SkipMimeTypeValidation bool     `yaml:"skip-mime-type-validation"`
    }
    
    type Environment struct {
        SkipHeaderValidation bool
        Versions             map[string]MajorVersion
    }
    
    func (e *Environment) UnmarshalYAML(unmarshal func(interface{}) error) error {
        var params struct {
            SkipHeaderValidation bool `yaml:"skip-header-validation"`
        }
        if err := unmarshal(&params); err != nil {
            return err
        }
        var versions map[string]MajorVersion
        if err := unmarshal(&versions); err != nil {
            // Here we expect an error because a boolean cannot be converted to a
            // a MajorVersion
            if _, ok := err.(*yaml.TypeError); !ok {
                return err
            }
        }
        e.SkipHeaderValidation = params.SkipHeaderValidation
        e.Versions = versions
        return nil
    }
    
    func main() {
        var e map[string]Environment
        if err := yaml.Unmarshal([]byte(data), &e); err != nil {
            fmt.Println(err.Error())
        }
        fmt.Printf("%#v
    ", e)
    }
    

    Output (after using makeup):

    map[string]main.Environment{
        "development": {
            SkipHeaderValidation: true,
            Versions:             {
                "V2": {
                    Current:                "2.0.0",
                    MimeTypes:              {"application/vnd.company.jk.identity+json;", "application/vnd.company.jk.user+json;", "application/vnd.company.jk.role+json;", "application/vnd.company.jk.scope+json;", "application/vnd.company.jk.test+json;"},
                    SkipVersionValidation:  false,
                    SkipMimeTypeValidation: false,
                },
                "V1": {
                    Current:                "1.0.0",
                    MimeTypes:              {"application/vnd.company.jk.identity+json;", "application/vnd.company.jk.user+json;", "application/vnd.company.jk.role+json;", "application/vnd.company.jk.scope+json;", "application/vnd.company.jk.test+json;"},
                    SkipVersionValidation:  true,
                    SkipMimeTypeValidation: true,
                },
            },
        },
    }
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥100 关于使用MATLAB中copularnd函数的问题
  • ¥20 在虚拟机的pycharm上
  • ¥15 jupyterthemes 设置完毕后没有效果
  • ¥15 matlab图像高斯低通滤波
  • ¥15 针对曲面部件的制孔路径规划,大家有什么思路吗
  • ¥15 钢筋实图交点识别,机器视觉代码
  • ¥15 如何在Linux系统中,但是在window系统上idea里面可以正常运行?(相关搜索:jar包)
  • ¥50 400g qsfp 光模块iphy方案
  • ¥15 两块ADC0804用proteus仿真时,出现异常
  • ¥15 关于风控系统,如何去选择