67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
|
package nacos_viper_remote
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"errors"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
var nacosOptions = &Option{}
|
||
|
|
|
||
|
|
func SetOptions(option *Option) {
|
||
|
|
nacosOptions = option
|
||
|
|
}
|
||
|
|
|
||
|
|
type remoteConfigProvider struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rc *remoteConfigProvider) Get(rp viper.RemoteProvider) (io.Reader, error) {
|
||
|
|
cmt, err := getConfigManager(rp)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var b []byte
|
||
|
|
switch cm := cmt.(type) {
|
||
|
|
case viperConfigManager:
|
||
|
|
b, err = cm.Get(rp.Path())
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return bytes.NewReader(b), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rc *remoteConfigProvider) Watch(rp viper.RemoteProvider) (io.Reader, error) {
|
||
|
|
return rc.Get(rp)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rc *remoteConfigProvider) WatchChannel(rp viper.RemoteProvider) (<-chan *viper.RemoteResponse, chan bool) {
|
||
|
|
cmt, err := getConfigManager(rp)
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
switch cm := cmt.(type) {
|
||
|
|
case viperConfigManager:
|
||
|
|
quit := make(chan bool)
|
||
|
|
viperResponseCh := cm.Watch("dataId", quit)
|
||
|
|
return viperResponseCh, quit
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func getConfigManager(rp viper.RemoteProvider) (interface{}, error) {
|
||
|
|
if rp.Provider() == "nacos" {
|
||
|
|
return NewNacosConfigManager(nacosOptions)
|
||
|
|
} else {
|
||
|
|
return nil, errors.New("The Nacos configuration manager is not supported!")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
viper.SupportedRemoteProviders = []string{"nacos"}
|
||
|
|
viper.RemoteConfig = &remoteConfigProvider{}
|
||
|
|
}
|