Swift中实现Method Swizzling

产品提出一个需求，进行页面和事件追踪
于是傻乎乎的在每个ViewControlelr的viewWillAppear和viewDidAppear中都调用友盟的接口。
后来接触到Method Swizzling，才了解到这个基于Runtime的强大的方式。
Method Swizzling是一个非常有用的技巧，可以全局hook，进行行为统计。

Method Swizzling原理

每个类都有一个方法列表，存放着selector的名字和方法实现的映射关系。IMP有点类似函数指针，指向具体的Method实现。

在Objective-C中调用一个方法，其实是向一个对象发送消息，查找消息的唯一依据是selector的名字。利用Objective-C的动态特性，可以实现在运行时偷换selector对应的方法实现，达到给方法挂钩的目的。

在实现Method Swizzling时，核心代码主要就是一个runtime的C语言API：

OBJC_EXPORT void method_exchangeImplementations(Method m1, Method m2) 
 __OSX_AVAILABLE_STARTING(__MAC_10_5, __IPHONE_2_0);

Swift中是如何使用的呢？

要在 Swift 自定义类中使用 Method Swizzling 有两个必要条件：

包含 swizzle 方法的类需要继承自 NSObject

需要 swizzle 的方法必须有动态属性（dynamic attribute）

extension UIViewController {

public class func initializeMethod() {
    // Make sure This isn't a subclass of UIViewController, So that It applies to all UIViewController childs
    
    if self != UIViewController.self {
        return
    }
    
    /**
     let _:() = {
     
     }()
    */

        
        let originalSelector = #selector(UIViewController.viewWillAppear(_:))
        let swizzledSelector = #selector(UIViewController.swizzled_viewWillAppear(animated:))
        
        let originalMethod = class_getInstanceMethod(self, originalSelector)
        let swizzledMethod = class_getInstanceMethod(self, swizzledSelector)
        //在进行 Swizzling 的时候,需要用 class_addMethod 先进行判断一下原有类中是否有要替换方法的实现
        let didAddMethod: Bool = class_addMethod(self, originalSelector, method_getImplementation(swizzledMethod!), method_getTypeEncoding(swizzledMethod!))
        //如果 class_addMethod 返回 yes,说明当前类中没有要替换方法的实现,所以需要在父类中查找,这时候就用到 method_getImplemetation 去获取 class_getInstanceMethod 里面的方法实现,然后再进行 class_replaceMethod 来实现 Swizzing
        
        if didAddMethod {
            class_replaceMethod(self, swizzledSelector, method_getImplementation(originalMethod!), method_getTypeEncoding(originalMethod!))
        } else {
            method_exchangeImplementations(originalMethod!, swizzledMethod!)
        }
    
}


@objc func swizzled_viewWillAppear(animated: Bool) {
    //需要注入的代码写在此处
    self.swizzled_viewWillAppear(animated: animated)
}

}

最后在didFinishLaunchingWithOptions中实现：

UIViewController.initializeMethod()