Method Swizzling

当我们需要更改某个类的方法,但却无法拿到到该类的的源码时,我们通常是通过继承并重写对应方法来完成目的的。当然还有另外一条路:Method Swizzling ,它允许我们动态地替换方法的实现偷天换日的目的。

比如有个需求要为每一个ViewController增加一个监控。将ViewController子类化是一个方案,但要为UITableViewController, UINavigationController等等一个个加入会造成大量的无聊且重复的代码。这就到Method Swizzling出场的时候了。

再次请出Mattt Thompson大神写的这篇文章

#import "UIViewController+Tracking.h"
#import 

@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]));}

参考链接

http://nshipster.com/method-swizzling/

你可能感兴趣的:(Method Swizzling)