只有缓存的和准备使用类型一样才可以

Author: josercc

ZHTableViewGroupExample/ZHTableViewGroupObjc/ZHTableViewDataSource.m

@@ -76,15 +76,18 @@ + (UITableViewCell *)cellForRowAtWithDataSource:(ZHTableViewDataSource *)dataSou
                                          config:(BOOL)config
                                        useCache:(BOOL)useCache {
     ZHTableViewGroup *group = [self groupForSectionWithDataSource:dataSource section:indexPath.section];
+    ZHTableViewCell *tableViewCell = [group tableViewCellForIndexPath:indexPath];
     UITableViewCell *cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:nil];
     if (!group) {
         return cell;
     }
     UITableViewCell *resultCell;
     if (useCache) {
-        resultCell = [dataSource cacheCellWithIndexPath:indexPath];
-        ZHTableViewCell *tableViewCell = [group tableViewCellForIndexPath:indexPath];
-        [group tableViewCell:tableViewCell configCell:resultCell atIndexPath:indexPath];
+        UITableViewCell *cacheCell = [dataSource cacheCellWithIndexPath:indexPath];
+        if ([cacheCell isKindOfClass:tableViewCell.anyClass]) {
+            resultCell = [dataSource cacheCellWithIndexPath:indexPath];
+            [group tableViewCell:tableViewCell configCell:resultCell atIndexPath:indexPath];
+        }
     }
     if (!resultCell) {
         resultCell = [group cellForTableViewWithTableView:dataSource.tableView
@@ -94,7 +97,6 @@ + (UITableViewCell *)cellForRowAtWithDataSource:(ZHTableViewDataSource *)dataSou
     if (!resultCell) {
         return cell;
     }
-    ZHTableViewCell *tableViewCell = [group tableViewCellForIndexPath:indexPath];
     NSIndexPath *realIndexPath = [group indexPathWithCell:tableViewCell indexPath:indexPath];
     BOOL isHidden = [tableViewCell isHiddenWithIndexPath:realIndexPath];
     resultCell.hidden = isHidden;

ZHTableViewGroupObjc.podspec

@@ -8,7 +8,7 @@
 
 Pod::Spec.new do |s|
   s.name             = 'ZHTableViewGroupObjc'
-  s.version          = '3.5.3'
+  s.version          = '3.5.5'
   s.summary          = 'Manger UITableView DataSource More Cell Style'
 
   s.homepage         = 'https://github.com/josercc/ZHTableViewGroup'