add:support file extension (#1626)

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
 }
 

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
 }

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)
+}