glenux-archive/spf13--viper
1
0
Fork 0
mirror of https://github.com/spf13/viper synced 2025-05-07 20:57:18 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix watch register.

Browse source
This commit is contained in:
cyprinus 2023-06-04 11:36:57 +08:00
parent e8a67492a6
commit f04ffd899d
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
viper.go
View file

@ -478,15 +478,11 @@ func (v *Viper) WatchConfig() {
(event.Has(fsnotify.Write) || event.Has(fsnotify.Create))) ||
(currentConfigFile != "" && currentConfigFile != realConfigFile) {
realConfigFile = currentConfigFile
tempViper := New()
tempViper.AddConfigPath(realConfigFile)
err := tempViper.ReadInConfig()
err := v.ReadInConfig()
if err != nil {
log.Printf("error reading config file: %v\n", err)
}
v.updateRegisteredConfig(tempViper)
if v.onConfigChange != nil {
v.onConfigChange(event)
}
@ -1623,11 +1619,13 @@ func (v *Viper) ReadInConfig() error {
}
config := make(map[string]interface{})
err = v.unmarshalReader(bytes.NewReader(file), config)
tempViper := New()
tempViper.configType = v.getConfigType()
err = tempViper.unmarshalReader(bytes.NewReader(file), config)
if err != nil {
return err
}
v.updateRegisteredConfig(tempViper)
v.config = config
return nil
@ -1982,7 +1980,10 @@ func (v *Viper) getRemoteConfig(provider RemoteProvider) (map[string]interface{}
if err != nil {
return nil, err
}
err = v.unmarshalReader(reader, v.kvstore)
tempViper := New()
tempViper.configType = v.getConfigType()
err = tempViper.unmarshalReader(reader, tempViper.kvstore)
v.updateRegisteredConfig(tempViper)
return v.kvstore, err
}
@ -1999,6 +2000,7 @@ func (v *Viper) watchKeyValueConfigOnChannel() error {
for {
b := <-rc
tempViper := New()
tempViper.configType = v.getConfigType()
reader := bytes.NewReader(b.Value)
tempViper.unmarshalReader(reader, tempViper.kvstore)
v.updateRegisteredConfig(tempViper)