100 lines
2.0 KiB
Go
100 lines
2.0 KiB
Go
package support
|
||
|
||
import (
|
||
"fmt"
|
||
"reflect"
|
||
"strings"
|
||
"unsafe"
|
||
)
|
||
|
||
func GetFieldValue(i any, k string) any {
|
||
v := reflect.ValueOf(i)
|
||
|
||
var kv reflect.Value
|
||
|
||
if v.Kind() == reflect.Ptr {
|
||
v = reflect.Indirect(v)
|
||
}
|
||
|
||
if v.Kind() == reflect.Struct {
|
||
kv = v.FieldByName(k)
|
||
} else if v.Kind() == reflect.Map {
|
||
for _, key := range v.MapKeys() {
|
||
if key.String() == k {
|
||
kv = v.MapIndex(key)
|
||
}
|
||
}
|
||
}
|
||
|
||
switch kv.Kind() {
|
||
case reflect.String:
|
||
return kv.String()
|
||
case reflect.Bool:
|
||
return kv.Bool()
|
||
case reflect.Float32, reflect.Float64:
|
||
return kv.Float()
|
||
case reflect.Int, reflect.Int64:
|
||
return int(kv.Int())
|
||
case reflect.Invalid:
|
||
return nil
|
||
}
|
||
|
||
return kv.Interface()
|
||
}
|
||
|
||
func InterfaceToSlice(i any) (ret []any) {
|
||
rv := reflect.ValueOf(i)
|
||
|
||
if rv.Kind() == reflect.Ptr {
|
||
rv = reflect.Indirect(rv)
|
||
}
|
||
|
||
if rv.Kind() != reflect.Slice {
|
||
panic("请传入切片类型数据")
|
||
}
|
||
|
||
for i := 0; i < rv.Len(); i++ {
|
||
ret = append(ret, rv.Index(i).Interface())
|
||
}
|
||
|
||
return ret
|
||
}
|
||
|
||
func SliceConvert(origSlice any, newSliceType reflect.Type) any {
|
||
sv := reflect.ValueOf(origSlice)
|
||
|
||
if sv.Kind() != reflect.Slice {
|
||
panic(fmt.Sprintf("无效切片类型 %T (来源)", origSlice))
|
||
}
|
||
|
||
if newSliceType.Kind() != reflect.Slice {
|
||
panic(fmt.Sprintf("无效切片类型 %T (目标)", newSliceType))
|
||
}
|
||
|
||
// 生成新类型的切片
|
||
newSlice := reflect.New(newSliceType)
|
||
|
||
// hdr指向到新生成切片的SliceHeader
|
||
hdr := (*reflect.SliceHeader)(unsafe.Pointer(newSlice.Pointer()))
|
||
|
||
var newElemSize = int(sv.Type().Elem().Size()) / int(newSliceType.Elem().Size())
|
||
|
||
// 设置SliceHeader的Cap，Len，以及数组的ptr
|
||
hdr.Cap = sv.Cap() * newElemSize
|
||
hdr.Len = sv.Len() * newElemSize
|
||
hdr.Data = uintptr(sv.Pointer())
|
||
|
||
return newSlice.Elem().Interface()
|
||
}
|
||
|
||
func UcFirst(s string) string {
|
||
if s == "" {
|
||
return ""
|
||
}
|
||
return strings.ToUpper(s[:1]) + s[1:]
|
||
}
|
||
|
||
func StudlyCase(s string) string {
|
||
return strings.Replace(strings.Title(strings.Replace(s, "-", " ", -1)), " ", "", -1)
|
||
}
|