Method Swizzling解惑

关于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，在第一种情况处加断点，程序运行起来。

lALOUbu_R80Bbs0DVA_852_366.png

Person *p = [[Person alloc] init];
[p doWork];

NSLog(@"--------------------------");

Student *s = [[Student alloc] init];
[s doWork];

控制台输出如下：

注释前.png

如果注释条件选择语句，分类变成下面这样

+ (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!");
}

再次运行程序，程序竟然崩溃了

崩溃信息.png

崩溃信息显示Person没有mrc_doWork方法，什么情况，完全搞不懂啊！其实程序是崩溃在下面这一句，具体是Person调用doWork方法时崩溃在这一句。

崩溃原因.png

我们想替换Student的doWork方法，Student的doWork方法继承自父类，如果按照上边注释条件选择语句的方式来写，实质上是这样的

Student的分类替换的是父类的doWork方法，我们的原意是仅仅替换Student的某个方法，实质上却替换了父类的该方法。

程序运行崩溃的原因呼之欲出了，Person调用doWork方法，经过method swizzling，实质上调用的是mrc_doWork，在mrc_doWork方法内部执行到[self mrc_doWork]时，因为Person没有该方法，程序崩溃。
原来被注释的条件选择语句有防止崩溃的功能，既然这样，我们看一下它做了什么。

注释干的事儿.png

其实总结起来就是，当子类需要替换某个方法，而该方法继承自父类，防止替换父类方法的实现。