diff options
author | Paramtamtam <[email protected]> | 2019-12-24 10:48:49 +0500 |
---|---|---|
committer | GitHub <[email protected]> | 2019-12-24 10:48:49 +0500 |
commit | fbd2022729ab7ffe4eee8ba8b7357ba179e4d010 (patch) | |
tree | fa6b66b18259282b2efe00a41c445a3fa40aa701 /cmd/util/config.go | |
parent | 3f7d8bc007a4efdd48bb16253c2686006f0f9cda (diff) | |
parent | 75a817edcff22e721dcc3a7fa5590b866f630403 (diff) |
Merge branch 'master' into master
Diffstat (limited to 'cmd/util/config.go')
-rw-r--r-- | cmd/util/config.go | 23 |
1 files changed, 20 insertions, 3 deletions
diff --git a/cmd/util/config.go b/cmd/util/config.go index 0a8d6005..08e01a89 100644 --- a/cmd/util/config.go +++ b/cmd/util/config.go @@ -1,6 +1,7 @@ package util import ( + "bytes" "fmt" "github.com/spf13/viper" "github.com/spiral/roadrunner/service" @@ -30,7 +31,7 @@ func (w *ConfigWrapper) Unmarshal(out interface{}) error { } // LoadConfig config and merge it's values with set of flags. -func LoadConfig(cfgFile string, path []string, name string, flags []string) (*ConfigWrapper, error) { +func LoadConfig(cfgFile string, path []string, name string, flags []string, jsonConfig string) (*ConfigWrapper, error) { cfg := viper.New() if cfgFile != "" { @@ -68,14 +69,13 @@ func LoadConfig(cfgFile string, path []string, name string, flags []string) (*Co // If a cfg file is found, read it in. if err := cfg.ReadInConfig(); err != nil { - if len(flags) == 0 { + if len(flags) == 0 && jsonConfig == "" { return nil, err } } // merge included configs if include, ok := cfg.Get("include").([]interface{}); ok { - for _, file := range include { filename, ok := file.(string) if !ok { @@ -117,6 +117,23 @@ func LoadConfig(cfgFile string, path []string, name string, flags []string) (*Co } } + if jsonConfig != "" { + jConfig := viper.New() + jConfig.AutomaticEnv() + jConfig.SetEnvPrefix("rr") + jConfig.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) + + jConfig.SetConfigType("json") + if err := jConfig.ReadConfig(bytes.NewBufferString(jsonConfig)); err != nil { + return nil, err + } + + // merging + if err := cfg.MergeConfigMap(jConfig.AllSettings()); err != nil { + return nil, err + } + } + merged := viper.New() // we have to copy all the merged values into new config in order normalize it (viper bug?) |