[dubbo-go] branch 3.0 updated: add:support file extension (#1626)

[laurence]

This is an automated email from the ASF dual-hosted git repository.

laurence pushed a commit to branch 3.0
in repository https://gitbox.apache.org/repos/asf/dubbo-go.git


The following commit(s) were added to refs/heads/3.0 by this push:
     new f7f1581  add:support file extension (#1626)
f7f1581 is described below

commit f7f15810cef0a85d4e538170bb3f1ea04502df11
Author: zhaoyunxing <zhaoyunx...@apache.org>
AuthorDate: Thu Dec 2 11:13:34 2021 +0800

    add:support file extension (#1626)
---
 config/config_center_config.go       | 15 ++++++---------
 config/config_loader_options.go      | 12 +++++++++---
 config/config_loader_options_test.go | 11 +++++++++++
 3 files changed, 26 insertions(+), 12 deletions(-)

diff --git a/config/config_center_config.go b/config/config_center_config.go
index 00cb05b..778e644 100644
--- a/config/config_center_config.go
+++ b/config/config_center_config.go
@@ -27,8 +27,6 @@ import (
        "github.com/creasty/defaults"
 
        "github.com/knadh/koanf"
-       "github.com/knadh/koanf/parsers/yaml"
-       "github.com/knadh/koanf/providers/rawbytes"
 
        "github.com/pkg/errors"
 )
@@ -61,6 +59,9 @@ type CenterConfig struct {
        AppID     string            `default:"dubbo" yaml:"app-id"  
json:"app-id,omitempty"`
        Timeout   string            `default:"10s" yaml:"timeout"  
json:"timeout,omitempty"`
        Params    map[string]string `yaml:"params"  json:"parameters,omitempty"`
+
+       //FileExtension the suffix of config dataId, also the file extension of 
config content
+       FileExtension string `default:"yaml" yaml:"file-extension" 
json:"file-extension" `
 }
 
 // Prefix dubbo.config-center
@@ -146,15 +147,11 @@ func startConfigCenter(rc *RootConfig) error {
                        "Please check if your config-center config is 
correct.", cc)
                return nil
        }
-       koan := koanf.New(".")
-       if err = koan.Load(rawbytes.Provider([]byte(strConf)), yaml.Parser()); 
err != nil {
-               return err
-       }
-       if err = koan.UnmarshalWithConf(rc.Prefix(),
-               rc, koanf.UnmarshalConf{Tag: "yaml"}); err != nil {
+       config := NewLoaderConf(WithDelim("."), WithGenre(cc.FileExtension), 
WithBytes([]byte(strConf)))
+       koan := GetConfigResolver(config)
+       if err = koan.UnmarshalWithConf(rc.Prefix(), rc, 
koanf.UnmarshalConf{Tag: "yaml"}); err != nil {
                return err
        }
-
        return nil
 }
 
diff --git a/config/config_loader_options.go b/config/config_loader_options.go
index 4994103..92ab4ea 100644
--- a/config/config_loader_options.go
+++ b/config/config_loader_options.go
@@ -18,7 +18,6 @@
 package config
 
 import (
-       "fmt"
        "io/ioutil"
        "os"
        "path/filepath"
@@ -122,6 +121,13 @@ func WithDelim(delim string) LoaderConfOption {
        })
 }
 
+// WithBytes set load config  bytes
+func WithBytes(bytes []byte) LoaderConfOption {
+       return loaderConfigFunc(func(conf *loaderConf) {
+               conf.bytes = bytes
+       })
+}
+
 // absolutePath get absolut path
 func absolutePath(inPath string) string {
 
@@ -155,11 +161,11 @@ func userHomeDir() string {
 
 // checkGenre check Genre
 func checkGenre(genre string) error {
-       genres := []string{"json", "toml", "yaml", "yml"}
+       genres := []string{"json", "toml", "yaml", "yml", "properties"}
        sort.Strings(genres)
        idx := sort.SearchStrings(genres, genre)
        if genres[idx] != genre {
-               return errors.New(fmt.Sprintf("no support %s", genre))
+               return errors.Errorf("no support file extension: %s", genre)
        }
        return nil
 }
diff --git a/config/config_loader_options_test.go 
b/config/config_loader_options_test.go
index 8943cc3..0ba55e0 100644
--- a/config/config_loader_options_test.go
+++ b/config/config_loader_options_test.go
@@ -44,3 +44,14 @@ func TestRootConfig(t *testing.T) {
        conf := NewLoaderConf(WithRootConfig(rc))
        assert.Equal(t, conf.rc.Application.Name, "test-app")
 }
+
+func TestNewLoaderConf_WithBytes(t *testing.T) {
+       str := `dubbo.application.name=dubbo-go
+dubbo.application.module=local
+dubbo.services.HelloService.registry=nacos,zk`
+
+       conf := NewLoaderConf(WithBytes([]byte(str)), WithGenre("properties"))
+
+       assert.NotNil(t, conf)
+       assert.NotNil(t, conf.bytes)
+}