Method Swizzling中的陷阱

作者：代培
地址：http://blog.csdn.net/dp948080952/article/details/54882811
转载请注明出处

这篇文章不是介绍什么是方法交换，这类文章很多，如果你不知道什么是方法交换可以看这篇文章：Method Swizzling

陷阱

方法交换是很危险的东西，会产生一些未知的错误，最近在使用方法交换时就遇到了这样的问题，在我交换了一个系统方法后，在调用这个方法时会在我交换的方法中循环无法跳出。

最终我找到了问题的关键，那就是这个方法可能不存在！

每次iOS版本的更新都会出现一些新的API，这些API在之前的版本中就不存在，而你在交换一个方法时如果不考虑它是否存在，那就会导致严重的错误。

比如我曾经交换过一个方法：

- (void)openURL:(NSURL*)url options:(NSDictionary<NSString *, id> *)options completionHandler:(void (^ __nullable)(BOOL success))completion

这个方法只在iOS10之后有，别人在使用的时候也会先判断这个方法是否会被响应，但是我们看下面的交换代码会先添加这个方法，如果这个方法不存在，那么本来不会响应就变成了会响应，那么在iOS10之前的系统就会进入这个方法，导致死循环。

    Class class = [self class];    Method method1 = class_getInstanceMethod(class, sel1);    Method method2 = class_getInstanceMethod(class, sel2);    BOOL didAddMethod =    class_addMethod(class,                    sel1,                    method_getImplementation(method2),                    method_getTypeEncoding(method2));    if (didAddMethod) {        class_replaceMethod(class,                            sel2,                            method_getImplementation(method1),                            method_getTypeEncoding(method1));    } else {        method_exchangeImplementations(method1, method2);    }

解决方案

解决这个问题也很简单就是交换前做一个判断：

    if (![class instancesRespondToSelector:sel1] || ![class instancesRespondToSelector:sel2]) {        return ;    }

如果不响应这个方法，直接返回。

封装

为了不重复写方法交换的代码，也能减少错误，我们可以将其封装，将其放在NSObject的Category中再合适不过了。

NSObject+DPExtension.h

#import <Foundation/Foundation.h>@interface NSObject (DPExtension)+ (void)intanceMethodExchangeWithOriginSelector:(SEL)sel1 swizzledSelector:(SEL)sel2;@end

NSObject+DPExtension.m

#import "NSObject+DPExtension.h"#import <objc/runtime.h>@implementation NSObject (DPExtension)+ (void)intanceMethodExchangeWithOriginSelector:(SEL)sel1 swizzledSelector:(SEL)sel2 {    Class class = [self class];    if (![class instancesRespondToSelector:sel1] || ![class instancesRespondToSelector:sel2]) {        return ;    }    Method method1 = class_getInstanceMethod(class, sel1);    Method method2 = class_getInstanceMethod(class, sel2);    BOOL didAddMethod =    class_addMethod(class,                    sel1,                    method_getImplementation(method2),                    method_getTypeEncoding(method2));    if (didAddMethod) {        class_replaceMethod(class,                            sel2,                            method_getImplementation(method1),                            method_getTypeEncoding(method1));    } else {        method_exchangeImplementations(method1, method2);    }}@end

就是这些^_^