DZNEmptyDataSet源码解析

DZNEmptyDataSet (github接近1万星)是一个能够为UITableView、UICollectionView自动添加空页面提示的第三方库，使用起来方便快捷。

最近看了实现原理，整理如下：
DZNEmptyDataSet 用类似 MethodSwizzle 的方法，重新实现了UITableView、UICollectionView 的 reloadData 方法。在新的方法实现中通过UITableView、UICollectionView 的DataSource代理方法获取要展示数据的条数item，如果item = 0, 则展示空页面。

1. 如何改写reloadData 方法的实现？

 - (void)swizzleIfPossible:(SEL)selector {
    
    if (![self respondsToSelector:selector]) {
        return;
    }
    
    // 新建检索字典
    if (!_impLookupTable) {
        _impLookupTable = [[NSMutableDictionary alloc] initWithCapacity:3];
        // 3 represent the supported base classes 
        // 字典key: className + selectorName
        // 字典value: dict { DZNSwizzleInfoOwnerKey:     class
                             // DZNSwizzleInfoPointerKey:   impValue
                             // DZNSwizzleInfoSelectorKey:  selector
    }
    
    // 查找是否已经改写过selector
    for (NSDictionary *info in [_impLookupTable allValues]) {
        Class class = [info objectForKey:DZNSwizzleInfoOwnerKey];
        NSString *selectorName = [info objectForKey:DZNSwizzleInfoSelectorKey];
        
        if ([selectorName isEqualToString:NSStringFromSelector(selector)]) {
            if ([self isKindOfClass:class]) {
                return;
            }
        }
    }
    
    // 获取self的类
    Class baseClass = dzn_baseClassToSwizzleForTarget(self);
    // 字符串拼接 className + selectorName
    NSString *key = dzn_implementationKey(baseClass, selector); 
    
    NSValue *impValue = [[_impLookupTable objectForKey:key] valueForKey:DZNSwizzleInfoPointerKey];
    
    
    // 如果实现已存在，跳过
    if (impValue || !key || !baseClass) {
        return;
    }
    
    // Swizzle
    Method method = class_getInstanceMethod(baseClass, selector);
    
    // 将dzn_original_implementation新的实现赋值给method, 其操作返回值为method的原有实现，将存储到检索字典中
    IMP dzn_newImplementation = method_setImplementation(method, (IMP)dzn_original_implementation);
    
    // 存储到检索字典中
    NSDictionary *swizzledInfo = @{DZNSwizzleInfoOwnerKey: baseClass,
                                   DZNSwizzleInfoSelectorKey: NSStringFromSelector(selector),
                                   DZNSwizzleInfoPointerKey: [NSValue valueWithPointer:dzn_newImplementation]};
    
    [_impLookupTable setObject:swizzledInfo forKey:key];
}

2. 这之后，调用reloadData方法将会调用dzn_original_implementation的实现

void dzn_original_implementation(id self, SEL _cmd) {
    // 在检索表中查找原有方法实现
    Class baseClass = dzn_baseClassToSwizzleForTarget(self);
    NSString *key = dzn_implementationKey(baseClass, _cmd);
    
    NSDictionary *swizzleInfo = [_impLookupTable objectForKey:key];
    NSValue *impValue = [swizzleInfo valueForKey:DZNSwizzleInfoPointerKey];
    
// 原有方法实现
    IMP impPointer = [impValue pointerValue];
    
    // 执行是否显示空页面的操作
    [self dzn_reloadEmptyDataSet];
    
    // 执行原有方法实现
    if (impPointer) {
        ((void(*)(id,SEL))impPointer)(self,_cmd);
    }
}

3. 其中[self dzn_reloadEmptyDataSet];通过UITableView、UICollectionView 的DataSource代理方法获取要展示数据的条数item，如果item = 0, 则展示空页面，如果item > 0将移除空页面。
   展示和移除空页面的UI操作不做详细分析。