50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package envutil
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/gookit/goutil/internal/comfunc"
|
|
)
|
|
|
|
// ValueGetter Env value provider func.
|
|
//
|
|
// TIPS: you can custom provide data.
|
|
var ValueGetter = os.Getenv
|
|
|
|
// VarReplace replaces ${var} or $var in the string according to the values.
|
|
//
|
|
// is alias of the os.ExpandEnv()
|
|
func VarReplace(s string) string { return os.ExpandEnv(s) }
|
|
|
|
// VarParse alias of the ParseValue
|
|
func VarParse(val string) string {
|
|
return comfunc.ParseEnvVar(val, ValueGetter)
|
|
}
|
|
|
|
// ParseEnvValue alias of the ParseValue
|
|
func ParseEnvValue(val string) string {
|
|
return comfunc.ParseEnvVar(val, ValueGetter)
|
|
}
|
|
|
|
// ParseValue parse ENV var value from input string, support default value.
|
|
//
|
|
// Format:
|
|
//
|
|
// ${var_name} Only var name
|
|
// ${var_name | default} With default value
|
|
//
|
|
// Usage:
|
|
//
|
|
// envutil.ParseValue("${ APP_NAME }")
|
|
// envutil.ParseValue("${ APP_ENV | dev }")
|
|
func ParseValue(val string) (newVal string) {
|
|
return comfunc.ParseEnvVar(val, ValueGetter)
|
|
}
|
|
|
|
// SetEnvs to os
|
|
func SetEnvs(mp map[string]string) {
|
|
for key, value := range mp {
|
|
_ = os.Setenv(key, value)
|
|
}
|
|
}
|