34 lines
887 B
Go
34 lines
887 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"reflect"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
func LoadOptionsFromEnv(options interface{}, cfg map[string]interface{}) {
|
||
|
|
val := reflect.ValueOf(options).Elem()
|
||
|
|
typ := val.Type()
|
||
|
|
for i := 0; i < typ.NumField(); i++ {
|
||
|
|
// pull out the struct tags:
|
||
|
|
// flag - the name of the command line flag
|
||
|
|
// deprecated - (optional) the name of the deprecated command line flag
|
||
|
|
// cfg - (optional, defaults to underscored flag) the name of the config file option
|
||
|
|
field := typ.Field(i)
|
||
|
|
flagName := field.Tag.Get("flag")
|
||
|
|
envName := field.Tag.Get("env")
|
||
|
|
cfgName := field.Tag.Get("cfg")
|
||
|
|
if cfgName == "" && flagName != "" {
|
||
|
|
cfgName = strings.Replace(flagName, "-", "_", -1)
|
||
|
|
}
|
||
|
|
if envName == "" || cfgName == "" {
|
||
|
|
// resolvable fields must have the `env` and `cfg` struct tag
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
v := os.Getenv(envName)
|
||
|
|
if v != "" {
|
||
|
|
cfg[cfgName] = v
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|