glenux-archive/spf13--viper
1
0
Fork 0
mirror of https://github.com/spf13/viper synced 2024-12-23 03:57:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

replace bytes.Buffer with io.Reader

Browse source
This commit is contained in:
oliveagle 2015-05-14 17:40:59 +08:00
parent 9c9f7f12c8
commit 5736fabdfa
2 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
viper.go
View file

@ -716,13 +716,20 @@ func (v *Viper) ReadInConfig() error {
return nil return nil
} }
func ReadBufConfig(buf *bytes.Buffer) error { return v.ReadBufConfig(buf) } func ReadConfig(in io.Reader) error { return v.ReadConfig(in) }
func (v *Viper) ReadBufConfig(buf *bytes.Buffer) error { func (v *Viper) ReadConfig(in io.Reader) error {
v.config = make(map[string]interface{}) v.config = make(map[string]interface{})
v.marshalReader(buf, v.config) v.marshalReader(in, v.config)
return nil return nil
} }
// func ReadBufConfig(buf *bytes.Buffer) error { return v.ReadBufConfig(buf) }
// func (v *Viper) ReadBufConfig(buf *bytes.Buffer) error {
// v.config = make(map[string]interface{})
// v.marshalReader(buf, v.config)
// return nil
// }
// Attempts to get configuration from a remote source // Attempts to get configuration from a remote source
// and read it in the remote configuration registry. // and read it in the remote configuration registry.
func ReadRemoteConfig() error { return v.ReadRemoteConfig() } func ReadRemoteConfig() error { return v.ReadRemoteConfig() }

2
viper_test.go
View file

@ -541,7 +541,7 @@ func TestFindsNestedKeys(t *testing.T) {
func TestReadBufConfig(t *testing.T) { func TestReadBufConfig(t *testing.T) {
v := New() v := New()
v.SetConfigType("yaml") v.SetConfigType("yaml")
v.ReadBufConfig(bytes.NewBuffer(yamlExample)) v.ReadConfig(bytes.NewBuffer(yamlExample))
t.Log(v.AllKeys()) t.Log(v.AllKeys())
assert.True(t, v.InConfig("name")) assert.True(t, v.InConfig("name"))