添加数据类型判断

Author: deatil

array/array.go

@@ -722,6 +722,40 @@ func (this *Array) flatten(includeEmpty bool) (map[string]any, error) {
 	return flattened, nil
 }
 
+// 获取 array, slice, map or string 的长度
+// get array, slice, map or string len
+func (this *Array) Len() (num int) {
+	defer func() {
+		if err := recover(); err != nil {
+			num = 0
+		}
+	}()
+
+	return reflect.ValueOf(this.source).Len()
+}
+
+// 判断是否是 slice 类型
+// if the source is slice and return true
+func (this *Array) IsSlice() bool {
+	kind := reflect.TypeOf(this.source).Kind()
+
+	return kind == reflect.Slice || kind == reflect.Array
+}
+
+// 判断是否是 map 类型
+// if the source is map and return true
+func (this *Array) IsMap() bool {
+	kind := reflect.TypeOf(this.source).Kind()
+
+	return kind == reflect.Map
+}
+
+// 判断是否是 slice or map 类型
+// if the source is slice or map and return true
+func (this *Array) IsArray() bool {
+	return this.IsSlice() || this.IsMap()
+}
+
 // 搜索
 // Search data with key from source
 func (this *Array) search(source any, path ...string) any {
@@ -980,7 +1014,7 @@ func (this *Array) anySliceFormat(data any) ([]any, bool) {
 	newData := dataValue.Interface()
 
 	newDataKind := reflect.TypeOf(newData).Kind()
-	if newDataKind == reflect.Slice {
+	if newDataKind == reflect.Slice || newDataKind == reflect.Array {
 		newDataValue := reflect.ValueOf(newData)
 		newDataLen := newDataValue.Len()
 

array/array_test.go

@@ -25,11 +25,23 @@ var (
 				"ddddd",
 				"fffff",
 			},
+			"ddeef": []any{
+				"ccc中文",
+				"ddddd",
+				"fffff",
+			},
 			"ddd": []int64{
 				22,
 				333,
 				555,
 			},
+			"qqq": [10]int64{
+				22,
+				333,
+				555,
+				5557,
+				5558,
+			},
 			"ff": map[any]any{
 				111: "fccccc",
 				222: "fddddd",
@@ -347,6 +359,11 @@ func Test_Find(t *testing.T) {
 			"ddddd",
 			"[]any",
 		},
+		{
+			"b.qqq.2",
+			"555",
+			"[num]any",
+		},
 		{
 			"b.ff.222",
 			"fddddd",
@@ -425,6 +442,223 @@ func Test_Find_func(t *testing.T) {
 
 }
 
+func Test_Len(t *testing.T) {
+	assert := assertDeepEqualT(t)
+
+	testData := []struct {
+		key      string
+		expected int
+		msg      string
+	}{
+		{
+			"a",
+			0,
+			"int",
+		},
+		{
+			"b.ddeef.1",
+			5,
+			"string",
+		},
+		{
+			"b.dd",
+			3,
+			"[]any",
+		},
+		{
+			"b.qqq",
+			10,
+			"[num]any",
+		},
+		{
+			"b.ff",
+			3,
+			"map[any]any",
+		},
+		{
+			"b.hhTy3",
+			0,
+			"&map[int]any",
+		},
+		{
+			"b.hhTy3.333",
+			3,
+			"map[any]string",
+		},
+		{
+			"b.hhTy3.666",
+			4,
+			"Slice",
+		},
+	}
+
+	for _, v := range testData {
+		check := New(arrData).Sub(v.key).Len()
+
+		assert(check, v.expected, v.msg)
+	}
+
+}
+
+func Test_IsArray(t *testing.T) {
+	assert := assertDeepEqualT(t)
+
+	testData := []struct {
+		key      string
+		expected bool
+		msg      string
+	}{
+		{
+			"a",
+			false,
+			"int",
+		},
+		{
+			"b.dd",
+			true,
+			"[]any",
+		},
+		{
+			"b.qqq",
+			true,
+			"[num]any",
+		},
+		{
+			"b.ff",
+			true,
+			"map[any]any",
+		},
+		{
+			"b.hhTy3",
+			false,
+			"&map[int]any",
+		},
+		{
+			"b.hhTy3.333",
+			true,
+			"map[any]string",
+		},
+		{
+			"b.hhTy3.666",
+			true,
+			"Slice",
+		},
+	}
+
+	for _, v := range testData {
+		check := New(arrData).Sub(v.key).IsArray()
+
+		assert(check, v.expected, v.msg)
+	}
+
+}
+
+func Test_IsMap(t *testing.T) {
+	assert := assertDeepEqualT(t)
+
+	testData := []struct {
+		key      string
+		expected bool
+		msg      string
+	}{
+		{
+			"a",
+			false,
+			"int",
+		},
+		{
+			"b.dd",
+			false,
+			"[]any",
+		},
+		{
+			"b.qqq",
+			false,
+			"[num]any",
+		},
+		{
+			"b.ff",
+			true,
+			"map[any]any",
+		},
+		{
+			"b.hhTy3",
+			false,
+			"&map[int]any",
+		},
+		{
+			"b.hhTy3.333",
+			true,
+			"map[any]string",
+		},
+		{
+			"b.hhTy3.666",
+			false,
+			"Slice",
+		},
+	}
+
+	for _, v := range testData {
+		check := New(arrData).Sub(v.key).IsMap()
+
+		assert(check, v.expected, v.msg)
+	}
+
+}
+
+func Test_IsSlice(t *testing.T) {
+	assert := assertDeepEqualT(t)
+
+	testData := []struct {
+		key      string
+		expected bool
+		msg      string
+	}{
+		{
+			"a",
+			false,
+			"int",
+		},
+		{
+			"b.dd",
+			true,
+			"[]any",
+		},
+		{
+			"b.qqq",
+			true,
+			"[num]any",
+		},
+		{
+			"b.ff",
+			false,
+			"map[any]any",
+		},
+		{
+			"b.hhTy3",
+			false,
+			"&map[int]any",
+		},
+		{
+			"b.hhTy3.333",
+			false,
+			"map[any]string",
+		},
+		{
+			"b.hhTy3.666",
+			true,
+			"Slice",
+		},
+	}
+
+	for _, v := range testData {
+		check := New(arrData).Sub(v.key).IsSlice()
+
+		assert(check, v.expected, v.msg)
+	}
+
+}
+
 func Test_Search(t *testing.T) {
 	assert := assertT(t)