iOS 中对Method Swizzling 的理解

对Method Swizzling 的理解。（更新了swift3.0 中MethodSwizzling 的用法）

首先，在OC中调一个方法，其实就是向对象发一条消息，根据selector进行查找，那么利用Runtime的特性，可以在运行时把selector对应方法的实现给换掉，Method Swizzling正是利用了这个原理
下面是示例
首先建立UIViewController的一个分类UIViewController+Swizzling
在.m文件中重写以下方法

+(void)load{

    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
//这里使用dispatch_once 是因为虽然load方法只会在初始化的时候被调用一次，但不能确保别的程序员调用你load的方法啊，所以这里用dispatch_once来保证以下。
        Class class = [self class];
        
        SEL originalSelector = @selector(viewDidLoad);
        SEL swizzledSelector = @selector(my_viewDidLoad);
        
        Method originalMethod = class_getInstanceMethod(class, originalSelector);
        Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
        //给viewController增加一个叫viewDidload的方法 具体实现的内容为swizzMehthod的实现，class_addMethod 这个方法若干在本类没有找到的话会去父类那里找，实在找不到了再给它增加。
        BOOL success = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
        
        if (success) {
        //如果添加成功的话，证明本身没有viewDidload这个方法，那viewDidLoad的这个方法对应的实现是my_viewDidLoad 的实现。那名为my_viewDidLoad的方法的实现实际就变成是viewDidLoad的实现。
  
            class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
//这里的method_getTypeEncoding 方法之后得到的是 const char* 类型，字符指针， 如果写死可以写成”v@：@”（）
"v@:@",解释v-返回值void类型,@-self指针id类型,:-SEL指针SEL类型,@-函数第一个参数为id类型
        } else {
            NSLog(@"fail");
            //本身就存在ViewDidLoad这个方法
            method_exchangeImplementations(originalMethod, swizzledMethod);
        }
    });

}


-(void)my_viewDidLoad{
    //父类的实现
    [self my_viewDidLoad];
//这里看是循环调用了， 但由于方法交换，viewController在调my_viewDidLoad的时候实际是进到这个方法里面来了，而上面又掉了my_viewDidLoad实际掉的是原来viewdidload的方法，所以这样就相当于在所有的UIViewController的viewDidload方法下增加了下面的逻辑，适合比如说在所有的页面增加统计的API ，就不用每个页面逐一的去写了
    //下面统一需要添加给系统的逻辑
    NSLog(@"self class%@",self.class);
    
}

===========================================================

import UIKit

extension UIViewController {
    
    open override static func initialize() {
        struct Static {
            static var token = NSUUID().uuidString
        }

    DispatchQueue.once(token: Static.token) {
            let originalSelector = #selector(UIViewController.viewDidLoad)
            let swizzledSelector = #selector(UIViewController.newViewDidLoad)
            
            let originalMethod = class_getInstanceMethod(self, originalSelector)
            let swizzledMethod = class_getInstanceMethod(self, swizzledSelector)
            
            let didAddMethod = class_addMethod(self, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod))
            
            if didAddMethod {
                class_replaceMethod(self, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod))
            } else {
                method_exchangeImplementations(originalMethod, swizzledMethod);
            }
  
        }
        
    }
    
    // MARK: - Method Swizzling
    func newViewDidLoad() {
       self.newViewDidLoad()
    ／／这里写要对ViewController里面viewDidLoad 要增加的方法
 print（“add something to all viewControllers”）
        }
    }
    
}

extension DispatchQueue {
    private static var onceTracker = [String]()
    
    open class func once(token: String, block:() -> Void) {
        objc_sync_enter(self)
        defer { objc_sync_exit(self) }
        
        if onceTracker.contains(token) {
            return
        }

        onceTracker.append(token)
        block()
  }
}