文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Go语言如何读取YAML配置文件

2023-07-04 19:51

关注

今天小编给大家分享一下Go语言如何读取YAML配置文件的相关知识点,内容详细,逻辑清晰,相信大部分人都还太了解这方面的知识,所以分享这篇文章给大家参考一下,希望大家阅读完这篇文章后有所收获,下面我们一起来了解一下吧。

yaml.v3 包

yaml.v3 的包,可以让我们在 Go 里面轻松地操作 yaml 格式的数据(如将 yaml 格式转成结构体等)。在使用 yaml.v3 包之前,我们需要先安装它:

go get gopkg.in/yaml.v3

读取 yaml 文件

yaml 测试文件内容:

mysql:  url: 127.0.0.1  port: 3306redis:  host: 127.0.0.1  port: 6379

yaml 文件的数据转成自定义的结构体或 Map

import ("fmt""gopkg.in/yaml.v3""os")type Config struct {Mysql Mysql `json:"mysql"`Redis Redis `json:"redis"`}type Mysql struct {Url  stringPort int}type Redis struct {Host stringPort int}func main() {dataBytes, err := os.ReadFile("test.yaml")if err != nil {fmt.Println("读取文件失败:", err)return}fmt.Println("yaml 文件的内容: \n", string(dataBytes))config := Config{}err = yaml.Unmarshal(dataBytes, &config)if err != nil {fmt.Println("解析 yaml 文件失败:", err)return}fmt.Printf("config → %+v\n", config) // config → {Mysql:{Url:127.0.0.1 Port:3306} Redis:{Host:127.0.0.1 Port:6379}}mp := make(map[string]any, 2)err = yaml.Unmarshal(dataBytes, mp)if err != nil {fmt.Println("解析 yaml 文件失败:", err)return}fmt.Printf("map → %+v", config) // config → {Mysql:{Url:127.0.0.1 Port:3306} Redis:{Host:127.0.0.1 Port:6379}}}

执行结果:

yaml 文件的内容:
 mysql:
  url: 127.0.0.1
  port: 3306

redis:
  host: 127.0.0.1
  port: 6379
config → {Mysql:{Url:127.0.0.1 Port:3306} Redis:{Host:127.0.0.1 Port:6379}}
map → {Mysql:{Url:127.0.0.1 Port:3306} Redis:{Host:127.0.0.1 Port:6379}}

viper 包

viper 包可以帮助我们做很多东西,比如读取 jsonyamlproperties 等配置文件,读取环境变量、读取命令行参数等。在使用 viper 包之前,我们需要先安装它:

go get github.com/spf13/viper

读取 yaml 文件

yaml 测试文件内容:

mysql:  url: 127.0.0.1  port: 3306redis:  host: 127.0.0.1  port: 6379

代码示例:

import (    "fmt"    "github.com/spf13/viper")func main() {    // 设置配置文件的名字    viper.SetConfigName("test")    // 设置配置文件的类型    viper.SetConfigType("yaml")    // 添加配置文件的路径,指定 config 目录下寻找    viper.AddConfigPath("./config")    // 寻找配置文件并读取    err := viper.ReadInConfig()    if err != nil {            panic(fmt.Errorf("fatal error config file: %w", err))    }    fmt.Println(viper.Get("mysql"))     // map[port:3306 url:127.0.0.1]    fmt.Println(viper.Get("mysql.url")) // 127.0.0.1}

通过 SetConfigName 函数,指定配置文件的名称;

通过 SetConfigType 函数,指定配置文件的类型;

通过 AddConfigPath 函数,指定配置文件所在目录,可以多次调用此函数,指定多个目录;

通过 ReadInConfig 函数,寻找配置文件并读取,操作的过程中可能会发生错误,如配置文件没找到,配置文件的内容格式不正确等;读取文件成功之后,可以通过 Get 函数,通过指定 key 获取对应的 value

以上就是“Go语言如何读取YAML配置文件”这篇文章的所有内容,感谢各位的阅读!相信大家阅读完这篇文章都有很大的收获,小编每天都会为大家更新不同的知识,如果还想学习更多的知识,请关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯