先说需求:
我们现在想在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 指向的是方法的具体实现。