51 lines
733 B
Go
51 lines
733 B
Go
|
package util
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Mysql struct {
|
||
|
|
User string
|
||
|
|
Password string
|
||
|
|
Host string
|
||
|
|
Port int
|
||
|
|
Db string
|
||
|
|
}
|
||
|
|
|
||
|
|
type Redis struct {
|
||
|
|
Host string
|
||
|
|
Port int
|
||
|
|
Db int
|
||
|
|
}
|
||
|
|
|
||
|
|
type AppConfig struct {
|
||
|
|
Mysql Mysql
|
||
|
|
Redis Redis
|
||
|
|
}
|
||
|
|
|
||
|
|
var Config *AppConfig
|
||
|
|
|
||
|
|
func InitConfig() {
|
||
|
|
Config = &AppConfig{}
|
||
|
|
v := viper.New()
|
||
|
|
v.SetConfigName("config")
|
||
|
|
wd, _ := os.Getwd()
|
||
|
|
v.AddConfigPath(wd)
|
||
|
|
v.SetConfigType("yml")
|
||
|
|
|
||
|
|
if err := v.ReadInConfig(); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
envConfig := os.Getenv("CONFIG_FILE_NAME")
|
||
|
|
if len(envConfig) > 0 {
|
||
|
|
v.SetConfigName(envConfig)
|
||
|
|
if err := v.ReadInConfig(); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if err := v.Unmarshal(Config); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|