mirror of
https://github.com/spf13/viper
synced 2024-12-22 03:27:03 +00:00
add MarshalToString
This commit is contained in:
parent
e71d7bf15c
commit
b3171c42b6
1 changed files with 48 additions and 0 deletions
48
viper.go
48
viper.go
|
@ -1601,6 +1601,11 @@ func (v *Viper) SafeWriteConfigAs(filename string) error {
|
|||
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 {
|
||||
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()
|
||||
}
|
||||
|
||||
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 {
|
||||
buf := new(bytes.Buffer)
|
||||
buf.ReadFrom(in)
|
||||
|
@ -1683,6 +1712,25 @@ func (v *Viper) marshalWriter(f afero.File, configType string) error {
|
|||
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 {
|
||||
lk := strings.ToLower(k)
|
||||
for mk := range m {
|
||||
|
|
Loading…
Reference in a new issue