provider with config type

This commit is contained in:
yoyofx
2021-05-12 13:56:58 +08:00
parent 59ac25d85d
commit 66c4e16397
5 changed files with 41 additions and 5 deletions

View File

@@ -6,17 +6,19 @@ import (
)
type ViperRemoteProvider struct {
configSet string
configType string
configSet string
}
func NewRemoteProvider() *ViperRemoteProvider {
func NewRemoteProvider(configType string) *ViperRemoteProvider {
return &ViperRemoteProvider{
configSet: "yoyogo.cloud.discovery.metadata"}
configType: configType,
configSet: "yoyogo.cloud.discovery.metadata"}
}
func (provider *ViperRemoteProvider) GetProvider(runtime_viper *viper.Viper) *viper.Viper {
var option *Option
err := runtime_viper.Sub("yoyogo.cloud.discovery.metadata").Unmarshal(&option)
err := runtime_viper.Sub(provider.configSet).Unmarshal(&option)
if err != nil {
panic(err)
return nil
@@ -24,7 +26,10 @@ func (provider *ViperRemoteProvider) GetProvider(runtime_viper *viper.Viper) *vi
SetOptions(option)
remote_viper := viper.New()
err = remote_viper.AddRemoteProvider("nacos", "localhost", "")
remote_viper.SetConfigType("yaml")
if provider.configType == "" {
provider.configType = "yaml"
}
remote_viper.SetConfigType(provider.configType)
err = remote_viper.ReadRemoteConfig()
if err == nil {
err = remote_viper.WatchRemoteConfigOnChannel()