77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
|
package flags
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"flag"
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
"reflect"
|
|||
|
|
"strconv"
|
|||
|
|
"unsafe"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ParseWithStructPointers 启动参数解析，如果启动参数没有指定，会去env里查找同名参数
|
|||
|
|
// flagStructPointers为结构体指针数组
|
|||
|
|
func ParseWithStructPointers(flagStructPointers ...interface{}) {
|
|||
|
|
|
|||
|
|
for _, st := range flagStructPointers {
|
|||
|
|
flagParseStruct2Flags(st)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
flag.Parse()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func flagParseStruct2Flags(st interface{}) {
|
|||
|
|
if st == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var stTo = reflect.TypeOf(st)
|
|||
|
|
var stVo = reflect.ValueOf(st)
|
|||
|
|
switch stTo.Kind() {
|
|||
|
|
case reflect.Ptr:
|
|||
|
|
stTo = stTo.Elem()
|
|||
|
|
stVo = stVo.Elem()
|
|||
|
|
// case reflect.Struct:
|
|||
|
|
// break
|
|||
|
|
default:
|
|||
|
|
panic(fmt.Errorf("invalid flags parse struct(%+v), must be pointer or struct", st))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for i := 0; i < stTo.NumField(); i++ {
|
|||
|
|
field := stTo.Field(i)
|
|||
|
|
|
|||
|
|
key, find := field.Tag.Lookup("env")
|
|||
|
|
if !find {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
desc := field.Tag.Get("desc")
|
|||
|
|
|
|||
|
|
defaultValue, find := os.LookupEnv(key)
|
|||
|
|
if !find {
|
|||
|
|
defaultValue, find = field.Tag.Lookup("default")
|
|||
|
|
if !find {
|
|||
|
|
defaultValue = ""
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var fieldValuePointer = unsafe.Pointer(stVo.Field(i).Addr().Pointer())
|
|||
|
|
switch field.Type.Kind() {
|
|||
|
|
case reflect.String:
|
|||
|
|
flag.StringVar((*string)(fieldValuePointer), key, defaultValue, desc)
|
|||
|
|
case reflect.Int:
|
|||
|
|
defaultValue1, _ := strconv.Atoi(defaultValue)
|
|||
|
|
flag.IntVar((*int)(fieldValuePointer), key, defaultValue1, desc)
|
|||
|
|
case reflect.Int64:
|
|||
|
|
defaultValue1, _ := strconv.ParseInt(defaultValue, 10, 64)
|
|||
|
|
flag.Int64Var((*int64)(fieldValuePointer), key, defaultValue1, desc)
|
|||
|
|
case reflect.Bool:
|
|||
|
|
flag.BoolVar((*bool)(fieldValuePointer), key, defaultValue == "true", desc)
|
|||
|
|
default:
|
|||
|
|
panic(fmt.Errorf("parse flag kind invalid,must be string/int/int64/bool, not %+v", field.Type.Kind()))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|