补充字段类型处理

infraboard · Jun 10, 2024 · a2453b5 · a2453b5
1 parent 7b22985
commit a2453b5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/desense/default_test.go b/desense/default_test.go
@@ -33,6 +33,7 @@ var set = &TestMaskStuctSet{
 	Items: []*TestMaskStuct{
 		{
 			PhoneNumber: "1234567890123",
+			Name:        "test",
 		},
 	},
 }
@@ -48,5 +49,6 @@ func TestMuskSet(t *testing.T) {
 
 	for _, v := range set.Items {
 		t.Log(v.PhoneNumber)
+		t.Log(v.Name)
 	}
 }
diff --git a/desense/desense.go b/desense/desense.go
@@ -20,11 +20,17 @@ func MaskStruct(s any) error {
 		switch fieldValue.Kind() {
 		case reflect.Slice:
 			for i := 0; i < fieldValue.Len(); i++ {
-				MaskStruct(fieldValue.Index(i).Interface())
+				err := MaskStruct(fieldValue.Index(i).Interface())
+				if err != nil {
+					return err
+				}
 			}
-		case reflect.Struct:
-			MaskStruct(fieldValue.Addr().Interface())
-		default:
+		case reflect.Ptr:
+			err := MaskStruct(fieldValue.Interface())
+			if err != nil {
+				return err
+			}
+		case reflect.String:
 			tag := t.Field(i).Tag.Get("mask")
 			if tag == "" {
 				continue