[feat] 新增 辅助函数
This commit is contained in:
parent
9b71c96e61
commit
9149dc140f
@ -1 +1,99 @@
|
|||||||
package support
|
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)
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user