Swift4 使用 Method Swizzling 实现 UIViewController方法的重写

先说需求：
我们现在想在ViewController的生命周期函数（ViewDidLod,viewWillAppear等）上加上一些统计和打点。传统的方法就是每个方法里面插一句代码，但是因为目前项目已经有很多UIViewController了，一个一个加很麻烦，那有没有简单的方法可以一劳永逸呢？

答案是有的，通过Method Swizzling实现。先说怎么做，再解释原理。
(以下代码通过Swift4实现)

private let swizzling: (UIViewController.Type) -> () = { viewController in
    
    let originalSelector = #selector(viewController.viewWillAppear(_:))
    let swizzledSelector = #selector(viewController.proj_viewWillAppear(animated:))
    
    let originalMethod = class_getInstanceMethod(viewController, originalSelector)
    let swizzledMethod = class_getInstanceMethod(viewController, swizzledSelector)
    
    method_exchangeImplementations(originalMethod!, swizzledMethod!)
}

extension UIViewController {
    
    open class func initClass() {
        guard self === UIViewController.self else { return }
        swizzling(self)
    }
    
    @objc func proj_viewWillAppear(animated: Bool) {
        self.proj_viewWillAppear(animated: animated)
        
        let viewControllerName = NSStringFromClass(type(of: self))
        print("viewWillAppear: \(viewControllerName)")
    }
}

在AppDelegate里初始化

@UIApplicationMain
class AppDelegate: UIResponder, UIApplicationDelegate {
    override init() {
        super.init()
        UIViewController.initClass()
    }
}

以上代码通过重写viewWillAppear再后面添加了一句print。

OK，接下来说原理：
参考Hook 原理之 Method Swizzling

Method Swizzling 其实就是利用了 runtime 机制，替换了调用的方法（method）的实现（imp）
runtime 运行时机制，主要是在 OC 和 C 语言（或汇编语言）之间架了一座桥梁。调用方法的本质是发送消息，方法调用是 OC 中的，而消息发送，找到方法的入口则是 C （或汇编）中的。这其中转换的过程，就是 runtime。runtime 是一个使用 C 语言以及汇编写的动态库。它封装了一些 C 语言的结构体和函数，这些函数可以让使用者在运行时创建、查看、修改类，对象以及方法等。同时也执行着较为底层的传递消息，寻找方法的执行代码的操作。

Method Swizzling，正如之前所提到的，本质就是替换了方法的实现。在 OC 中调用一个方法，这个方法的本质是一条消息，而这条消息的本质，就是一个 selector。也就是说 selector 就代表着这个方法，比如在程序中，我们经常会使用 @selector() 这样的方式来调用另一个方法。

每个类都存储着一个方法列表，又叫调度表（dispatch table），这个表是 selector 与方法的具体实现 IMP 的对应关系，类似于函数名和函数指针。IMP 指向的是方法的具体实现。

Swift4 使用 Method Swizzling 实现 UIViewController方法的重写

推荐阅读更多精彩内容