iOS--Runtime--Method Swizzling
2017-05-23 13:56 doudo 阅读(410) 评论(0) 编辑 收藏 举报
这篇文章的大部分内容来自互联网，我只是从中摘取我认为解释的比较合理的易懂的，然后根据我当时学习时容易出现理解误差的地方加以说明，希望读者在看的过程中同时参考最新的资料，毕竟知识也是在随时更新变化的。
一、Method Swizzling简介
叫法各有不同，有叫方法混淆，也有叫方法重组的，意思都差不多。Method Swizzling是改变一个selector的实际实现的技术。通过这一技术，我们可以在运行时通过修改类的分发表中selector对应的函数，来修改方法的实现。简单来说我们主要是使用Method Swizzling来把系统的方法交换为我们自己的方法，从而给系统方法添加一些我们想要的功能。
二、实例讲解
单纯的讲，不如在实际应用中来的更直观，也更容易理解，所以先上一个例子来讲解他的用法：
例如，我们想跟踪在程序中每一个view controller展示给用户的次数：当然，我们可以在每个view controller的viewDidAppear中添加跟踪代码；但是这太过麻烦，需要在每个view controller中写重复的代码。创建一个子类可能是一种实现方式，但需要同时创建UIViewController, UITableViewController, UINavigationController及其它UIKit中view controller的子类，这同样会产生许多重复的代码。
这种情况下，我们就可以使用Method Swizzling，代码如下：
#import <objc/runtime.h> @implementation UIViewController (Tracking) + (void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ Class class = [self class]; // When swizzling a class method, use the following: // Class class = object_getClass((id)self); SEL originalSelector = @selector(viewWillAppear:); SEL swizzledSelector = @selector(xxx_viewWillAppear:); //原有方法 Method originalMethod = class_getInstanceMethod(class, originalSelector); //替换原有方法的新方法 Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector); //先尝试給源SEL添加IMP，这里是为了避免源SEL没有实现IMP的情况（这种情况是原方法继承自父类，而本身并没有实现） BOOL didAddMethod = class_addMethod(class,originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod)); if (didAddMethod) {//添加成功：说明源SEL没有实现IMP，将源SEL的IMP替换到交换SEL的IMP class_replaceMethod(class,swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod)); } else {//添加失败：说明源SEL已经有IMP，直接将两个SEL的IMP交换即可 method_exchangeImplementations(originalMethod, swizzledMethod); } }); } #pragma mark - Method Swizzling - (void)xxx_viewWillAppear:(BOOL)animated { [self xxx_viewWillAppear:animated]; NSLog(@"viewWillAppear: %@", self); } @end
说明：
1.为什么要增加didAddMethod 判断？
先尝试添加原SEL其实是为了做一层保护，因为如果这个类没有实现originalSelector，但其父类实现了，那class_getInstanceMethod会返回父类的方法。这样method_exchangeImplementations替换的是父类的那个方法，这时候如果父类调用自己的那个方法是会crash的，这当然不是我们想要的。所以我们先尝试添加 orginalSelector，如果已经存在，再用 method_exchangeImplementations 把原方法的实现跟新的方法实现给交换掉。
2.方法说明：
①.
class_addMethod(class,originalSelector,
method_getImplementation(swizzledMethod),
method_getTypeEncoding(swizzledMethod));
该方法会为一个类添加方法（包括方法名称（SEL）和方法的实现（IMP）），返回值为BOOL类型，表示方法是否成功添加。添加成功与否，完全由该类本身来决定，与父类有该方法无关：该类本身已经实现该方法，则添加失败；该类本身没有实现，则添加成功。添加的方法，也不会对父类的方法造成影响。
②.
class_replaceMethod(class,swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
该方法，内部首先会判断selector（方法）是否存在，如果不存在，会执行class_addMethod；如果方法存在，则会执行method_setImplementation。
③.
method_exchangeImplementations(originalMethod, swizzledMethod);
该方法，单纯交换两个方法的实现，无需过多说明。
三、注意事项：
1.为什么方法交换调用在+load方法中？
在Objective-C runtime会自动调用两个类方法，分别为+load与+ initialize。+load 方法是在类被加载的时候调用的，也就是一定会被调用。而+initialize方法是在类或它的子类收到第一条消息之前被调用的，这里所指的消息包括实例方法和类方法的调用。也就是说+initialize方法是以懒加载的方式被调用的，如果程序一直没有给某个类或它的子类发送消息，那么这个类的+initialize方法是永远不会被调用的。此外+load方法还有一个非常重要的特性，那就是子类、父类和分类中的+load方法的实现是被区别对待的。换句话说在 Objective-C runtime自动调用+load方法时，分类中的+load方法并不会对主类中的+load方法造成覆盖。综上所述，+load 方法是实现 Method Swizzling 逻辑的最佳“场所”。如需更深入理解，可参考Objective-C 深入理解 +load 和 +initialize。
2.为什么方法交换要在dispatch_once中执行？
方法交换应该要线程安全，而且保证在任何情况下（多线程环境，或者被其他人手动再次调用+load方法）只交换一次，防止再次调用又将方法交换回来。除非只是临时交换使用，在使用完成后又交换回来。 最常用的解决方案是在+load方法中使用dispatch_once来保证交换是安全的。之前有读者反馈+load方法本身即为线程安全，为什么仍需添加dispatch_once，其原因就在于+load方法本身无法保证其中代码只被执行一次。
3.避免冲突：给自定义的分类方法加前缀，从而使其与所依赖的代码库不会存在命名冲突。
如：- (void)xxx_viewWillAppear:(BOOL)animated。
4.总是调用方法的原始实现(除非有更好的理由不这么做)。
如： [self xxx_viewWillAppear:animated]。
API提供了一个输入与输出约定，但其内部实现是一个黑盒。Swizzle一个方法而不调用原始实现可能会打破私有状态底层操作，从而影响到程序的其它部分。
五、项目实例
在列举之前，我们可以将Method Swizzling功能封装为类方法，作为NSObject的类别，这样我们后续调用也会方便些。
#import <Foundation/Foundation.h> #import <objc/runtime.h> @interface NSObject (Swizzling) + (void)methodSwizzlingWithOriginalSelector:(SEL)originalSelector bySwizzledSelector:(SEL)swizzledSelector; @end
#import "NSObject+Swizzling.h" @implementation NSObject (Swizzling) + (void)methodSwizzlingWithOriginalSelector:(SEL)originalSelector bySwizzledSelector:(SEL)swizzledSelector{ Class class = [self class]; //原有方法 Method originalMethod = class_getInstanceMethod(class, originalSelector); //替换原有方法的新方法 Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector); //先尝试給源SEL添加IMP，这里是为了避免源SEL没有实现IMP的情况 BOOL didAddMethod = class_addMethod(class,originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod)); if (didAddMethod) {//添加成功：说明源SEL没有实现IMP，将源SEL的IMP替换到交换SEL的IMP class_replaceMethod(class,swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod)); } else {//添加失败：说明源SEL已经有IMP，直接将两个SEL的IMP交换即可 method_exchangeImplementations(originalMethod, swizzledMethod); } } @end
实例一：替换ViewController生命周期方法
App跳转到某具有网络请求的界面时，为了用户体验效果常会添加加载栏或进度条来显示当前请求情况或进度。这种界面都会存在这样一个问题，在请求较慢时，用户手动退出界面，这时候需要去除加载栏。
当然可以依次在每个界面的viewWillDisappear方法中添加去除方法，但如果类似的界面过多，一味的复制粘贴也不是方法。这时候就能体现Method Swizzling的作用了，我们可以替换系统的viewWillDisappear方法，使得每当执行该方法时即自动去除加载栏。
#import "UIViewController+Swizzling.h" #import "NSObject+Swizzling.h" @implementation UIViewController (Swizzling) + (void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ [self methodSwizzlingWithOriginalSelector:@selector(viewWillDisappear:) bySwizzledSelector:@selector(sure_viewWillDisappear:)]; }); } - (void)sure_viewWillDisappear:(BOOL)animated { [self sure_viewWillDisappear:animated]; [SVProgressHUD dismiss]; }
代码如上，这样就不用考虑界面是否移除加载栏的问题了。补充一点，通常我们也会在生命周期方法中设置默认界面背景颜色，因若背景颜色默认为透明对App的性能也有一定影响，这大家可以在UIKit性能优化那篇文章中查阅。但类似该类操作也可以书写在通用类中，所以具体使用还要靠自己定夺。
实例二：解决获取索引、添加、删除元素越界崩溃问题
对于NSArray、NSDictionary、NSMutableArray、NSMutableDictionary不免会进行索引访问、添加、删除元素的操作，越界问题也是很常见，这时我们可以通过Method Swizzling解决这些问题，越界给予提示防止崩溃。
这里以NSMutableArray为例说明:
#import "NSMutableArray+Swizzling.h" #import "NSObject+Swizzling.h" @implementation NSMutableArray (Swizzling) + (void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ [objc_getClass("__NSArrayM") methodSwizzlingWithOriginalSelector:@selector(removeObject:) bySwizzledSelector:@selector(safeRemoveObject:) ]; [objc_getClass("__NSArrayM") methodSwizzlingWithOriginalSelector:@selector(addObject:) bySwizzledSelector:@selector(safeAddObject:)]; [objc_getClass("__NSArrayM") methodSwizzlingWithOriginalSelector:@selector(removeObjectAtIndex:) bySwizzledSelector:@selector(safeRemoveObjectAtIndex:)]; [objc_getClass("__NSArrayM") methodSwizzlingWithOriginalSelector:@selector(insertObject:atIndex:) bySwizzledSelector:@selector(safeInsertObject:atIndex:)]; [objc_getClass("__NSArrayM") methodSwizzlingWithOriginalSelector:@selector(objectAtIndex:) bySwizzledSelector:@selector(safeObjectAtIndex:)]; }); } - (void)safeAddObject:(id)obj { if (obj == nil) { NSLog(@"%s can add nil object into NSMutableArray", __FUNCTION__); } else { [self safeAddObject:obj]; } } - (void)safeRemoveObject:(id)obj { if (obj == nil) { NSLog(@"%s call -removeObject:, but argument obj is nil", __FUNCTION__); return; } [self safeRemoveObject:obj]; } - (void)safeInsertObject:(id)anObject atIndex:(NSUInteger)index { if (anObject == nil) { NSLog(@"%s can't insert nil into NSMutableArray", __FUNCTION__); } else if (index > self.count) { NSLog(@"%s index is invalid", __FUNCTION__); } else { [self safeInsertObject:anObject atIndex:index]; } } - (id)safeObjectAtIndex:(NSUInteger)index { if (self.count == 0) { NSLog(@"%s can't get any object from an empty array", __FUNCTION__); return nil; } if (index > self.count) { NSLog(@"%s index out of bounds in array", __FUNCTION__); return nil; } return [self safeObjectAtIndex:index]; } - (void)safeRemoveObjectAtIndex:(NSUInteger)index { if (self.count <= 0) { NSLog(@"%s can't get any object from an empty array", __FUNCTION__); return; } if (index >= self.count) { NSLog(@"%s index out of bound", __FUNCTION__); return; } [self safeRemoveObjectAtIndex:index]; } @end
对应大家可以举一反三，相应的实现添加、删除等，以及NSArray、NSDictionary等操作，因代码篇幅较大，这里就不一一书写了。
这里没有使用self来调用，而是使用objc_getClass("__NSArrayM")来调用的。因为NSMutableArray的真实类只能通过后者来获取，而不能通过[self class]来获取，而method swizzling只对真实的类起作用。这里就涉及到一个小知识点：类簇。补充以上对象对应类簇表。
实例三：防止按钮重复暴力点击
程序中大量按钮没有做连续响应的校验，连续点击出现了很多不必要的问题，例如发表帖子操作，用户手快点击多次，就会导致同一帖子发布多次。
#import <UIKit/UIKit.h> //默认时间间隔 #define defaultInterval 1 @interface UIButton (Swizzling) //点击间隔 @property (nonatomic, assign) NSTimeInterval timeInterval; //用于设置单个按钮不需要被hook @property (nonatomic, assign) BOOL isIgnore; @end
#import "UIButton+Swizzling.h" #import "NSObject+Swizzling.h" @implementation UIButton (Swizzling) + (void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ [self methodSwizzlingWithOriginalSelector:@selector(sendAction:to:forEvent:) bySwizzledSelector:@selector(sure_SendAction:to:forEvent:)]; }); } - (NSTimeInterval)timeInterval{ return [objc_getAssociatedObject(self, _cmd) doubleValue]; } - (void)setTimeInterval:(NSTimeInterval)timeInterval{ objc_setAssociatedObject(self, @selector(timeInterval), @(timeInterval), OBJC_ASSOCIATION_RETAIN_NONATOMIC); } //当按钮点击事件sendAction 时将会执行sure_SendAction - (void)sure_SendAction:(SEL)action to:(id)target forEvent:(UIEvent *)event{ if (self.isIgnore) { //不需要被hook [self sure_SendAction:action to:target forEvent:event]; return; } if ([NSStringFromClass(self.class) isEqualToString:@"UIButton"]) { self.timeInterval =self.timeInterval == 0 ?defaultInterval:self.timeInterval; if (self.isIgnoreEvent){ return; }else if (self.timeInterval > 0){ [self performSelector:@selector(resetState) withObject:nil afterDelay:self.timeInterval]; } } //此处 methodA和methodB方法IMP互换了，实际上执行 sendAction；所以不会死循环 self.isIgnoreEvent = YES; [self sure_SendAction:action to:target forEvent:event]; } //runtime 动态绑定 属性 - (void)setIsIgnoreEvent:(BOOL)isIgnoreEvent{ // 注意BOOL类型 需要用OBJC_ASSOCIATION_RETAIN_NONATOMIC 不要用错，否则set方法会赋值出错 objc_setAssociatedObject(self, @selector(isIgnoreEvent), @(isIgnoreEvent), OBJC_ASSOCIATION_RETAIN_NONATOMIC); } - (BOOL)isIgnoreEvent{ //_cmd == @select(isIgnore); 和set方法里一致 return [objc_getAssociatedObject(self, _cmd) boolValue]; } - (void)setIsIgnore:(BOOL)isIgnore{ // 注意BOOL类型 需要用OBJC_ASSOCIATION_RETAIN_NONATOMIC 不要用错，否则set方法会赋值出错 objc_setAssociatedObject(self, @selector(isIgnore), @(isIgnore), OBJC_ASSOCIATION_RETAIN_NONATOMIC); } - (BOOL)isIgnore{ //_cmd == @select(isIgnore); 和set方法里一致 return [objc_getAssociatedObject(self, _cmd) boolValue]; } - (void)resetState{ [self setIsIgnoreEvent:NO]; } @end
实例四：全局更换控件初始效果
以UILabel为例，在项目比较成熟的基础上，应用中需要引入新的字体，需要更换所有Label的默认字体，但是同时，对于一些特殊设置了字体的label又不需要更换。乍看起来，这个问题确实十分棘手，首先项目比较大，一个一个设置所有使用到的label的font工作量是巨大的，并且在许多动态展示的界面中，可能会漏掉一些label，产生bug。其次，项目中的label来源并不唯一，有用代码创建的，有xib和storyBoard中的，这也将浪费很大的精力。这时Method Swizzling可以解决此问题，避免繁琐的操作。
#import "UILabel+Swizzling.h" #import "NSObject+Swizzling.h" @implementation UILabel (Swizzling) + (void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ [self methodSwizzlingWithOriginalSelector:@selector(init) bySwizzledSelector:@selector(sure_Init)]; [self methodSwizzlingWithOriginalSelector:@selector(initWithFrame:) bySwizzledSelector:@selector(sure_InitWithFrame:)]; [self methodSwizzlingWithOriginalSelector:@selector(awakeFromNib) bySwizzledSelector:@selector(sure_AwakeFromNib)]; }); } - (instancetype)sure_Init{ id __self = [self sure_Init]; UIFont * font = [UIFont fontWithName:@"Zapfino" size:self.font.pointSize]; if (font) { self.font=font; } return __self; } - (instancetype)sure_InitWithFrame:(CGRect)rect{ id __self = [self sure_InitWithFrame:rect]; UIFont * font = [UIFont fontWithName:@"Zapfino" size:self.font.pointSize]; if (font) { self.font=font; } return __self; } - (void)sure_AwakeFromNib{ [self sure_AwakeFromNib]; UIFont * font = [UIFont fontWithName:@"Zapfino" size:self.font.pointSize]; if (font) { self.font=font; } } @end
这一实例个人认为使用率可能不高，对于产品的设计这些点都是已经确定好的，更改的几率很低。况且我们也可以使用appearance来进行统一设置。
实例五：App热修复
因为AppStore上线审核时间较长，且如果在线上版本出现bug修复起来也是很困难，这时App热修复就可以解决此问题。热修复即在不更改线上版本的前提下，对线上版本进行更新甚至添加模块。国内比较好的热修复技术：JSPatch。JSPatch能做到通过JS调用和改写OC方法最根本的原因是Objective-C是动态语言，OC上所有方法的调用/类的生成都通过Objective-C Runtime在运行时进行，我们可以通过类名/方法名反射得到相应的类和方法，进而替换出现bug的方法或者添加方法等。bang的博客上有详细的描述有兴趣可以参考，这里就不赘述了。
实例六：App异常加载占位图通用类封装（更新于：2016/12/01）
详情可见文章:《零行代码为App添加异常加载占位图》
在该功能模块中，使用Runtime Method Swizzling进行替换tableView、collectionView的reloadData方法，使得每当执行刷新操作时，自动检测当前组数与行数，从而实现零代码判断占位图是否显示的功能，同样也适用于网络异常等情况，详细设置可前往阅读。
实例七：全局修改导航栏后退（返回）按钮（更新于：2016/12/05）
在真实项目开发中，会全局统一某控件样式，以导航栏后退（返回）按钮为例，通常项目中会固定为返回字样，或者以图片进行显示等。
iOS默认的返回按钮样式如下，默认为蓝色左箭头，文字为上一界面标题文字。
这里我们仍可以通过Runtime Method Swizzling来实现该需求，在使用Method Swizzling进行更改之前，必须考虑注意事项，即尽可能的不影响原有操作，比如对于系统默认的返回按钮，与其对应的是有界面边缘右滑返回功能的，因此我们进行统一更改后不可使其功能废弃。
闲话少说，我们创建基于UINavigationItem的类别，在其load方法中替换方法backBarButtonItem
代码如下
#import "UINavigationItem+Swizzling.h" #import "NSObject+Swizzling.h" static char *kCustomBackButtonKey; @implementation UINavigationItem (Swizzling) + (void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ [self methodSwizzlingWithOriginalSelector:@selector(backBarButtonItem) bySwizzledSelector:@selector(sure_backBarButtonItem)]; }); } - (UIBarButtonItem*)sure_backBarButtonItem { UIBarButtonItem *backItem = [self sure_backBarButtonItem]; if (backItem) { return backItem; } backItem = objc_getAssociatedObject(self, &kCustomBackButtonKey); if (!backItem) { backItem = [[UIBarButtonItem alloc] initWithTitle:@"" style:UIBarButtonItemStylePlain target:nil action:NULL]; objc_setAssociatedObject(self, &kCustomBackButtonKey, backItem, OBJC_ASSOCIATION_RETAIN_NONATOMIC); } return backItem; } @end
这里进行将返回按钮的文字清空操作，其他需求样式大家也可随意替换，现在再次运行程序，就会发现所有的返回按钮均只剩左箭头，并右滑手势依然有效。如图所示:
以上内容，都是结合前人的果实，加上自己的理解，并根据我当时学的时候可能出的问题加以针对说明，应该会对后人有用吧。
参考：Runtime Method Swizzling开发实例汇总（持续更新中）
Objective-C Runtime 运行时之四：Method Swizzling
