当前位置： > > > > go语言读取并合并两个Yaml文件

go语言读取并合并两个Yaml文件

来源：stackoverflow
2024-04-20 12:24:40
0浏览
收藏

从现在开始，我们要努力学习啦！今天我给大家带来《go语言读取并合并两个Yaml文件》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

问题内容

假设我们有两个 yaml 文件

master.yaml

someproperty: "somevaue"
anotherproperty: "anothervalue"

override.yaml

someproperty: "overriddenvaue"

是否可以解组、合并，然后将这些更改写入文件，而无需为 yaml 文件中的每个属性定义 struct？

主文件中有超过 500 个属性，这些属性在执行时对服务一点也不重要，因此理想情况下，我能够将其解组到映射中，进行合并并再次在 yaml 中写入但我相对较新，所以想要一些意见。

我有一些代码可以将 yaml 读入 interface，但我不确定合并两者的最佳方法。

var masteryaml interface{}
yamlbytes, _ := ioutil.readfile("master.yaml")
yaml.unmarshal(yamlbytes, &masteryaml)

var overrideyaml interface{}
yamlbytes, _ = ioutil.readfile("override.yaml")
yaml.unmarshal(yamlbytes, &overrideyaml)

我研究过像 mergo 这样的库，但我不确定这是否是正确的方法。

我希望在掌握大师之后我能够将属性写入文件

someProperty: "overriddenVaue"
anotherProperty: "anotherValue"

解决方案

假设您只想在顶层合并，则可以解组为 map[string]interface{} 类型的映射，如下所示：

主包

import (
    "io/ioutil"

    "gopkg.in/yaml.v2"
)

func main() {
    var master map[string]interface{}
    bs, err := ioutil.readfile("master.yaml")
    if err != nil {
        panic(err)
    }
    if err := yaml.unmarshal(bs, &master); err != nil {
        panic(err)
    }

    var override map[string]interface{}
    bs, err = ioutil.readfile("override.yaml")
    if err != nil {
        panic(err)
    }
    if err := yaml.unmarshal(bs, &override); err != nil {
        panic(err)
    }

    for k, v := range override {
        master[k] = v
    }

    bs, err = yaml.marshal(master)
    if err != nil {
        panic(err)
    }
    if err := ioutil.writefile("merged.yaml", bs, 0644); err != nil {
        panic(err)
    }
}

对于更广泛的解决方案（具有 n 个输入文件），您可以使用此函数。我已经使用@robox答案来完成我的解决方案：

func readvalues(filenames ...string) (string, error) {
    if len(filenames) <= 0 {
        return "", errors.new("you must provide at least one filename for reading values")
    }
    var resultvalues map[string]interface{}
    for _, filename := range filenames {

        var override map[string]interface{}
        bs, err := ioutil.readfile(filename)
        if err != nil {
            log.info(err)
            continue
        }
        if err := yaml.unmarshal(bs, &override); err != nil {
            log.info(err)
            continue
        }

        //check if is nil. this will only happen for the first filename
        if resultvalues == nil {
            resultvalues = override
        } else {
            for k, v := range override {
                resultvalues[k] = v
            }
        }

    }
    bs, err := yaml.marshal(resultvalues)
    if err != nil {
        log.info(err)
        return "", err
    }

    return string(bs), nil
}

因此，对于这个示例，您应该按以下顺序调用它：

result, _ := readvalues("master.yaml", "overwrite.yaml")

如果你有一个额外的文件 newfile.yaml，你也可以使用这个函数：

result, _ := ReadValues("master.yaml", "overwrite.yaml", "newFile.yaml")

本篇关于《go语言读取并合并两个Yaml文件》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注米云公众号！