引
小玩具的Git传送门.这是一个缓存cell高度的小玩具.目前在自己项目中使用.
关于在tableview的使用中,对于变高的cell,我们一般是会做缓存以避免重复计算.
最初是将高度保存在model中,不过稍显麻烦.
然后使用sunny大的UITableView+FDTemplateLayoutCell.自动计算height蛮爽,不过如果是使用al的话,对al有一定的要求.并且对于历史项目,则需要改动历史项目的heightForRowAtIndexPath代理方法,改动文件较多.
所以设想,如果能给tableview增加一个开关(cacheEnable属性),则tableview可以自行cahce,是否可行呢?
开始
可以确定,是一个UITableview的分类,这样不会造成侵入性.然后给分类添加一个属性作为开关,那么自然是通过associate的方式关联.开关负责功能的开启和关闭,那么具体缓存在哪里呢?所以还需要增加一个属性:cache,用来存储height数据:
- (void)setZc_enableCache:(BOOL)zc_enableCache {
if (zc_enableCache) {
[self startCache];
}
objc_setAssociatedObject(self, @selector(zc_enableCache), @(zc_enableCache), OBJC_ASSOCIATION_ASSIGN);
}
- (BOOL)zc_enableCache {
return [objc_getAssociatedObject(self, _cmd) boolValue];
}
- (void)setRowHeightCache:(NSMutableDictionary *)cache {
objc_setAssociatedObject(self, @selector(rowHeightCache), cache, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (NSMutableDictionary *)rowHeightCache {
return objc_getAssociatedObject(self, _cmd);
}
只有当设置cache enable为YES以后,才会开启cache功能.因为属性是可以反复设置的,但是功能开启只允许一次(通过swizzling),所以通过dispatch once来保证.
- (void)startCache {
static dispatch_once_t zcCacheToken;
dispatch_once(&zcCacheToken, ^{
[self initRowHeightCache];
[self setHookToDelegate];
});
}
对于存储height的cache,就是一个NSDictionary.字典的key是NSNumber,代表tableview的section;字典的value是NSArray属性,代表tableview下的所有row.initRowHeightCache方法就是根据section和row来初始化字典.
对于setHookToDelegate就稍显关键.缓存的关键在于替换掉tableview的代理方法:tableView:heightForRowAtIndexPath:,而代理方法所存在的类是不确定的.为了避免上篇博文中关于self的问题,我们通过以下方式处理.
- (void)setHookToDelegate {
Method currentRowHeightMethod = class_getInstanceMethod([self class], @selector(zc_tableView:heightForRowAtIndexPath:));
IMP currentRowHeightImp = method_getImplementation(currentRowHeightMethod);
const char *type = method_getTypeEncoding(currentRowHeightMethod);
BOOL isSuccess = class_addMethod([self.delegate class], @selector(zc_tableView:heightForRowAtIndexPath:), currentRowHeightImp, type);
if (isSuccess) {
Method oriRowHeightMethod = class_getInstanceMethod([self.delegate class], @selector(tableView:heightForRowAtIndexPath:));
Method latestRowHeightMethod = class_getInstanceMethod([self.delegate class], @selector(zc_tableView:heightForRowAtIndexPath:));
method_exchangeImplementations(oriRowHeightMethod, latestRowHeightMethod);
}
}
将新方法(zc_tableView:heightForRowAtIndexPath:)加入到代理类中,再交换代理类中的新方法与旧方法(tableView:heightForRowAtIndexPath:),这样就避免了self的相关问题.
新方法则是解决缓存的核心了.挺简单的一个原理.根据indexPath中的secton和row,来查找当前cache字典中的值,有则返回,无则调用原方法(tableView:heightForRowAtIndexPath:),并且保存,然后返回.
- (CGFloat)zc_tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {
BOOL flag = [objc_getAssociatedObject(tableView, @selector(zc_enableCache)) boolValue];
if (!flag) {
return [self zc_tableView:tableView heightForRowAtIndexPath:indexPath];
}
NSMutableDictionary *cache = objc_getAssociatedObject(tableView, @selector(rowHeightCache));
if (!cache) {
SEL selector = @selector(initRowHeightCache);
void (*methodPointer) (id, SEL) = (void (*) (id, SEL))[tableView methodForSelector:selector];
methodPointer(tableView,selector);
}
NSMutableArray *cacheArrayInSection = cache[@(indexPath.section)];
CGFloat cachedHeight = [cacheArrayInSection[indexPath.row] floatValue];
if (cachedHeight == -1) {
CGFloat height = [self zc_tableView:tableView heightForRowAtIndexPath:indexPath];
cacheArrayInSection[indexPath.row] = @(height);
return height;
}
else {
return cachedHeight;
}
}
这其中仍然涉及到了一个self问题:这里的self并不指该tableview的分类,而是指的tableview的delegate类.所以调用[self xxxx]时,会出现crash的情况.所以能通过参数:tableview来代替self的地方,则用tableview.绕不过去的地方,则直接使用imp函数指针来进行invoke.
剩下的工作就没太多了,一共有9个影响数据源的方法.包括一个总体的reload,4个对section的reload,delete,insert,move,4个对row的reload,delete,inert,move.每当数据源变动,则缓存也需要跟着变动.
于是我们在+load方法里面交换一下相关方法,然后做一些简单的数据结构:
reload整体/section/row,则清零整体/section/row的缓存.
move则是交换相应缓存.
delete则是删除相应缓存,并且后面的缓存都往前移动.
insert则是添加一个新的空缓存,后面的缓存都往后移动.
只是一个玩具
因为自己的特殊需求和代码偏好(不喜欢侵入),有了这么一个小工具.所以目前的情况是:
- 功能不够强大(没有自动计算高度)
- 基于runloop的预缓存暂未完成.(git上的代码有一个preCache的方法.不过仅添加了runloop的observer,handler尚未实现)
- 仍然强烈推荐sunny大的fd:git地址
- 有bug/建议请告诉我~~
