mirror of
https://github.com/spf13/viper
synced 2024-12-22 19:47:01 +00:00
Added method ReadConfigNoNil
This patch adds a new method called ReadConfigNoNil that will read in a configuration file, but only set existing keys if they exist in the provided file.
This commit is contained in:
parent
a7ef020a9a
commit
4d238ec572
2 changed files with 51 additions and 0 deletions
31
viper.go
31
viper.go
|
@ -805,12 +805,43 @@ func (v *Viper) ReadInConfig() error {
|
||||||
return v.unmarshalReader(bytes.NewReader(file), v.config)
|
return v.unmarshalReader(bytes.NewReader(file), v.config)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Viper will discover and load the configuration file from disk
|
||||||
|
// and key/value stores, searching in one of the defined paths.
|
||||||
|
// This method will only set existing keys if they exist in the
|
||||||
|
// provided file.
|
||||||
|
func ReadInConfigNoNil() error { return v.ReadInConfigNoNil() }
|
||||||
|
func (v *Viper) ReadInConfigNoNil() error {
|
||||||
|
jww.INFO.Println("Attempting to read in config file")
|
||||||
|
if !stringInSlice(v.getConfigType(), SupportedExts) {
|
||||||
|
return UnsupportedConfigError(v.getConfigType())
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := ioutil.ReadFile(v.getConfigFile())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return v.ReadConfigNoNil(bytes.NewReader(file))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Viper will read a configuration file, setting existing keys to nil if the
|
||||||
|
// key does not exist in the file.
|
||||||
func ReadConfig(in io.Reader) error { return v.ReadConfig(in) }
|
func ReadConfig(in io.Reader) error { return v.ReadConfig(in) }
|
||||||
func (v *Viper) ReadConfig(in io.Reader) error {
|
func (v *Viper) ReadConfig(in io.Reader) error {
|
||||||
v.config = make(map[string]interface{})
|
v.config = make(map[string]interface{})
|
||||||
return v.unmarshalReader(in, v.config)
|
return v.unmarshalReader(in, v.config)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Viper will read a configuration file, but only set existing keys if they
|
||||||
|
// exist in the provided file.
|
||||||
|
func ReadConfigNoNil(in io.Reader) error { return v.ReadConfigNoNil(in) }
|
||||||
|
func (v *Viper) ReadConfigNoNil(in io.Reader) error {
|
||||||
|
if v.config == nil {
|
||||||
|
v.config = make(map[string]interface{})
|
||||||
|
}
|
||||||
|
return v.unmarshalReader(in, v.config)
|
||||||
|
}
|
||||||
|
|
||||||
// func ReadBufConfig(buf *bytes.Buffer) error { return v.ReadBufConfig(buf) }
|
// func ReadBufConfig(buf *bytes.Buffer) error { return v.ReadBufConfig(buf) }
|
||||||
// func (v *Viper) ReadBufConfig(buf *bytes.Buffer) error {
|
// func (v *Viper) ReadBufConfig(buf *bytes.Buffer) error {
|
||||||
// v.config = make(map[string]interface{})
|
// v.config = make(map[string]interface{})
|
||||||
|
|
|
@ -37,6 +37,13 @@ eyes : brown
|
||||||
beard: true
|
beard: true
|
||||||
`)
|
`)
|
||||||
|
|
||||||
|
var yamlExample2 = []byte(`Hacker: false
|
||||||
|
name: robert
|
||||||
|
age: 40
|
||||||
|
eyes : blue
|
||||||
|
height: 5' 8"
|
||||||
|
`)
|
||||||
|
|
||||||
var tomlExample = []byte(`
|
var tomlExample = []byte(`
|
||||||
title = "TOML Example"
|
title = "TOML Example"
|
||||||
|
|
||||||
|
@ -255,6 +262,19 @@ func TestYML(t *testing.T) {
|
||||||
assert.Equal(t, "steve", Get("name"))
|
assert.Equal(t, "steve", Get("name"))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestReadConfigNoNil(t *testing.T) {
|
||||||
|
initYAML()
|
||||||
|
assert.Equal(t, "steve", Get("name"))
|
||||||
|
|
||||||
|
r := bytes.NewReader(yamlExample2)
|
||||||
|
|
||||||
|
v.ReadConfigNoNil(r)
|
||||||
|
assert.Equal(t, "robert", Get("name"))
|
||||||
|
assert.Equal(t, 40, Get("age"))
|
||||||
|
assert.Equal(t, true, Get("beard"))
|
||||||
|
assert.Equal(t, "5' 8\"", Get("height"))
|
||||||
|
}
|
||||||
|
|
||||||
func TestJSON(t *testing.T) {
|
func TestJSON(t *testing.T) {
|
||||||
initJSON()
|
initJSON()
|
||||||
assert.Equal(t, "0001", Get("id"))
|
assert.Equal(t, "0001", Get("id"))
|
||||||
|
|
Loading…
Reference in a new issue