golang 修改yaml数据

2023-05-14 21:05:13 821人浏览泡泡鱼

摘要

对于使用golang开发的应用程序来说，在常见的数据存储方式中，YAML格式的数据文件也是常用的一种方式。由于YAML格式的文件可读性好、结构清晰，因此在多种场景下被广泛使用，如kubernetes的配置文件就是以YAML格式来存储。当需要

对于使用golang开发的应用程序来说，在常见的数据存储方式中，YAML格式的数据文件也是常用的一种方式。由于YAML格式的文件可读性好、结构清晰，因此在多种场景下被广泛使用，如kubernetes的配置文件就是以YAML格式来存储。当需要修改YAML格式数据时，如何使用Golang来实现呢？下面将以修改Kubernetes的YAML配置文件为例，介绍如何使用Golang来实现。

导入必要的库

在代码中需要使用到“gopkg.in/yaml.v2”这个库来读取和修改YAML格式的数据。由于Golang的依赖管理工具go modules自带了依赖管理功能，因此我们只需要引用这个库即可。

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

读取YAML数据

在修改YAML数据之前，首先需要将YAML数据读取出来，这里可以用ioutil库中的ReadFile函数来读取文件。由于Kubernetes的配置文件是一个可以包含多个对象的文档，因此需要将文件中的所有对象使用“---”分隔开。

func readYamlFile(filepath string) ([]map[interface{}]interface{}, error) {
    yamlFile, err := ioutil.ReadFile(filepath)
    if err != nil {
        return nil, err
    }
    var data []map[interface{}]interface{}
    for _, item := range bytes.Split(yamlFile, []byte("
---
")) {
        var obj map[interface{}]interface{}
        if err := yaml.Unmarshal(item, &obj); err != nil {
            return nil, err
        }
        data = append(data, obj)
    }
    return data, nil
}

这个函数的返回值是一个[]map[interface{}]interface{}类型的数据，每一个元素表示一个YAML对象，在Kubernetes的配置文件中，每一个对象通常代表一个Deployment、Service或者Pod等。

修改YAML数据

假设我们需要将某个Deployment的Replicas从原来的3个改为4个，那么我们需要找到这个Deployment对象，然后修改Replicas字段的值。由于YAML格式的数据每个字段都是一个map，因此修改某个字段的值非常简单，只需要找到这个字段所在的map，然后修改这个map的值即可。这里在遍历每个YAML对象时，检查当前是否是我们需要修改的Deployment，如果是则查找Replicas字段并修改。

func modifyYamlData(data []map[interface{}]interface{}, objectName string, fieldName string, fieldValue interface{}) error {
    for _, obj := range data {
        if obj["kind"] == "Deployment" && obj["metadata"].(map[interface{}]interface{})["name"] == objectName {
            spec := obj["spec"].(map[interface{}]interface{})
            replicas := spec["replicas"].(int)
            if replicas != fieldValue {
                spec["replicas"] = fieldValue
                yamlData, err := yaml.Marshal(data)
                if err != nil {
                    return err
                }
                return ioutil.WriteFile("k8s.yaml", yamlData, 0644)
            }
            break
        }
    }
    return nil
}

这个函数的参数依次是读取出来的YAML数据，需要修改的对象名、属性名和属性值。如果找到了需要修改的对象并修改完成，就可以直接写入文件了。

完整代码

下面是一个完整的程序，用于修改Kubernetes配置文件中的Deployment的Replicas属性。可以在此基础上扩展修改其它属性。

package main

import (
    "bytes"
    "fmt"
    "io/ioutil"
  
    "gopkg.in/yaml.v2"
)

func readYamlFile(filepath string) ([]map[interface{}]interface{}, error) {
    yamlFile, err := ioutil.ReadFile(filepath)
    if err != nil {
        return nil, err
    }
    var data []map[interface{}]interface{}
    for _, item := range bytes.Split(yamlFile, []byte("
---
")) {
        var obj map[interface{}]interface{}
        if err := yaml.Unmarshal(item, &obj); err != nil {
            return nil, err
        }
        data = append(data, obj)
    }
    return data, nil
}

func modifyYamlData(data []map[interface{}]interface{}, objectName string, fieldName string, fieldValue interface{}) error {
    for _, obj := range data {
        if obj["kind"] == "Deployment" && obj["metadata"].(map[interface{}]interface{})["name"] == objectName {
            spec := obj["spec"].(map[interface{}]interface{})
            replicas := spec["replicas"].(int)
            if replicas != fieldValue {
                spec["replicas"] = fieldValue
                yamlData, err := yaml.Marshal(data)
                if err != nil {
                    return err
                }
                return ioutil.WriteFile("k8s.yaml", yamlData, 0644)
            }
            break
        }
    }
    return nil
}

func main() {
    data, err := readYamlFile("k8s.yaml")
    if err != nil {
        fmt.Println(err)
        return
    }
    if err := modifyYamlData(data, "Nginx", "replicas", 4); err != nil {
        fmt.Println(err)
        return
    }
}