golang 使用 viper 读取自定义配置文件

时间:2022-09-18 20:09:21

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