如何在golang中使用viper读取自定义配置文件

如何在golang中使用viper读取自定义配置文件？针对这个问题，这篇文章详细介绍了相对应的分析和解答，希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。

创建 config.yaml 文件

database:
 driver: mysql
 host: 127.0.0.1
 port: 3306
 username: blog
 dbname: blog
 password: 123456

建一个 config.go 用于初始化配置文件

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)
  }
}

简单使用:

  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)

关于如何在golang中使用viper读取自定义配置文件问题的解答就分享到这里了，希望以上内容可以对大家有一定的帮助，如果你还有很多疑惑没有解开，可以关注亿速云行业资讯频道了解更多相关知识。