DZNEmptyDataSet (github接近1万星)是一个能够为UITableView、UICollectionView自动添加空页面提示的第三方库，使用起来方便快捷。
最近看了实现原理，整理如下：
DZNEmptyDataSet 用类似 MethodSwizzle 的方法，重新实现了UITableView、UICollectionView 的 reloadData 方法。在新的方法实现中通过UITableView、UICollectionView 的DataSource代理方法获取要展示数据的条数item，如果item = 0, 则展示空页面。
- 如何改写
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];
}
- 这之后，调用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);
}
}
- 其中
[self dzn_reloadEmptyDataSet];通过UITableView、UICollectionView的DataSource代理方法获取要展示数据的条数item，如果item = 0, 则展示空页面，如果item > 0将移除空页面。
展示和移除空页面的UI操作不做详细分析。
