FunTester Go 语言处理 yaml 文件

FunTester · 2024年10月24日 · 2300 次阅读

背景

在当前的微服务时代,每个服务通常独立部署,并且可能有其自身的配置需求。应用程序配置文件通常用于存储设置和参数,以配置应用程序的操作方式。这些文件与代码分离,使开发人员可以在不修改源代码的情况下调整应用程序的行为。有几种流行的配置文件格式,包括 JSON、YAML、TOML、XML 和 INI。每种格式都有其优势和适用场景。

YAML 文件

YAML 是一种简洁、易于人类书写和阅读的数据序列化语言,近年来迅速流行。其简洁的语法让它在配置文件编写中备受欢迎,尤其是在需要高度可读性时。许多主流工具和平台,如 AWS CloudFormation 模板、OpenAPI、Swagger 以及 Kubernetes,广泛使用 YAML 来创建结构清晰、可读性强的配置文件,极大地提升了配置管理的效率和灵活性。

以下是 YAML 的规则简要整理:

  1. 缩进规则: 使用空格表示层级,空格数量需一致,不能使用制表符。
  2. 键值对表示: 键和值之间用冒号加空格分隔。
  3. 列表表示: 使用连字符加空格表示列表项。
  4. 布尔值: 使用 true / false 来表示布尔类型,区分大小写。
  5. 注释: 使用 # 表示注释。
  6. 字符串表示: 字符串无需加引号,特殊字符除外。
  7. 换行: 长文本可以通过换行符 |> 进行折叠表示。
  8. 引用: 支持锚点(&)和别名(*)进行引用,避免重复配置。

下面我们来请出今天的主角。

ymal.v3 基础

gopkg.in/yaml.v3 是 Go 语言中用于解析和生成 YAML(YAML Ain't Markup Language)数据的一个非常流行的库。它支持 YAML 1.2 版本,能够提供完整的编码和解码功能,适用于处理结构化数据。

主要功能:

  • YAML 解析与生成:支持从 YAML 文件或数据流中读取和写入数据。
  • 支持复杂数据结构:可以解析包括映射、列表、嵌套结构在内的各种复杂数据类型。
  • 序列化和反序列化:可以将 Go 结构体转为 YAML 文件,或将 YAML 数据解析为 Go 结构体。
  • 流处理:支持从输入流中逐行解析数据,适合处理大型 YAML 文件。

示例

下面是一个解析 YAML 数据的例子:

package main

import (
    "fmt"
    "gopkg.in/yaml.v3"
)

var data = `
name: FunTester
age: 30
languages:
  - Go
  - Java
  - JavaScript
`

func main() {
    var person map[string]interface{}
    err := yaml.Unmarshal([]byte(data), &person)
    if err != nil {
        panic(err)
    }
    fmt.Printf("解析后的数据: %v\n", person)
}

控制台输出:

解析后的数据: map[age:30 languages:[Go Java JavaScript] name:Alice]

下面来一个生成 YAML 数据的演示:

package main

import (
    "fmt"
    "gopkg.in/yaml.v3"
)

func main() {
    person := map[string]interface{}{
        "name":      "Bob",
        "age":       25,
        "languages": []string{"Go", "JavaScript"},
    }

    out, err := yaml.Marshal(&person)
    if err != nil {
        panic(err)
    }
    fmt.Printf("生成的 YAML 数据:\n%s", out)
}

控制台输出:

生成的 YAML 数据:
age: 25
languages:
- Go
- JavaScript
name: Bob

ymal.v3 进阶

下面我们来研究一下进阶用法中的结构体与 yaml 文件转换,主要是从 yaml 文件读取数据,转成 Go 语言的结构体的过程,为了方便,我写到一个文件当中了。

下面是本地的 config.yaml 文件内容:

dbConfig:
  host: db-host
  username: admin
  password: admin_pwd

下面是我的测试代码:

package main  

import (  
    "encoding/json"  
    "fmt"
    "os"
    "sigs.k8s.io/yaml")  

func main() {  
    //读取当前目录下的文件  
    file, err := os.ReadFile("config.yaml")  
    if err != nil {  
       panic(err)  
    }  
    //打印文件内容  
    println(string(file))  

    defer func() {  
       if err := recover(); err != nil {  
          println(err)  
       }  
    }()  
    var o DbConfigRoot  
    yaml.Unmarshal(file, &o)  

    //把结构体转成json  
    jsonData, _ := json.Marshal(o.DbConfig)  
    fmt.Println(string(jsonData))  

}  

type DbConfigRoot struct {  
    DbConfig DbConfig `yaml:"dbConfig"`  
}  

type DbConfig struct {  
    Host     string `yaml:"host"`  
    UserName string `yaml:"username"`  
    Password string `yaml:"password"`  
}

下面我们来看看控制台打印的内容:

dbConfig:
  host: db-host
  username: admin
  password: admin_pwd
  port: 3306
{"Host":"db-host","UserName":"admin","Password":"admin_pwd"}

非常完美地实现了我们的预期。

夸夸

下面是 yaml.v3 的优点:

  1. 全面支持 YAML 1.2:提供完整的解析和生成功能,支持最新的 YAML 规范。
  2. 简单易用:提供了直观的 API,可以轻松将 YAML 数据和 Go 结构体进行相互转换,简化了配置文件的解析和处理。
  3. 支持复杂数据结构:能够处理嵌套的映射、列表、数组等复杂数据类型,适应各种实际应用场景。
  4. 序列化与反序列化灵活:支持将 Go 结构体序列化为 YAML 文件,也可以将 YAML 文件内容解析为 Go 结构体或 map,非常适合配置管理和数据交换。
  5. 流处理支持:可以按流(如从文件或网络接收的流式数据)解析 YAML,适合处理大型文件或数据流的场景。
  6. 错误处理清晰:解析和生成过程中有明确的错误提示,方便调试与排查问题。
  7. 跨版本兼容性:使用 gopkg.in 版本管理,确保兼容性,同时方便用户根据需要选择合适的版本。

gopkg.in/yaml.v3 是 Go 语言中处理 YAML 格式数据的强大工具,适用于解析和生成各种复杂的配置文件。它不仅支持 YAML 1.2 规范,还提供了灵活的序列化和反序列化功能,帮助开发者轻松实现数据和结构体的转换。其流式处理支持特别适合大规模数据解析场景,提升了处理效率。同时,清晰的错误提示和友好的 API 设计使得开发体验更加顺畅。无论是在微服务架构中加载配置文件,还是在复杂系统中处理嵌套数据结构,gopkg.in/yaml.v3 都是一个不可或缺的工具。对于任何需要 YAML 解析或生成的应用场景,它都能以简洁、高效的方式提供可靠的支持,是 Go 开发者值得信赖的选择。

FunTester 原创精华
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
暂无回复。
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册