method swizzling你应该注意的点

原文地址

method swizzling

方法交换是OC运行时特征之一，经过方法交换能够hook方法，在实现一些需求时能够达到事半功倍的效果，但使用不慎一样可能致使不可思议的后果。在使用method swizzling前都应该理解如下注意点。

1.避免交换父类方法

若是当前类未实现被交换的方法而父类实现了的状况下，此时父类的实现会被交换，若此父类的多个继承者都在交换时会致使方法被交换屡次而混乱，同时当调用父类的方法时会由于找不到而发生崩溃。
因此在交换前都应该先尝试为当前类添加被交换的函数的新的实现IMP，若是添加成功则说明类没有实现被交换的方法，则只须要替代分类交换方法的实现为原方法的实现，若是添加失败，则原类中实现了被交换的方法，则能够直接进行交换。

这个过程主要涉及如下三个函数：

• class_addMethod

BOOL class_addMethod(Class cls, SEL name, IMP imp, const char *types)

给类cls的SEL添加一个实现IMP, 返回YES则代表类cls并未实现此方法，返回NO则代表类已实现了此方法。注意：添加成功与否，彻底由该类自己来决定，与父类有无该方法无关。

• class_replaceMethod

class_replaceMethod(Class _Nullable cls, SEL _Nonnull name, IMP _Nonnull imp, 
                    const char * _Nullable types)

替换类cls的SEL的函数实现为imp

• method_exchangeImplementations

method_exchangeImplementations(Method _Nonnull m1, Method _Nonnull m2)

交换两个方法的实现m1,m1

最终的过程大体以下代码：

BOOL didAddMethod = class_addMethod(class,originalSelector,
                                            method_getImplementation(swizzledMethod),
                                            method_getTypeEncoding(swizzledMethod));
if (didAddMethod) {//添加成功：则代表没有实现IMP，将源SEL的IMP替换到交换SEL的IMP
    class_replaceMethod(class,swizzledSelector,
                        method_getImplementation(originalMethod),
                        method_getTypeEncoding(originalMethod));
} else {//添加失败：代表已实现，则能够直接交换实现IMP
    method_exchangeImplementations(originalMethod, swizzledMethod);
}

2.交换方法应在+load方法

方法交换应当在调用前完成交换，+load方法发生在运行时初始化过程当中类被加载的时候调用，且每一个类被加载的时候只会调用一次load方法，调用的顺序是父类、类、分类，且他们之间是相互独立的，不会存在覆盖的关系，因此放在+load方法中能够确保在使用时已经完成交换。

3.交换方法应该放到dispatch_once中执行

在第2点已经写到，+load方法在类被加载的时候调用，且只会调用一次，那为何还须要dispatch_once呢？这是为了防止手动调用+load方法而致使反复的被交换，由于这是存在可能的。

4.交换的分类方法应该添加自定义前缀，避免冲突

这个毫无疑问，由于分类的方法会覆盖类中同名的方法，这样会致使没法预估的后果

5.交换的分类方法应调用原实现

不少状况咱们不清楚被交换的的方法具体作了什么内部逻辑，并且不少被交换的方法都是系统封装的方法，因此为了保证其逻辑性都应该在分类的交换方法中去调用原被交换方法，注意：调用时方法交换已经完成，在分类方法中应该调用分类方法自己才正确。

在每次的方法交换时都应该注意以上几点，最终咱们能够将其封装到NSObject的分类中，方便在调用：

#import <Foundation/Foundation.h>
#import <objc/runtime.h>
@interface NSObject (Swizzling) 

+ (void)methodSwizzlingWithOriginalSelector:(SEL)originalSelector
                         swizzledSelector:(SEL)swizzledSelector;
@end

#import "NSObject+Swizzling.h"
@implementation NSObject (Swizzling)

+ (void)methodSwizzlingWithOriginalSelector:(SEL)originalSelector swizzledSelector:(SEL)swizzledSelector{
    Class class = [self class];

    //原有被交换方法
    Method originalMethod = class_getInstanceMethod(class, originalSelector);

    //要交换的分类新方法
    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);

    //避免交换到父类的方法，先尝试添加被交换方法的新实现IMP
    BOOL didAddMethod = class_addMethod(class,originalSelector,
                                        method_getImplementation(swizzledMethod),
                                        method_getTypeEncoding(swizzledMethod));

    if (didAddMethod) {//添加成功：则代表没有实现IMP，将源SEL的IMP替换到交换SEL的IMP
        class_replaceMethod(class,swizzledSelector,
                            method_getImplementation(originalMethod),
                            method_getTypeEncoding(originalMethod));
    } else {//添加失败：代表已实现，则能够直接交换实现IMP
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}
@end