登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-20 10:27:35 265浏览 收藏

在 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 内部将是这样的:

city: washington
name: john

您需要注意的一件事是 go 中的映射不会维护顺序(据我所知 go 没有内置的 orderedmap 类型,2022 年 5 月 7 日),因此新文件中的顺序键将是随机

附加说明:对于错误处理,您最好立即处理(在收到错误后立即处理)。这是一篇关于它的好文章Error handling and Go

今天关于《比较和更新两个yaml文件的Golang实现》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>