glenux-archive/spf13--viper
1
0
Fork 0
mirror of https://github.com/spf13/viper synced 2025-05-11 22:57:21 +00:00
Code Issues Projects Releases Packages Wiki Activity

add: write file with old config

Browse source
This commit is contained in:
olongfen 2020-09-14 18:00:38 +08:00
parent 9f5de68da1
commit f819f4304d
1 changed files with 18 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
viper.go
View file

@ -1447,13 +1447,13 @@ func (v *Viper) writeConfig(filename string, force bool) error {
flags |= os.O_EXCL flags |= os.O_EXCL
} }
q,_:=ioutil.ReadFile(filename) q, _ := ioutil.ReadFile(filename)
f, err := v.fs.OpenFile(filename, flags, v.configPermissions) f, err := v.fs.OpenFile(filename, flags, v.configPermissions)
if err != nil { if err != nil {
return err return err
} }
defer f.Close() defer f.Close()
if err := v.marshalWriter(f, configType,q); err != nil { if err := v.marshalWriter(f, configType, q); err != nil {
return err return err
} }
@ -1547,18 +1547,18 @@ func (v *Viper) unmarshalReader(in io.Reader, c map[string]interface{}) error {
} }
// Marshal a map into Writer. // Marshal a map into Writer.
func (v *Viper) marshalWriter(f afero.File, configType string,q []byte) error { func (v *Viper) marshalWriter(f afero.File, configType string, q []byte) error {
c := v.AllSettings() c := v.AllSettings()
switch configType { switch configType {
case "json": case "json":
d:= map[string]interface{}{} d := map[string]interface{}{}
// unmarshal old file data // unmarshal old file data
_ = json.Unmarshal(q,&d) _ = json.Unmarshal(q, &d)
if reflect.DeepEqual(c,d){ if reflect.DeepEqual(c, d) {
return nil return nil
} }
// decode to settings , use old file data // decode to settings , use old file data
_ = mapstructure.Decode(d,&c) _ = mapstructure.Decode(d, &c)
b, err := json.MarshalIndent(c, "", " ") b, err := json.MarshalIndent(c, "", " ")
if err != nil { if err != nil {
return ConfigMarshalError{err} return ConfigMarshalError{err}
@ -1611,15 +1611,15 @@ func (v *Viper) marshalWriter(f afero.File, configType string,q []byte) error {
} }
case "toml": case "toml":
d:= map[string]interface{}{} d := map[string]interface{}{}
// unmarshal old file data // unmarshal old file data
_ = toml.Unmarshal(q,&d) _ = toml.Unmarshal(q, &d)
if reflect.DeepEqual(c,d){ if reflect.DeepEqual(c, d) {
return nil return nil
} }
// decode to settings , use old file data // decode to settings , use old file data
_ = mapstructure.Decode(d,&c) _ = mapstructure.Decode(d, &c)
t, err := toml.Marshal(c) t, err := toml.Marshal(c)
if err != nil { if err != nil {
return ConfigMarshalError{err} return ConfigMarshalError{err}
@ -1630,14 +1630,14 @@ func (v *Viper) marshalWriter(f afero.File, configType string,q []byte) error {
} }
case "yaml", "yml": case "yaml", "yml":
d:= map[string]interface{}{} d := map[string]interface{}{}
// unmarshal old file data // unmarshal old file data
_ = yaml.Unmarshal(q,&d) _ = yaml.Unmarshal(q, &d)
if reflect.DeepEqual(c,d){ if reflect.DeepEqual(c, d) {
return nil return nil
} }
// decode to settings , use old file data // decode to settings , use old file data
_ = mapstructure.Decode(d,&c) _ = mapstructure.Decode(d, &c)
b, err := yaml.Marshal(c) b, err := yaml.Marshal(c)
if err != nil { if err != nil {
return ConfigMarshalError{err} return ConfigMarshalError{err}