DZNEmptyDataSet源码解析

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

最近看了实现原理,整理如下:
DZNEmptyDataSet 用类似 MethodSwizzle 的方法,重新实现了UITableView、UICollectionViewreloadData 方法。在新的方法实现中通过UITableView、UICollectionViewDataSource代理方法获取要展示数据的条数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];
}
  1. 这之后,调用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);
    }
}
  1. 其中[self dzn_reloadEmptyDataSet];通过UITableView、UICollectionViewDataSource代理方法获取要展示数据的条数item,如果item = 0, 则展示空页面,如果item > 0将移除空页面。
    展示和移除空页面的UI操作不做详细分析。

推荐阅读更多精彩内容

  • 发现 关注 消息 iOS 第三方库、插件、知名博客总结 作者大灰狼的小绵羊哥哥关注 2017.06.26 09:4...
    肇东周阅读 7,106评论 4 39
  • 概述在iOS开发中UITableView可以说是使用最广泛的控件,我们平时使用的软件中到处都可以看到它的影子,类似...
    liudhkk阅读 7,911评论 3 38
  • 老虎伤人事件接二连三,有人说野生动物园管理不到位,没有及时提醒与规劝;也有说因为动物园门票太贵,农民工舍不得...
    微风与暖阳阅读 120评论 0 1
  • 握住我的手 我给予你力量 握住我的手 我给予你信念 握住我的手 我给予你信心 握住我的手 我给予你激情 握住我的手...
    如镜若水阅读 64评论 0 0
  • 已经完结一段时间的日剧《贤者之爱》简直是颠覆了我的三观,看完以后整个人抖三抖啊,太短了,没看够啊,来继续刷新我的三...
    今日排行榜阅读 981评论 0 9