read config from env

This commit is contained in:
wfhtqp
2020-12-26 11:39:54 +08:00
committed by dushixiang
parent e61f19233c
commit 712e2cfe84

View File

@ -1,9 +1,11 @@
package config package config
import ( import (
"log"
"strings"
"github.com/spf13/pflag" "github.com/spf13/pflag"
"github.com/spf13/viper" "github.com/spf13/viper"
"log"
) )
type Config struct { type Config struct {
@ -30,6 +32,8 @@ func SetupConfig() *Config {
viper.AddConfigPath("/etc/next-terminal/") viper.AddConfigPath("/etc/next-terminal/")
viper.AddConfigPath("$HOME/.next-terminal") viper.AddConfigPath("$HOME/.next-terminal")
viper.AddConfigPath(".") viper.AddConfigPath(".")
viper.AutomaticEnv()
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
//pflag.String("mysql.hostname", "127.0.0.1", "mysql hostname") //pflag.String("mysql.hostname", "127.0.0.1", "mysql hostname")
//pflag.Int("mysql.port", 3306, "mysql port") //pflag.Int("mysql.port", 3306, "mysql port")