golang 修改yaml數據-Golang-PHP中文網

golang 修改yaml數據

WBOY

發布： 2023-05-14 16:18:08

原創

907 人瀏覽過

對於使用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
    }
}

登入後複製