Merge pull request #9 from h3110w0r1d-y/main

🐛 修复对interface{}类型的支持

Author: liu-cn

filter/example_test.go

@@ -70,3 +70,24 @@ func ExampleOmit() {
 	//{"hot":120,"tags":[{"name":"foo"},{"name":"bar"}]}
 	//{"hot":120,"tags":[{"icon":"icon"},{"icon":"icon"}]}
 }
+
+type InterfaceTest struct {
+	Data interface{} `json:"data"`
+}
+
+func ExampleFilter_Interface() {
+
+	fmt.Println(Omit("", InterfaceTest{Data: map[string]interface{}{
+		"1": 1,
+	}}))
+	fmt.Println(Omit("", InterfaceTest{Data: map[string]interface{}{
+		"1": 1,
+		"2": map[string]interface{}{
+			"3": 3,
+		},
+	}}))
+	//Output:
+	//{"data":{"1":1}}
+	//{"data":{"1":1,"2":{"3":3}}}
+
+}

filter/parser.go

@@ -18,6 +18,10 @@ TakePointerValue: //取指针的值
 		valueOf = valueOf.Elem()
 		typeOf = typeOf.Elem()
 		goto TakePointerValue
+	case reflect.Interface:
+		valueOf = reflect.ValueOf(valueOf.Interface())
+		typeOf = valueOf.Type()
+		goto TakePointerValue
 	case reflect.Struct:
 		parserStruct(typeOf, valueOf, t, scene, key, isSelect)
 	case reflect.Bool,

test/main.go

@@ -40,6 +40,10 @@ func (u *Us) GetAvatar2() string {
 }
 
 func main() {
+	fmt.Println(filter.Omit("1111", Us2{Data: map[string]interface{}{
+		"1": 1,
+	}}))
+
 	//var bb = []byte(`{"a":"1"}`)
 	u := Us{
 		BB:         [3]byte{1, 2, 4},