viper 支持 Yaml、Json、 TOML、HCL 等格式,读取非常的方便。
viper 官网有案例:https://github.com/spf13/viper
1
|
go get github.com/spf13/viper |
创建 config.yaml 文件
1
2
3
4
5
6
7
|
database: driver: mysql host: 127.0.0.1 port: 3306 username: blog dbname: blog password: 123456 |
建一个 config.go 用于初始化配置文件
1
2
3
4
5
6
7
8
9
10
11
12
|
func InitConfig() { path, err := os.Getwd() if err != nil { panic(err) } viper.AddConfigPath(path + "/config/dev") viper.SetConfigName("config") viper.SetConfigType("yaml") if err := viper.ReadInConfig(); err != nil { panic(err) } } |
简单使用:
1
2
3
4
5
6
7
|
username := viper.GetString("database.username") password := viper.GetString("database.password") host := viper.GetString("database.host") port := viper.GetInt("database.port") dbname := viper.GetString("database.dbname") dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local",username,password,host, port, dbname) GormPool, err = gorm.Open("mysql", dsn) |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://learnku.com/articles/39558