mirror of https://github.com/spf13/viper.git
re-implemented WatchRemoteConfigOnChannel()
Now WatchRemoteConfigOnChannel() returns '<-chan bool'. This channel signals the change.
This commit is contained in:
parent
71509d2887
commit
cf9182539d
15
viper.go
15
viper.go
|
@ -1639,7 +1639,7 @@ func (v *Viper) WatchRemoteConfig() error {
|
||||||
return v.watchKeyValueConfig()
|
return v.watchKeyValueConfig()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (v *Viper) WatchRemoteConfigOnChannel() error {
|
func (v *Viper) WatchRemoteConfigOnChannel() (<-chan bool, error) {
|
||||||
return v.watchKeyValueConfigOnChannel()
|
return v.watchKeyValueConfigOnChannel()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1670,20 +1670,23 @@ func (v *Viper) getRemoteConfig(provider RemoteProvider) (map[string]interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Retrieve the first found remote configuration.
|
// Retrieve the first found remote configuration.
|
||||||
func (v *Viper) watchKeyValueConfigOnChannel() error {
|
func (v *Viper) watchKeyValueConfigOnChannel() (<-chan bool, error) {
|
||||||
|
wCh := make(chan bool)
|
||||||
for _, rp := range v.remoteProviders {
|
for _, rp := range v.remoteProviders {
|
||||||
respc, _ := RemoteConfig.WatchChannel(rp)
|
respc, _ := RemoteConfig.WatchChannel(rp)
|
||||||
//Todo: Add quit channel
|
//Todo: Add quit channel
|
||||||
go func(rc <-chan *RemoteResponse) {
|
go func(rc <-chan *RemoteResponse, rc2 chan<- bool) {
|
||||||
for {
|
for {
|
||||||
b := <-rc
|
b := <-rc
|
||||||
reader := bytes.NewReader(b.Value)
|
reader := bytes.NewReader(b.Value)
|
||||||
v.unmarshalReader(reader, v.kvstore)
|
v.unmarshalReader(reader, v.kvstore)
|
||||||
|
rc2 <- true
|
||||||
}
|
}
|
||||||
}(respc)
|
}(respc, wCh)
|
||||||
return nil
|
return wCh, nil
|
||||||
}
|
}
|
||||||
return RemoteConfigError("No Files Found")
|
close(wCh)
|
||||||
|
return nil, RemoteConfigError("No Files Found")
|
||||||
}
|
}
|
||||||
|
|
||||||
// Retrieve the first found remote configuration.
|
// Retrieve the first found remote configuration.
|
||||||
|
|
Loading…
Reference in New Issue