快捷導(dǎo)航

Golang合并yaml文件過(guò)程逐步講解

更新時(shí)間：2023年01月10日 10:24:59 作者：夢(mèng)想畫(huà)家

之前一直從事java開(kāi)發(fā),習(xí)慣了使用yaml文件的格式,尤其是清晰的層次結(jié)構(gòu)、注釋,下面這篇文章主要給大家介紹了關(guān)于Golang合并yaml文件的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下

當(dāng)前有趨勢(shì)使用環(huán)境變量代替配置文件，主要是為了方便容器部署，因?yàn)槲募枰~外文件權(quán)限。但環(huán)境變量也有缺點(diǎn)，如長(zhǎng)度和因數(shù)量太多造成的復(fù)雜性，另外嵌套也很難實(shí)現(xiàn)，此時(shí)配置文件則更合適。

本文就介紹Golang讀寫(xiě)配置文件，以及合并配置獲得最終配置信息。

讀配置

首先準(zhǔn)備配置文件config.yaml:

name: test
image: tom.png
environment:
  os: linux
  group: admin
  user: testAdmin
limits:
  cpu: 4
  memory: 512M

為了加載配置，需要定義結(jié)構(gòu)體及標(biāo)簽，和json標(biāo)記類(lèi)似：

type Spec struct {
	// Name: name of the function
	Name string `yaml:"name"`
	// Image: docker image name of the function
	Image       string            `yaml:"image"`
	Environment map[string]string `yaml:"environment,omitempty"`
	// Limits for the function
	Limits *FunctionResources `yaml:"limits,omitempty"`
}
// FunctionResources Memory and CPU
type FunctionResources struct {
	Memory string ` yaml:"memory"`
	CPU    string ` yaml:"cpu"`
}

這里僅用yaml代替json，下面解析也同樣使用Unmarshal接口生成對(duì)象：

import (
    "fmt"
    "os"
	"gopkg.in/yaml.v2"
)
func ReadYaml() {
	bytesOut, err := os.ReadFile("config.yaml")
	if err != nil {
		panic(err)
	}
	spec := Spec{}
	if err := yaml.Unmarshal(bytesOut, &spec); err != nil {
		panic(err)
	}
	fmt.Printf("Function name: %s\tImage: %s\tEnvs: %d\n", spec.Name, spec.Image,
		len(spec.Environment))
	fmt.Printf("Limists: %s, %s\n", spec.Limits.CPU, spec.Limits.Memory)
}

輸出結(jié)果：

Function name: test Image: tom.png Envs: 3
Limists: 4, 512M

寫(xiě)配置

寫(xiě)文件首先使用接口Marshal把對(duì)象序列化為字節(jié)數(shù)組，然后調(diào)用os方法寫(xiě)yaml文件：

func WriteYaml() {
	spec := Spec{
		Image: "docker.io/functions/figlet:latest",
		Name:  "figlet",
	}
	bytesOut, err := yaml.Marshal(spec)
	if err != nil {
		panic(err)
	}
	err = os.WriteFile("figlet.yaml", bytesOut, os.ModePerm)
	if err != nil {
		panic(err)
	}
	fmt.Printf("Wrote: figlet.yaml. . OK. \n")
}

運(yùn)行程序成功生成figlet.yaml文件并輸出：

Wrote: figlet.yaml. . OK.

使用yaml.v2庫(kù)可以快速實(shí)現(xiàn)配置文件相關(guān)工作，下面介紹如何合并多個(gè)配置信息。

合并配置

如果配置信息包括包括很多字段，則可能需要維護(hù)一個(gè)示例文件，并允許用戶提供新的值進(jìn)行修改。我們可以使用mergo庫(kù)實(shí)現(xiàn)合并功能，當(dāng)然其他庫(kù)也可以實(shí)現(xiàn)類(lèi)似功能，該方法支持內(nèi)存中任何結(jié)構(gòu)體，不一定為YAML:

import "github.com/imdario/mergo"
func MergeYaml() {
	base := Spec{
		Image:       "docker.io/functions/figlet:latest",
		Name:        "figlet",
		Environment: map[string]string{"stage": "develop"},
		Limits:      &FunctionResources{Memory: "512Mi", CPU: "60Mi"},
	}
	production := Spec{
		Environment: map[string]string{"stage": "production"},
		Limits:      &FunctionResources{Memory: "1Gi", CPU: "100Mi"},
	}
	overrides := []Spec{
		base,
		production,
	}
	merged := Spec{}
	for _, override := range overrides {
		err := mergo.Merge(&merged, override, mergo.WithOverride)
		if err != nil {
			panic(err)
		}
	}
	bytesOut, err := yaml.Marshal(merged)
	if err != nil {
		panic(err)
	}
	fmt.Printf("Merged content: \n\n%s\n", string(bytesOut))
}

運(yùn)行測(cè)試輸出結(jié)果：