在go语言中参数配置解决方案:viper
Go  /  管理员 发布于 2个月前   58
在go语言应用中使用viper包来获取配置文件的配置信息,以及实现开发模式下的配置和上线产品如何使用进行viper做配置映射。
简单介绍一下viper
Viper是Go应用程序(包括 12 要素应用程序)的完整配置解决方案。
它设计用于在应用程序中运行,可以处理所有类型的配置需求和格式。
Viper git地址:
https://github.com/spf13/viper
进入步骤
安装viper
go get github.com/spf13/viper
目录结构:
ch1
├── config.yaml //配置文件
└── main.go //viper读取配置
config.yaml:
name: 'user-web'
mysql:
host: '127.0.0.1'
port: 3306
main.go:
package main
import (
"fmt"
"github.com/spf13/viper"
)
type mysqlConfig struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
}
//ServerConfig 将配置文件内容映射为struct
type ServerConfig struct {
ServerName string `mapstructure:"name"`
SqlConfig mysqlConfig `mapstructure:"mysql"`
}
func main() {
//new对象
v := viper.New()
//读取配置文件
v.SetConfigFile("viper_test/ch1/config.yaml")
if err := v.ReadInConfig(); err != nil {
panic(err)
}
ServerConfig := ServerConfig{}
v.Unmarshal(&ServerConfig)
fmt.Println(ServerConfig)
}
在上述代码中我们定义了两个结构体:
type mysqlConfig struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
}
//ServerConfig 将配置文件内容映射为struct
type ServerConfig struct {
ServerName string `mapstructure:"name"`
SqlConfig mysqlConfig `mapstructure:"mysql"`
}
这两个结构体用来做配置映射,这里使用mapstructure为配置做tag
完整代码:
package main
import (
"fmt"
"github.com/spf13/viper"
)
type mysqlConfig struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
}
//ServerConfig 将配置文件内容映射为struct
type ServerConfig struct {
ServerName string `mapstructure:"name"`
SqlConfig mysqlConfig `mapstructure:"mysql"`
}
func main() {
//new一个viper对象
v := viper.New()
//获取配置文件
v.SetConfigFile("viper_test/ch1/config.yaml")
//将配置读入viper对象中
if err := v.ReadInConfig(); err != nil {
panic(err)
}
ServerConfig := ServerConfig{}
//将读取到的配置信息映射至ServerConfig中
v.Unmarshal(&ServerConfig)
fmt.Println(ServerConfig)
}
输出:
{user-web {127.0.0.1 3306}}
实时监控配置(这个场景在某些时候是非常有用的)
viper为我们提供了实时监控配置的功能
v.WatchConfig()
v.OnConfigChange(func(e fsnotify.Event) {
fmt.Println("配置变化:", e)
_ = v.ReadInConfig()
v.Unmarshal(&ServerConfig)
fmt.Println(ServerConfig)
})
事例:
package main
import (
"fmt"
"time"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
)
type mysqlConfig struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
}
//ServerConfig 将配置文件内容映射为struct
type ServerConfig struct {
ServerName string `mapstructure:"name"`
SqlConfig mysqlConfig `mapstructure:"mysql"`
}
func main() {
//new对象
v := viper.New()
//读取配置文件
v.SetConfigFile("viper_test/ch1/config.yaml")
if err := v.ReadInConfig(); err != nil {
panic(err)
}
ServerConfig := ServerConfig{}
v.Unmarshal(&ServerConfig)
fmt.Println(ServerConfig)
v.WatchConfig()
v.OnConfigChange(func(e fsnotify.Event) {
fmt.Println("配置变化:", e)
_ = v.ReadInConfig()
v.Unmarshal(&ServerConfig)
fmt.Println(ServerConfig)
})
//延迟程序退出时间
time.Sleep(300 * time.Second)
}
运行程序后我们对port进行更改,会有如下输出:
{user-web {127.0.0.1 3306}}
配置变化: WRITE "viper_test/ch1/config.yaml"
{user-web {127.0.0.1 3303}}
配置变化: WRITE "viper_test/ch1/config.yaml"
{user-web {127.0.0.1 3306}}
线上线下环境隔离
目录结构:
ch02
├── config-debug.yaml //开发环境
├── config-por.yaml //线上环境
└── main.go
config-debug.yaml:
name: 'user-web-debug'
mysql:
host: '127.0.0.1'
port: 3306
config-por.yaml:
name: 'user-web'
mysql:
host: '195.1.43.12'
port: 2000
根据条件读写线上\线下配置并做监控
package main
import (
"fmt"
"time"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
)
//将配置文件嵌套映射
type MysqlServerConfig struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
}
//将配置文件内容映射为struct
type ServerConfig struct {
ServerName string `mapstructure:"name"`
Mysql MysqlServerConfig `mapstructure:"mysql"`
}
//通过环境变量,将线上线下环境隔离
func GetEnvInfo(env string) bool {
viper.AutomaticEnv()
return viper.GetBool(env)
}
func main() {
debug := GetEnvInfo("MXSHOP_DEBUG")
configFilePrefix := "config"
//默认为线上配置
configFileName := fmt.Sprintf("viper_test/ch02/%s-por.yaml", configFilePrefix)
if debug {
configFileName = fmt.Sprintf("viper_test/ch02/%s-debug.yaml", configFilePrefix)
}
v := viper.New()
v.SetConfigFile(configFileName)
if err := v.ReadInConfig(); err != nil {
panic(err)
}
ServerConfig := ServerConfig{}
if err := v.Unmarshal(&ServerConfig); err != nil {
panic(err)
}
fmt.Println(ServerConfig)
v.WatchConfig()
v.OnConfigChange(func(e fsnotify.Event) {
fmt.Println("配置变化:", e)
_ = v.ReadInConfig()
v.Unmarshal(&ServerConfig)
fmt.Println(ServerConfig)
})
time.Sleep(300 * time.Second)
}
完,有需要或者有兴趣可以一试。
路人 在
php中使用hyperf框架调用讯飞星火大模型实现国内版chatgpt功能示例中评论 教程很详细,如果加个前端chatgpt对话页面就完美了..博主 在
科学上网翻墙之v2rayN-Core客户端免费公益节点使用教程中评论 @ mashrdn 多切换几个节点测试,免费ssr是没那么稳..mashrdn 在
科学上网翻墙之v2rayN-Core客户端免费公益节点使用教程中评论 V2rayn免费节点添加上去了,youtobe无法打开网页,是怎么回事..张伟 在
科学上网翻墙之v2rayN-Core客户端免费公益节点使用教程中评论 3q!有用,不过免费节点隔天就要去git上复制新的导进去..博主 在
科学上网翻墙访问Google , 上外网神器佛跳墙VPN(永久免费)使用流程步骤中评论 该篇教程已不能用了,告知大家,免的老有老铁问我!..
Copyright·© 2019 侯体宗版权所有·
粤ICP备20027696号