golang 读取yaml配置文件中的数据 两种方式:yaml.v2 和 Viper

golang 读取yaml配置文件中的数据

yaml 配置文件 config.yaml 中 写数据:

app:
  host: 127.0.0.1
  port: 3306
  username: admin
  password: admin
log:
  suffix: log
  maxSize: 5

config 文件 下 是 配置文件对应的结构体 config.go

package config

type Config struct {
	App App `yaml:"app" json:"app"`
	Log Log `yaml:"log" json:"log"`
}

下面是 具体的结构体 app.golog.go

type App struct {
	Host     string `yaml:"host" json:"host"`
	Port     int    `yaml:"port" json:"port"`
	Username string `yaml:"username" json:"username"`
	Password string `yaml:"password" json:"password"`
}

type Log struct {
	Suffix  string `yaml:"suffix" json:"suffix"`
	MaxSize int    `yaml:"maxSize" json:"maxsize"`
}

这种写结构体的方法比较有层级,后续配置文件需要增加数据,可以对不同的配置用不同的结构体,使用起来不会杂乱;

方法一: 使用yaml.v2 读取配置文件数据,需要导入对应的包 :yaml.v2
需要注意的是:yaml.v2只能读取yaml格式的文件

import (
	"gopkg.in/yaml.v2"
)

使用命令:go get gopkg.in/yaml.v2 导入对应 的包。

使用ioutil导入配置文件,使用yaml.Unmarshal将配置文件读取到结构体中,示例代码如下。

package main

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

func GetConfigData() {
	//导入配置文件
	yamlFile, err := ioutil.ReadFile("./config.yml")
	if err != nil {
		fmt.Println(err.Error())
	}
	var _config *config.Config
	//将配置文件读取到结构体中
	err = yaml.Unmarshal(yamlFile, &_config)
	if err != nil {
		fmt.Println(err.Error())
	}
	username := _config.App.Username
	fmt.Println("username:" + username )
	fmt.Printf("config.log: %#v\n", _config.Log)

}

func main() {
	GetConfigData()
}

方法二:使用Viper读取配置文件数据
Viper可以处理多种格式的配置,目前已支持从JSON、TOML、YAML、HCL、INI和Java properties文件中读取配置数据。

Viper还能监视配置文件的变动、重新读取配置文件。在项目中可以用作热加载

1.导入 Viper .
go get github.com/spf13/viper

2.使用Viper导入配置文件,并将配置文件读取到结构体中,示例代码如下:

package main

import (
	"GoReadConfig/config"
	"fmt"
	"github.com/spf13/viper"
)

func GetConfigData() {
	//导入配置文件
	viper.SetConfigType("yaml")
	viper.SetConfigFile("./config.yml")
	//读取配置文件
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println(err.Error())
	}
	var _config *config.Config
	//将配置文件读到结构体中
	err = viper.Unmarshal(&_config)
	if err != nil {
		fmt.Println(err.Error())
	}
	
	username := _config.App.Username
	fmt.Println("username:" + username )
	fmt.Printf("config.log: %#v\n", _config.Log)
}

func main() {
	GetConfigData()
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值