Browse Source

Merge branch 'develop' of https://github.com/piexlmax/QMPlus into gva_gormv2_dev

main
pixel 4 years ago
parent
commit
a16646c543
  1. 16
      server/core/config.go

16
server/core/config.go

@ -6,13 +6,27 @@ import (
_ "gin-vue-admin/packfile" _ "gin-vue-admin/packfile"
"github.com/fsnotify/fsnotify" "github.com/fsnotify/fsnotify"
"github.com/spf13/viper" "github.com/spf13/viper"
"github.com/spf13/pflag"
) )
const defaultConfigFile = "config.yaml" const defaultConfigFile = "config.yaml"
func init() { func init() {
pflag.StringP("configFile","c", "", "choose config file.")
pflag.Parse()
// 优先级: 命令行 > 环境变量 > 默认值
v := viper.New() v := viper.New()
v.SetConfigFile(defaultConfigFile)
v.BindPFlags(pflag.CommandLine)
v.SetEnvPrefix("gva")
v.BindEnv("configFile") // GVA_CONFIGFILE
configFile := v.GetString("configFile")
if configFile == ""{
configFile = defaultConfigFile
}
v.SetConfigFile(configFile)
err := v.ReadInConfig() err := v.ReadInConfig()
if err != nil { if err != nil {
panic(fmt.Errorf("Fatal error config file: %s \n", err)) panic(fmt.Errorf("Fatal error config file: %s \n", err))

Loading…
Cancel
Save