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];是干嘛用的，麻烦哪位好心人看懂了告诉我下....
走进aopUtils的injectTableView方法:

- (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的时候，会走到_IMYAOPTableView的aop_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];
复制代码

会走到_IMYAOPTableView的aop_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+UITableViewDataSource的numberOfRowsInSection方法，核心方法就在于

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 sectionsection = [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;}///如果该位置不是广告， 则转为逻辑indexsection = [self userSectionByFeeds:section];NSIndexPath *userIndexPath = [NSIndexPath indexPathForRow:row - cutCount inSection:section];return userIndexPath;
}
复制代码

END

还有很多地方没看明白，还需要多学习 :)