|
@@ -9,8 +9,11 @@ import (
|
|
)
|
|
)
|
|
|
|
|
|
func Initialize(path string, cfgStruct interface{}) error {
|
|
func Initialize(path string, cfgStruct interface{}) error {
|
|
- dir := fmt.Sprintf("%s/config/%s", os.Getenv("GO_WORK_DIR"), path)
|
|
|
|
- // dir = fmt.Sprintf("D:\\project\\golangNew\\kpt-tmr-group\\config\\%s", path)
|
|
|
|
|
|
+ workDir := os.Getenv("GO_WORK_DIR")
|
|
|
|
+ if workDir == "" {
|
|
|
|
+ workDir = "./"
|
|
|
|
+ }
|
|
|
|
+ dir := fmt.Sprintf("%s/config/%s", workDir, path)
|
|
viper.SetConfigType("yaml")
|
|
viper.SetConfigType("yaml")
|
|
viper.SetConfigFile(dir)
|
|
viper.SetConfigFile(dir)
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
if err := viper.ReadInConfig(); err != nil {
|