当我们需要更改某个类的方法,但却无法拿到到该类的的源码时,我们通常是通过继承并重写对应方法来完成目的的。当然还有另外一条路:Method Swizzling ,它允许我们动态地替换方法的实现偷天换日的目的。
比如有个需求要为每一个ViewController增加一个监控。将ViewController子类化是一个方案,但要为UITableViewController, UINavigationController等等一个个加入会造成大量的无聊且重复的代码。这就到Method Swizzling出场的时候了。
再次请出Mattt Thompson大神写的这篇文章
#import "UIViewController+Tracking.h"
#import <objc/runtime.h>
@implementation UIViewController (Tracking)
+(void)load {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
Class class = [self class];
SEL originalSelector = @selector(viewWillAppear:);
SEL swizzledSelector = @selector(xxx_viewWillAppear:);
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
// When swizzling a class method, use the following:
// Class class = object_getClass((id)self);
// ...
// Method originalMethod = class_getClassMethod(class, originalSelector);
// Method swizzledMethod = class_getClassMethod(class, swizzledSelector);
BOOL didAddMethos = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
if(didAddMethos){
class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
}else{
method_exchangeImplementations(originalMethod, swizzledMethod);
}
});
}
#pragma mark - Method Swizzling
-(void)xxx_viewWillAppear:(BOOL)animated {
[self xxx_viewWillAppear:animated];
NSLog(@"viewWillAppear: %@", self);
}
@end
这样每一个UIViewController或者其子类调用viewWillAppear:时候就会打印出日志。
上面的代码通过给UIViewController添加一个名为Tracking的Category,将UIViewController类的viewWillAppear:方法和Category中xxx_viewWillAppear:方法实现了互换。
值得注意的几件事:
1.Swizzling应该写在+load方法中,因为+load是在类被初始化时候就被调用的。+initialize是在收到消息之后才调用,如果应用不发送消息给它,它就永远不可能执行。
2.Swizzling应该被写在dispatch_once中,保证只被执行一次和线程安全。
3.如果类中已经有了可替换的方法,那么就调用method_exchangeImplementations交换,否则调用class_addMethod和class_replaceMethod来完成替换。
4.xxx_viewWillAppear:方法的看似会引发死循环,但其实不会。在Swizzling的过程中xxx_viewWillAppear:已经被重新指定到UIViewController类的-viewWillAppear:中。不过如果我们调用的是viewWillAppear:反而会产生无限循环,因为这个方法的实现在运行时已经被重新指定为xxx_viewWillAppear:了。
- (void)xxx_viewWillAppear:(BOOL)animated { [self xxx_viewWillAppear:animated]; NSLog(@"viewWillAppear: %@", NSStringFromClass([self class]));}