From 062309ab9e17b6f57c8a868bd9f01b4ce840171f Mon Sep 17 00:00:00 2001
From: Yin Heng <yinheng@kuaishou.com>
Date: Tue, 17 Jan 2023 16:39:19 +0800
Subject: [PATCH] add type check when it is class  foundation

---
 MJExtension/NSObject+MJKeyValue.m | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/MJExtension/NSObject+MJKeyValue.m b/MJExtension/NSObject+MJKeyValue.m
index c48e003..26a6d87 100755
--- a/MJExtension/NSObject+MJKeyValue.m
+++ b/MJExtension/NSObject+MJKeyValue.m
@@ -290,8 +290,16 @@ + (NSMutableArray *)mj_objectArrayWithKeyValuesArray:(id)keyValuesArray context:
     MJExtensionAssertError([keyValuesArray isKindOfClass:[NSArray class]], nil, [self class], @"keyValuesArray参数不是一个数组");
     
     // 如果数组里面放的是NSString、NSNumber等数据
-    if ([MJFoundation isClassFromFoundation:self]) return [NSMutableArray arrayWithArray:keyValuesArray];
-    
+    // 则直接判断类型是否一致
+    if ([MJFoundation isClassFromFoundation:self]) {
+        NSMutableArray *modelArray = [NSMutableArray array];
+        [keyValuesArray enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
+            if ([obj isKindOfClass:self]) {
+                [modelArray addObject:obj];
+            }
+        }];
+        return modelArray;
+    }
 
     // 2.创建数组
     NSMutableArray *modelArray = [NSMutableArray array];