Browse Source

Merge pull request #208 from maplepie/master

Update config.go
main
奇淼(piexlmax 4 years ago
committed by GitHub
parent
commit
7764f8d22f
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
  1. 16
      server/core/config.go

16
server/core/config.go

@ -6,13 +6,27 @@ import (
_ "gin-vue-admin/packfile"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
"github.com/spf13/pflag"
)
const defaultConfigFile = "config.yaml"
func init() {
pflag.StringP("configFile","c", "", "choose config file.")
pflag.Parse()
// 优先级: 命令行 > 环境变量 > 默认值
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()
if err != nil {
panic(fmt.Errorf("Fatal error config file: %s \n", err))

Loading…
Cancel
Save