golang 使用 viper 讀取自定義配置檔案

viper 支援Yaml、Json、 TOML、HCL 等格式，讀取非常的方便。
viper官網有案例:https://github.com/spf13/viper

go get github.com/spf13/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)

本作品採用《CC 協議》，轉載必須註明作者和本文連結