关于Method Swizzling有很多非常好的文章,如雷纯锋的技术博客、NSHipster,国内外大大小小的博客也都有多多少少的介绍,我也看了不少,每次看都有一些疑惑,今天就记下来方便以后的查看。
BOOL success = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
if (success) {
class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
} else {
method_exchangeImplementations(originalMethod, swizzledMethod);
}
其实疑惑就是为什么要做这个判断?当然,在其他博客里作者也都提到一二。
第1种情况:主类本身没有实现需要替换的方法,而是继承了父类的实现,即 class_addMethod 方法返回 YES 。这时使用 class_getInstanceMethod 函数获取到的 originalSelector 指向的就是父类的方法,我们再通过执行 class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod)); 将父类的实现替换到我们自定义的 mrc_viewWillAppear 方法中。这样就达到了在 mrc_viewWillAppear 方法的实现中调用父类实现的目的。
第2种情况:主类本身有实现需要替换的方法,也就是 class_addMethod 方法返回 NO 。这种情况的处理比较简单,直接交换两个方法的实现就可以了.
看了之后还是似懂非懂,什么情况下会是第一种情况,什么情况下会是第二种情况呢?既然这样,自己动手做一下实验吧。根据上面的信息,如果父类没有需要的替换的方法时会是第一种情况,有的话是第二种。首先声明两个类,Person
和Student
,其中Student
继承于Person
。
Person类:
#import <Foundation/Foundation.h>
@interface Person : NSObject
- (void)doWork;
@end
#import "Person.h"
@implementation Person
- (void)doWork {
NSLog(@"Person is doing work!");
}
@end
Student类
#import "Person.h"
@interface Student : Person
@end
#import "Student.h"
@implementation Student
@end
Student类的Method Swizzling分类
+ (void)load {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
Class class = [self class];
SEL originalSelector = @selector(doWork);
SEL swizzledSelector = @selector(mrc_doWork);
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
BOOL success = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
if (success) {
class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
} else {
method_exchangeImplementations(originalMethod, swizzledMethod);
}
});
}
- (void)mrc_doWork {
[self mrc_doWork];
NSLog(@"MRCStudent is doing work!");
}
Person
类中有-(void)doWork
方法,Student
类继承于Person
类,我们想要将Student
类中的-(void)doWork
方法替换为-(void)mrc_doWork
,在第一种情况处加断点,程序运行起来。
程序果然跑进了第一种情况,同时分别调用
Person
和Student
的doWork
方法,
Person *p = [[Person alloc] init];
[p doWork];
NSLog(@"--------------------------");
Student *s = [[Student alloc] init];
[s doWork];
控制台输出如下:
如果注释条件选择语句,分类变成下面这样
+ (void)load {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
Class class = [self class];
SEL originalSelector = @selector(doWork);
SEL swizzledSelector = @selector(mrc_doWork);
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
// BOOL success = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
// if (success) {
// class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
// } else {
method_exchangeImplementations(originalMethod, swizzledMethod);
// }
});
}
- (void)mrc_doWork {
[self mrc_doWork];
NSLog(@"MRCStudent is doing work!");
}
再次运行程序,程序竟然崩溃了
崩溃信息显示Person
没有mrc_doWork
方法,什么情况,完全搞不懂啊!其实程序是崩溃在下面这一句,具体是Person
调用doWork
方法时崩溃在这一句。
我们想替换Student
的doWork
方法,Student
的doWork
方法继承自父类,如果按照上边注释条件选择语句的方式来写,实质上是这样的
Student
的分类替换的是父类的doWork
方法,我们的原意是仅仅替换Student
的某个方法,实质上却替换了父类的该方法。
程序运行崩溃的原因呼之欲出了,Person
调用doWork
方法,经过method swizzling,实质上调用的是mrc_doWork
,在mrc_doWork
方法内部执行到[self mrc_doWork]
时,因为Person
没有该方法,程序崩溃。
原来被注释的条件选择语句有防止崩溃的功能,既然这样,我们看一下它做了什么。
其实总结起来就是,当子类需要替换某个方法,而该方法继承自父类,防止替换父类方法的实现。