OC- Method Swizzling

1、Method Swizzling 原理

Objective-C 的每一个方法都有一个唯一对应的SEL，而方法是运行时绑定的，利用这些特性，我们可以在运行时调换方法的实现。前面我们说过，IMP 是一个函数指针，通过SEL获得指向方法的入口地址。下面是正常情况下SEL对应IMP函数指针的情况：

Method Swizzling 实现的原理就是调换IMP的指向：

2、Method Swizzling 应用

#import "UIViewController+MethodSwizzling.h"
#include <objc/runtime.h>

@implementation UIViewController (MethodSwizzling)

/*
 +load 方法会在类加入运行期调用，这是把方法调配放在+load方法里面的原因
 */
+ (void)load
{
    static dispatch_once_t onceToken;
    /*单例保证并行情况下不会有问题*/
    dispatch_once(&onceToken, ^{
        Class class = [self class];

        SEL originalSelector = @selector(viewWillAppear:);
        SEL swizzledSelector = @selector(my_viewWillAppear);

        /*获取原来的方法*/
        Method originalMethod = class_getInstanceMethod(class, originalSelector);
        /*获取要替换的方法*/
        Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);

        /*originalSelector是否有实现，如果存在返回YES，不存在返回NO*/
        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);
        }

    });
}

#pragma mark - Method Swizzling
/*
 看上去会递归调用，进入死循环。但是这个方法是准备和viewWillAppear方法交换的。在运行期，my_viewWillAppear 选择子实际上对应于原有的viewWillAppear方法实现。
 */
- (void)my_viewWillAppear
{
    [self my_viewWillAppear];
    NSLog(@"%s", __func__);
}

@end

3、注意事项

1、在+load方法中使用；

2、命名规范，带前缀；

3、在dispatch_once中执行，保证代码不管在多少个线程情况下，只被执行一次。