当前位置：首页 > 文章列表 > Golang > Go问答 > 比较和更新两个yaml文件的Golang实现

比较和更新两个yaml文件的Golang实现

来源：stackoverflow 2024-03-20 10:27:35 0浏览收藏

在 Go 语言中，比较和更新两个 YAML 文件的过程涉及读取和反序列化文件，然后将数据存储在映射中。通过遍历这两个映射并比较键，可以识别出差异。对于存在的键，可以更新第二个映射中的值。对于不存在的键，可以将其附加到第二个映射中。通过使用内置的 YAML 库或第三方库（如 mergo），可以简化这一过程，从而使比较和更新变得更加容易。

问题内容

我是 golang 的新手。我正在尝试比较两个 yaml 文件，如果第一个 yaml 中有该特定键的新值，则更新第二个文件的值。

因此文件的格式为：这些是示例 yaml 文件。真实的 yaml 文件具有更多嵌套的复杂映射，每个键具有不同的数据类型。第一个 yaml：

name: john
city: washington

第二个yaml：

name: peter
city: washington

第二个 yaml 文件的最终结果应该是：

name: john
city: washington

尝试使用 unmarshal 为两个 yaml 文件创建映射字符串接口。但在如何比较两张地图时遇到了麻烦。试图循环遍历地图的每个键并在第二个 yaml 地图中搜索该键。如果键存在，则更新第二个 yaml 映射中的值。但我无法实现这一点。有什么建议/更好的想法吗？

编辑：更新代码

package main

import (
    "fmt"
    "io/ioutil"
    "log"

    "github.com/imdario/mergo"
    "gopkg.in/yaml.v3"
)

func main() {

    yfile, err := ioutil.ReadFile("C:/Users/212764682/lifecycle/userconfig.yaml")

    if err != nil {

        log.Fatal(err)
    }

    data := make(map[string]interface{})

    err2 := yaml.Unmarshal(yfile, &data)

    if err2 != nil {

        log.Fatal(err2)
    } else {
        
        yfile1, err3 := ioutil.ReadFile("C:/Users/212764682/lifecycle/conf.yaml")
        yfile2, err4 := ioutil.ReadFile("C:/Users/212764682/lifecycle/prof.yaml")

        if err3 != nil && err4 != nil {

            log.Fatal(err3)
            log.Fatal(err4)
        } else {
            dat := make(map[string]interface{})
            dat2 := make(map[string]interface{})
            err5 := yaml.Unmarshal(yfile1, &dat)
            err6 := yaml.Unmarshal(yfile2, &dat2)

            _ = err5
            _ = err6

            for key1, element1 := range data {
                for key2, element2 := range dat {
                    if key1 == key2 {
                        if element1 == element2 {

                        } else {
                            element2 = element1
                        }
                    } else {
                        dat[key1] = data[key1]
                    }
                }

            }
        }

    }

}

所以我想将数据的每个键与 dat 进行比较。如果该键存在于 dat 中，请检查 data 中的值。如果 dat 中的值不同，则使用该键的 dat 中的数据值进行更新。另外，如果 dat 中不存在任何数据键，则将该键附加到 dat 中。但无法正确实现。

正确答案

您可以尝试比较地图，如果密钥存在则更新它。这是使用您的案例的一些示例。

package main

import (
    "fmt"
    "io/ioutil"
    "log"

    "gopkg.in/yaml.v3"
)

func main() {
    // read file
    yfile1, err := ioutil.readfile("file1.yaml")
    if err != nil {
        log.fatal(err)
        return
    }

    yfile2, err := ioutil.readfile("file2.yaml")
    if err != nil {
        log.fatal(err)
        return
    }

    // unmarshal ymal
    data1 := make(map[string]interface{})
    if err := yaml.unmarshal(yfile1, &data1); err != nil {
        log.fatal(err)
        return
    }

    data2 := make(map[string]interface{})
    if err := yaml.unmarshal(yfile2, &data2); err != nil {
        log.fatal(err)
        return
    }

    // from this we can iterate the key in data2 then check whether it exists in data1
    // if so then we can update the value in data2

    // iterate key in data2
    for key2 := range data2 {
        // check whether key2 exists in data1
        if val1, ok := data1[key2]; ok {
            // update the value of key2 in data2
            data2[key2] = val1
        }
    }

    fmt.printf("data2: %v", data2)
    // output:
    // data2: map[city:washington name:john]

    // you can write the data2 into ymal
    newyfile2, err := yaml.marshal(&data2)
    if err != nil {
        log.fatal(err)
        return
    }

    // write to file
    if err = ioutil.writefile("new_file2.yaml", newyfile2, 0644); err != nil {
        log.fatal(err)
        return
    }
}

new_file2.yaml 内部将是这样的：