You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.9 KiB
80 lines
1.9 KiB
package config |
|
|
|
import ( |
|
"errors" |
|
"flag" |
|
"github.com/spf13/viper" |
|
"os" |
|
"sonet/pkg/utils/logger" |
|
"strings" |
|
) |
|
|
|
func parseConfPathFlag(confPath string) (filePath, fileName, confName, confType string) { |
|
idx := strings.LastIndex(confPath, "/") |
|
filePath = confPath[:idx+1] |
|
fileName = confPath[idx+1:] |
|
idx2 := strings.LastIndex(fileName, ".") |
|
confName = fileName[:idx2] |
|
confType = fileName[idx2+1:] |
|
return |
|
} |
|
|
|
// LoadConfig load config.toml |
|
// appConf service custom config |
|
// return common service configuration |
|
func LoadConfig(appConf any, confPathArg ...string) *Configuration { |
|
confPath := "" |
|
confName := "config" |
|
confType := "toml" |
|
envPrefix := "SO" |
|
|
|
if len(confPathArg) > 0 { |
|
confPath = confPathArg[0] |
|
} |
|
|
|
// go run xx -conf=config/xx.toml |
|
confPathFlag := flag.String("conf", "", "config file path.") |
|
envPrefixFlag := flag.String("envPrefix", "", "env config key prefix.") |
|
flag.Parse() |
|
if *confPathFlag != "" { |
|
confPath, _, confName, confType = parseConfPathFlag(*confPathFlag) |
|
} |
|
if *envPrefixFlag != "" { |
|
envPrefix = *envPrefixFlag |
|
} |
|
if confPath == "" { |
|
confPath = "./" |
|
} |
|
logger.Infof("use config file: %s%s.%s, env prefix=%s\n", confPath, confName, confType, envPrefix) |
|
|
|
viper.AddConfigPath(confPath) |
|
viper.SetConfigName(confName) |
|
viper.SetConfigType(confType) |
|
viper.SetEnvPrefix(envPrefix) |
|
viper.AutomaticEnv() |
|
viper.AllowEmptyEnv(true) |
|
|
|
conf := &Configuration{} |
|
|
|
if err := viper.ReadInConfig(); err != nil { |
|
panic(errors.New("viper read config fail: " + err.Error())) |
|
} |
|
if err := viper.Unmarshal(conf); err != nil { |
|
panic(errors.New("viper unmarshal config failed: " + err.Error())) |
|
} |
|
if appConf != nil && len(conf.App) > 0 { |
|
if err := viper.UnmarshalKey("app", appConf); err != nil { |
|
panic(errors.New("viper unmarshal app config failed: " + err.Error())) |
|
} |
|
} |
|
|
|
return conf |
|
} |
|
|
|
func LoadIdlPath(idlPath string) (string, error) { |
|
idl, err := os.ReadFile(idlPath) |
|
if err != nil { |
|
return "", err |
|
} |
|
return string(idl), nil |
|
}
|
|
|