mirror of
https://github.com/spf13/viper
synced 2024-12-22 19:47:01 +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)
|
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 {
|
||||||
|
|
Loading…
Reference in a new issue