39 lines
697 B
Go
39 lines
697 B
Go
package core
|
|
|
|
import (
|
|
"ai4m/core/internal"
|
|
"ai4m/global"
|
|
"fmt"
|
|
"github.com/fsnotify/fsnotify"
|
|
"github.com/spf13/viper"
|
|
"os"
|
|
)
|
|
|
|
func Viper() *viper.Viper {
|
|
config := internal.ConfigDefaultFile
|
|
getwd, _ := os.Getwd()
|
|
|
|
println(getwd)
|
|
v := viper.New()
|
|
v.SetConfigFile(config)
|
|
v.SetConfigType("yaml")
|
|
err := v.ReadInConfig()
|
|
if err != nil {
|
|
panic(fmt.Errorf("Fatal error config file: %s \n", err))
|
|
}
|
|
v.WatchConfig()
|
|
|
|
v.OnConfigChange(func(e fsnotify.Event) {
|
|
fmt.Println("config file changed:", e.Name)
|
|
if err = v.Unmarshal(&global.PCM_CONFIG); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
})
|
|
|
|
if err = v.Unmarshal(&global.PCM_CONFIG); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
return v
|
|
}
|