Golang比较两个YAML文件并更新它们

Question

我是戈兰的新手。我正在尝试比较两个YAML文件，并更新第二个文件的值，如果该特定键的第1个YAML中有任何新值。

因此，文件是格式的：这些是​​示例yaml文件。真正的YAML文件具有每个密钥的不同数据类型的嵌套复杂地图。 第1 YAML：

name: john
city: washington

第二个YAML：

name: peter
city: washington

第二个YAML文件的最终结果应为：

name: john
city: washington

尝试使用Unmarshal为两个YAML文件创建MAP String接口。但是如何比较两个地图。试图在地图的每个键上循环，并在第二个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中，请检查数据中的值。如果dat的值不同，请使用该密钥中的数据值进行更新。另外，如果DAT中存在数据的任何键，请在DAT中附加该密钥。但无法正确实施。

Answer 1

您可以尝试比较地图，然后如果键存在，然后将其更新。这是使用您的案例的一些示例。

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中的地图无法维护订单（Afaik Go没有内置orderedmap键入每2022年5月7日），因此新文件中的顺序键将是随机的

附加注意事项：对于错误处理，您最好立即处理（立即在遇到错误之后）。这是一篇关于它的好文章错误处理并进行

Answer 2

我相信有一个地图包，因为1.18。如果您不在乎添加到目标映射中的新密钥，则可以使用其复制函数。

  func副本[m〜map [k] v，k bodable，v any]（dst，src m）
 

复制在SRC中复制所有键/值对，将它们添加到DST中。当DST中已经存在SRC中的密钥时，DST中的值将被与SRC中的密钥关联的值所覆盖。

source code> source code 该功能非常简单：

func Copy[M ~map[K]V, K comparable, V any](dst, src M) {
    for k, v := range src {
        dst[k] = v
    }
}

---

您也可以自己做。以下代码来自。与上面的复制功能不同，它递归起作用：

func mergeMaps(a, b map[string]interface{}) map[string]interface{} {
    out := make(map[string]interface{}, len(a))
    for k, v := range a {
        out[k] = v
    }
    for k, v := range b {
        if v, ok := v.(map[string]interface{}); ok {
            if bv, ok := out[k]; ok {
                if bv, ok := bv.(map[string]interface{}); ok {
                    out[k] = mergeMaps(bv, v)
                    continue
                }
            }
        }
        out[k] = v
    }
    return out
}