glenux-archive/spf13--viper
1
0
Fork 0
mirror of https://github.com/spf13/viper synced 2025-05-06 20:27:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix toml return count error

Browse source
This commit is contained in:
Herkermer Sherwood 2017-04-23 14:00:53 -07:00
parent c588a0c383
commit 4b195d0ed8
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
viper.go
View file

@ -1365,20 +1365,21 @@ func (v *Viper) marshalWriter(f afero.File, configType string) error {
}
case "toml":
t := toml.TreeFromMap(c)
s := t.String()
_, err := f.WriteString(s)
t, err := toml.TreeFromMap(c)
if err != nil {
return ConfigMarshalError{err}
}
s := t.String()
if _, err := f.WriteString(s); err != nil {
return ConfigMarshalError{err}
}
case "yaml", "yml":
b, err := yaml.Marshal(c)
if err != nil {
return ConfigMarshalError{err}
}
_, err = f.WriteString(string(b))
if err != nil {
if _, err = f.WriteString(string(b)); err != nil {
return ConfigMarshalError{err}
}
}