first commit

go.mod

@@ -0,0 +1,5 @@
+module git.fsdpf.net/go/utils
+
+go 1.21
+
+require github.com/spf13/cast v1.6.0

go.sum

@@ -0,0 +1,12 @@
+github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
+github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
+github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
+github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
+github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
+github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
+github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0=
+github.com/spf13/cast v1.6.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=

utils.go

@@ -0,0 +1,175 @@
+package utils
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"reflect"
+	"strings"
+	"unsafe"
+
+	"github.com/spf13/cast"
+)
+
+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 GetStructField(item any, key string) (field reflect.StructField, ok bool) {
+	vItem := reflect.TypeOf(item)
+
+	if vItem.Kind() == reflect.Ptr {
+		vItem = vItem.Elem()
+	}
+
+	if vItem.Kind() != reflect.Struct {
+		return field, false
+	}
+
+	return vItem.FieldByName(UcFirst(key))
+}
+
+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 DeepSearch(m map[string]any, path []string) map[string]any {
+	for _, k := range path {
+		m2, ok := m[k]
+		if !ok {
+			// intermediate key does not exist
+			// => create it and continue from there
+			m3 := make(map[string]interface{})
+			m[k] = m3
+			m = m3
+			continue
+		}
+		m3, ok := m2.(map[string]interface{})
+		if !ok {
+			// intermediate key is a value
+			// => replace with a new map
+			m3 = make(map[string]interface{})
+			m[k] = m3
+		}
+		// continue search from here
+		m = m3
+	}
+	return m
+}
+
+func MergeMap(shadow map[string]any, m map[string]any) map[string]any {
+	if shadow == nil {
+		shadow = make(map[string]any)
+	}
+
+	var m2 map[string]any
+
+	for k, val := range m {
+		switch val.(type) {
+		case map[string]any:
+			m2 = val.(map[string]any)
+		case map[any]any:
+			m2 = cast.ToStringMap(val)
+		default:
+			// immediate value
+			shadow[k] = val
+			continue
+		}
+		// recursively merge to shadow map
+		shadow[k] = MergeMap(cast.ToStringMap(shadow[k]), m2)
+	}
+	return shadow
+}
+
+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)
+}
+
+func GetAbsPath(s string) string {
+	if strings.HasPrefix(s, "/") {
+		return s
+	} else if dir, err := os.Getwd(); err == nil {
+		return filepath.Join(dir, s)
+	}
+
+	return s
+}

utils_test.go

@@ -0,0 +1,46 @@
+package utils
+
+import (
+	"testing"
+)
+
+func TestDeepSearch(t *testing.T) {
+	m := map[string]any{
+		"a": 1,
+		"b": map[string]any{
+			"c": 2,
+		},
+	}
+
+	mb := DeepSearch(m, []string{"b", "a"})
+
+	t.Log(m)
+
+	mb["cc"] = "3"
+
+	t.Log(m)
+}
+
+func TestMergeMap(t *testing.T) {
+	m1 := map[string]any{
+		"a": 1,
+		"b": map[string]any{
+			"c": 2,
+			"a": 1,
+			"b": map[string]any{
+				"c": 2,
+				"a": 1,
+			},
+		},
+	}
+
+	m2 := map[string]any{
+		"a": 11,
+		"b": map[string]any{
+			"c":  22,
+			"cc": 33,
+		},
+	}
+
+	t.Log(MergeMap(m1, m2))
+}