IMYAOPTableView源碼學習筆記

NO IMAGE

Head

近期由於入職了新公司,接觸到了資訊業務的模塊,看了看代碼發現資訊業務的廣告植入是由IMYAOPTableView 實現的,出於好奇,探索了下源碼,走了一邊流程,雖然框架中還有挺多東西沒看懂[ 🙁 ],這邊先將流程記錄下來

Content

IMYAOPTableView 總體是將業務流與廣告流分開,再記錄原數據源以及代理、新數據源以及新代理,最後再分發給對應的流
框架中的三行代碼,對應的位置就是YYFeedListExample 中的 tableView:didSelectRowAtIndexPath:

UITableView *feedsTableView = [ctrl valueForKey:@"feedsView"];
self.aopDemo = [IMYAOPTableDemo new];
self.aopDemo.aopUtils = feedsTableView.aop_utils;

這裡使用kvc取出業務流,精髓就在於設置aop_utils這個屬性上,我們點擊右邊的aop_utils進入:

- (IMYAOPTableViewUtils *)aop_utils {
IMYAOPTableViewUtils *aopUtils = objc_getAssociatedObject(self, kIMYAOPTableUtilsKey);
if (!aopUtils) {
@synchronized(self) {
aopUtils = objc_getAssociatedObject(self, kIMYAOPTableUtilsKey);
if (!aopUtils) {
///初始化部分配置
[_IMYAOPTableView aop_setupConfigs];
// 獲取aop utils,設置aopUtils的tableView對象
aopUtils = [IMYAOPTableViewUtils aopUtilsWithTableView:self];
// 設置關聯
objc_setAssociatedObject(self, kIMYAOPTableUtilsKey, aopUtils, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
// 注入tableView
[aopUtils injectTableView];
}
}
}
return aopUtils;
}

這裡創建單例對象,使用runtime關聯,沒看懂[_IMYAOPTableView aop_setupConfigs];是幹嘛用的,麻煩哪位好心人看懂了告訴我下….
走進aopUtilsinjectTableView方法:

- (void)injectTableView {
UITableView *tableView = self.tableView;
// 記錄原始的數據源以及代理對象
// 這裡如果你點Twitter 就是 :T1HomeTimelineItemsViewController
_origDataSource = tableView.dataSource;
_origDelegate = tableView.delegate;
[self injectFeedsView:tableView];
}

這邊把原數據源、原代理存儲在aopUtils_origDataSource以及_origDelegate,這邊也就是T1HomeTimelineItemsViewController 對象,再走進injectFeedsView方法:

- (void)injectFeedsView:(UIView *)feedsView {
struct objc_super objcSuper = {.super_class = [self msgSendSuperClass], .receiver = feedsView};
// 設置新的數據源以及代理對象: objcSuper結構體的地址即為第一個成員(receiver)的地址
// objc_msgSendSuper 消息接收者還是 feedsView 只是查詢方法是去父類查找
// feedsView.delegate = self;
// feedsView.dataSource = self;
((void (*)(void *, SEL, id))(void *)objc_msgSendSuper)(&objcSuper, @selector(setDelegate:), self);
((void (*)(void *, SEL, id))(void *)objc_msgSendSuper)(&objcSuper, @selector(setDataSource:), self);
self.origViewClass = [feedsView class];
Class aopClass = [self makeSubclassWithClass:self.origViewClass];
if (![self.origViewClass isSubclassOfClass:aopClass]) {
[self bindingFeedsView:feedsView aopClass:aopClass];
}
}

這裡構造了一個結構體objcSuper,使用objc_msgSendSuper發送消息,個人感覺

((void (*)(void *, SEL, id))(void *)objc_msgSendSuper)(&objcSuper, @selector(setDelegate:), self);
((void (*)(void *, SEL, id))(void *)objc_msgSendSuper)(&objcSuper, @selector(setDataSource:), self);

等價於:

feedsView.delegate = self;
feedsView.dataSource = self;

接下來,走進makeSubclassWithClass

- (Class)makeSubclassWithClass:(Class)origClass {
NSString *className = NSStringFromClass(origClass);
NSString *aopClassName = [kA`setupAopClass`PFeedsViewPrefix, stringByAppendingString:className];
Class aopClass = NSClassFromString(aopClassName);
if (aopClass) {
return aopClass;
}
aopClass = objc_allocateClassPair(origClass, aopClassName.UTF8String, 0);
[self setupAopClass:aopClass];
objc_registerClassPair(aopClass);
return aopClass;
}

這裡動態創建子類kIMYAOP_ClassName,並注入實現該子類方法的類為_IMYAOPTableView,覆蓋父類的實現,比如進入到setupAopClass,查看

[self addOverriteMethod:@selector(reloadData) aopClass:aopClass];
- (void)addOverriteMethod:(SEL)seletor aopClass:(Class)aopClass {
NSString *seletorString = NSStringFromSelector(seletor);
NSString *aopSeletorString = [NSString stringWithFormat:@"aop_%@", seletorString];
SEL aopMethod = NSSelectorFromString(aopSeletorString);
[self addOverriteMethod:seletor toMethod:aopMethod aopClass:aopClass];
}
- (void)addOverriteMethod:(SEL)seletor toMethod:(SEL)toSeletor aopClass:(Class)aopClass {
Class implClass = [self implAopViewClass];
Method method = class_getInstanceMethod(implClass, toSeletor);
if (method == NULL) {
method = class_getInstanceMethod(implClass, seletor);
}
const char *types = method_getTypeEncoding(method);
IMP imp = method_getImplementation(method);
class_addMethod(aopClass, seletor, imp, types);
}

這裡動態生成aop_seletor,並添加到子類kIMYAOP_ClassName的方法列表中:

class_addMethod(aopClass, seletor, imp, types);

所以當你再調用aopUtils.tableView.reloadData的時候,會走到_IMYAOPTableViewaop_reloadData方法實現,再往下看bindingFeedsView:aopClass:
啊….這些是什麼,不懂不懂,看懂的快告訴我….

到這裡,就配置好了原始的數據源、代理、動態創建子類、子類覆蓋方法等,接下來就看廣告類的設置

點擊左邊的aopUtils

 self.aopDemo.aopUtils = feedsTableView.aop_utils;

進入injectTableView

- (void)injectTableView {
[self.aopUtils.tableView registerClass:[UITableViewCell class] forCellReuseIdentifier:@"AD"];
///廣告回調,跟TableView的Delegate,DataSource 一樣。
self.aopUtils.delegate = self;
self.aopUtils.dataSource = self;
dispatch_async(dispatch_get_main_queue(), ^{
[self insertRows];
});
}

這裡,就將aopUtils的代理設置成了廣告類,用於最後的分發,往下看insertRows:

- (void)insertRows {
NSMutableArray<IMYAOPTableViewInsertBody *> *insertBodys = [NSMutableArray array];
///隨機生成了5個要插入的位置
for (int i = 0; i < 5; i++) {
NSIndexPath *indexPath = [NSIndexPath indexPathForRow:arc4random() % 10 inSection:0];
[insertBodys addObject:[IMYAOPTableViewInsertBody insertBodyWithIndexPath:indexPath]];
}
///清空 舊數據
[self.aopUtils insertWithSections:nil];
[self.aopUtils insertWithIndexPaths:nil];
///插入 新數據, 同一個 row 會按數組的順序 row 進行 遞增
[self.aopUtils insertWithIndexPaths:insertBodys];
///調用tableView的reloadData,進行頁面刷新
[self.aopUtils.tableView reloadData];
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
NSLog(@"%@", self.aopUtils.allModels);
});
}

進入insertWithIndexPaths方法:

- (void)insertWithIndexPaths:(NSArray<IMYAOPBaseInsertBody *> *)indexPaths {
NSArray<IMYAOPBaseInsertBody *> *array = [indexPaths sortedArrayUsingComparator:^NSComparisonResult(IMYAOPBaseInsertBody *_Nonnull obj1, IMYAOPBaseInsertBody *_Nonnull obj2) {
return [obj1.indexPath compare:obj2.indexPath];
}];
NSMutableDictionary *insertMap = [NSMutableDictionary dictionary];
[array enumerateObjectsUsingBlock:^(IMYAOPBaseInsertBody *_Nonnull obj, NSUInteger idx, BOOL *_Nonnull stop) {
NSInteger section = obj.indexPath.section;
NSInteger row = obj.indexPath.row;
NSMutableArray *rowArray = insertMap[@(section)];
if (!rowArray) {
rowArray = [NSMutableArray array];
[insertMap setObject:rowArray forKey:@(section)];
}
while (YES) {
BOOL hasEqual = NO;
for (NSIndexPath *inserted in rowArray) {
if (inserted.row == row) {
row++;
hasEqual = YES;
break;
}
}
if (hasEqual == NO) {
break;
}
}
NSIndexPath *insertPath = [NSIndexPath indexPathForRow:row inSection:section];
[rowArray addObject:insertPath];
obj.resultIndexPath = insertPath;
}];
self.sectionMap = insertMap;
}

原諒我比較懶,我直接看了結果,就是把廣告的indexPath記錄到sectionMap裡把,嗯沒錯,應該是。。
最後就是調用過程了

[self.aopUtils.tableView reloadData];

會走到_IMYAOPTableViewaop_reloadData方法實現

- (void)aop_reloadData {
AopDefineVars;
aop_utils.isUICalling += 1;
AopCallSuper(@selector(reloadData));
aop_utils.isUICalling -= 1;
}

這裡會調用父類(YYTableView)的reloadData方法,YYTableView又調用了[super reloadData],所以最終是也是調用[UITableView]reloadData,即走到aop_utils的數據源方法上,查看IMYAOPTableViewUtils+UITableViewDataSourcenumberOfRowsInSection方法,核心方法就在於

NSIndexPath *feedsIndexPath = [self feedsIndexPathByUser:[NSIndexPath indexPathForRow:rowCount inSection:section]];
- (NSIndexPath *)feedsIndexPathByUser:(NSIndexPath *)userIndexPath {
if (userIndexPath == nil) {
return nil;
}
NSInteger section = userIndexPath.section;
NSInteger row = userIndexPath.row;
///轉為table section
section = [self feedsSectionByUser:section];
NSMutableArray<NSIndexPath *> *array = self.sectionMap[@(section)];
for (NSIndexPath *obj in array) {
if (obj.row <= row) {
row += 1;
} else {
break;
}
}
NSIndexPath *feedsIndexPath = [NSIndexPath indexPathForRow:row inSection:section];
return feedsIndexPath;
}

這裡計算出了最終業務流+廣告流的cell數量
再往下看tableView:cellForRowAtIndexPath:方法:

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
kAOPUICallingSaved;
kAOPUserIndexPathCode;
UITableViewCell *cell = nil;
if ([dataSource respondsToSelector:@selector(tableView:cellForRowAtIndexPath:)]) {
cell = [dataSource tableView:tableView cellForRowAtIndexPath:indexPath];
}
if (![cell isKindOfClass:[UITableViewCell class]]) {
cell = [UITableViewCell new];
if (dataSource) {
NSAssert(NO, @"Cell is Nil");
}
}
kAOPUICallingResotre;
return cell;
}

核心在於kAOPUserIndexPathCode:
這裡區分該indexPath是廣告流還是業務流,查看userIndexPathByFeeds,最終取出dataSource,然後分發

#define kAOPUserIndexPathCode                                           \
NSIndexPath *userIndexPath = [self userIndexPathByFeeds:indexPath]; \
id<IMYAOPTableViewDataSource> dataSource = nil;                     \
if (userIndexPath) {                                                \
dataSource = (id)self.origDataSource;                           \
indexPath = userIndexPath;                                      \
} else {                                                            \
dataSource = self.dataSource;                                   \
isInjectAction = YES;                                           \
}                                                                   \
if (isInjectAction) {                                               \
self.isUICalling += 1;                                          \
}
- (NSIndexPath *)userIndexPathByFeeds:(NSIndexPath *)feedsIndexPath {
if (!feedsIndexPath) {
return nil;
}
NSInteger section = feedsIndexPath.section;
NSInteger row = feedsIndexPath.row;
NSMutableArray<NSIndexPath *> *array = self.sectionMap[@(section)];
NSInteger cutCount = 0;
for (NSIndexPath *obj in array) {
if (obj.row == row) {
cutCount = -1;
break;
}
if (obj.row < row) {
cutCount++;
} else {
break;
}
}
if (cutCount < 0) {
return nil;
}
///如果該位置不是廣告, 則轉為邏輯index
section = [self userSectionByFeeds:section];
NSIndexPath *userIndexPath = [NSIndexPath indexPathForRow:row - cutCount inSection:section];
return userIndexPath;
}

END

還有很多地方沒看明白,還需要多學習 🙂

相關文章

Android重修課Handler機制

iOSRunLoop的使用及底層原理

iOSload和initialize的異同

iOS性能優化(初級)