Swift3中的 Method Swizzling
先聊聊Method Swizzling
从 Objective-C 开始, runtime一直是解决坑爹需求和面试装逼的一大利器,然而聊起 runtime 很多人都第一个想到 Method Swizzling,.因为 Objective-C中调用方法都是动态实现的,当运行时的才确定到底执行哪个方法,而 Method Swizzling 就是利用这个特点来解决很多问题.
现在关于 Runtime 和 Method Swizzling 的文章太多了,我推荐一篇:
神经病院Objective-C Runtime出院第三天——如何正确使用Runtime
@一缕殇流化隐半边冰霜 大神写的关于 runtime这几篇,看完基本对 runtime 就没什么问题了吧...
再看看 Swift3.0中的 Method Swizzling
先来看看 swizzling 在 Objective-C 中的注意点:(对比上文链接中)
1.Swizzling应该总在category的 +load中执行 ( Objective-C )
那在 Swift 中, extension 并不是运行时加载的, 因此也没有加载时候就会被调用的类似 +load 的方法. 事实上,Swift 实现的 load 并不是在 app 运行开始就被调用的。基于这些理由,我们使用另一个类初始化时会被调用的方法来进行交换:
open override static func initialize() {
// Method Swizzling
}
这一条部分来自喵神 swiift tips 第二版, 喵神在第三版中删除了 swizzling 这个章节,理由是这部分更多是 Objective-C的内容. 我个人觉得如果在 Swift 中还需要用 Swizzling 这种技术来实现需求, 不如用更 Swifty的方式去解决问题, 函数式或者面向协议等等等😀
2.Swizzling应该总是在dispatch_once中执行
那么,问题来了,在3.0版本 dispatch once 已经被废弃,这怎么办?
刚巧的是前几天群里的老司机 @没故事的卓同学 写了篇 [译]Swift 3 中实现Dispatch once扩展
通过给DispatchQueue实现一个扩展方法来实现 Dispatch once.
至于没什么要 dispatch_once呢? 因为 Swizzling会改变全局状态,所以用dispatch_once来确保无论多少线程都只会被执行一次.
3. Swift自定义类中使用 Method Swizzling
因为Method Swizzling的实现是基于 Objective-C 的动态派发机制,所以有两条限制
1.包含 swizzle 方法的类需要继承自 NSObject
2.如果要 Swizzle 的是 Swift 类型的方法的话,需要将原方法和替换方法都加上 dynamic 标记,以指明它们需要使用动态派发机制
上个 sample:
extension UIViewController {
open override static func initialize() {
struct Static {
static var token = NSUUID().uuidString
}
if self != UIViewController.self {
return
}
DispatchQueue.once(token: Static.token) {
let originalSelector = #selector(UIViewController.viewWillAppear(_:))
let swizzledSelector = #selector(UIViewController.xl_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)
}
}
}
func xl_viewWillAppear(animated: Bool) {
self.xl_viewWillAppear(animated: animated)
print("xl_viewWillAppear in swizzleMethod")
}
}
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()
}
}