Method Swizzling中的陷阱

2019-11-09 18:15:10

字体：大中小

来源：转载

供稿：网友

作者：代培地址：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

就是这些^_^

上一篇：第十一章粒子系统

下一篇：APP数据备份