add MarshalToString

This commit is contained in:
lin 2024-06-27 18:06:10 +08:00
parent e71d7bf15c
commit b3171c42b6
1 changed files with 48 additions and 0 deletions

View File

@ -1601,6 +1601,11 @@ func (v *Viper) SafeWriteConfigAs(filename string) error {
return v.writeConfig(filename, false) return v.writeConfig(filename, false)
} }
// MarshalToString writes current configuration to a string
func MarshalToString(configTypeOps ...string) (string, error) {
return v.MarshalToString(configTypeOps...)
}
func (v *Viper) writeConfig(filename string, force bool) error { func (v *Viper) writeConfig(filename string, force bool) error {
v.logger.Info("attempting to write configuration to file") v.logger.Info("attempting to write configuration to file")
@ -1639,6 +1644,30 @@ func (v *Viper) writeConfig(filename string, force bool) error {
return f.Sync() return f.Sync()
} }
func (v *Viper) MarshalToString(configTypeOps ...string) (string, error) {
v.logger.Info("MarshalToString: attempting to marshal configuration to string")
//default by origin configType
var configType = v.configType
//set from param if configTypeOps len > 0
if len(configTypeOps) > 0 {
configType = configTypeOps[0]
}
if configType == "" {
return "", fmt.Errorf("MarshalToString: config type could not be determined")
}
if !stringInSlice(configType, SupportedExts) {
return "", UnsupportedConfigError(configType)
}
str, err := v.marshalToString(configType)
if err != nil {
return "", err
}
return str, nil
}
func (v *Viper) unmarshalReader(in io.Reader, c map[string]any) error { func (v *Viper) unmarshalReader(in io.Reader, c map[string]any) error {
buf := new(bytes.Buffer) buf := new(bytes.Buffer)
buf.ReadFrom(in) buf.ReadFrom(in)
@ -1683,6 +1712,25 @@ func (v *Viper) marshalWriter(f afero.File, configType string) error {
return nil return nil
} }
// Marshal a map into string.
func (v *Viper) marshalToString(configType string) (string, error) {
c := v.AllSettings()
switch configType {
case "yaml", "yml", "json", "toml", "hcl", "tfvars", "ini", "prop", "props", "properties", "dotenv", "env":
encoder, err := v.encoderRegistry.Encoder(configType)
if err != nil {
return "", ConfigMarshalError{err}
}
b, err := encoder.Encode(c)
if err != nil {
return "", ConfigMarshalError{err}
}
return string(b), nil
}
return "", ConfigMarshalError{fmt.Errorf("config type could not be determined, type:%s", configType)}
}
func keyExists(k string, m map[string]any) string { func keyExists(k string, m map[string]any) string {
lk := strings.ToLower(k) lk := strings.ToLower(k)
for mk := range m { for mk := range m {