综合技术

Method Swizzling 为什么要先调用 class_addMethod?

微信扫一扫,分享到朋友圈

Method Swizzling 为什么要先调用 class_addMethod?
0

先上个 Swift 中的 demo
Method Swizzling

Swift 中的实现

其实 Swift 中实现原理和 OC 基本一致,只是苹果爸爸不再允许在 Swift 中使用 +load()
+initialize()
方法,这当然难不倒各种大神,那么我就做次农夫山泉。。。

Swizzling

先抽取 swizzling 的实现到 NSObject
的扩展当中:

extension NSObject {
    static func swizzlingForClass(_ forClass: AnyClass, originalSelector: Selector, swizzledSelector: Selector) {
        guard let originalMethod = class_getInstanceMethod(forClass, originalSelector),
              let swizzledMethod = class_getInstanceMethod(forClass, swizzledSelector) else {
            return
        }
        
        let isAddSuccess = class_addMethod(forClass, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod))
        if isAddSuccess {
            class_replaceMethod(forClass, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod))
        } else {
            method_exchangeImplementations(originalMethod, swizzledMethod)
        }
    }
}
复制代码

可以看到核心实现和 OC 是完全一致的,那么剩下的就是模拟 OC 版本实现中的 +load()
dispatch_once

dispatch_once

我们用 viewDidLoad
来做个 dispatch_once
的示范:

extension UIViewController {
    static func swizzleViewDidLoad() {
        _ = self.swizzleMethod
    }
    
    @objc func swizzled_viewDidLoad() {
        swizzled_viewDidLoad()
        print("嘻嘻")
    }
    
    private static let swizzleMethod: Void = {
        let originalSelector = #selector(viewDidLoad)
        let swizzledSelector = #selector(swizzled_viewDidLoad)
        swizzlingForClass(UIViewController.self, originalSelector: originalSelector, swizzledSelector: swizzledSelector)
    }()
}
复制代码

在 Swift 中 static let
这样声明的变量其实已经用到 dispatch_once
,而且 static
自带 lazy
属性,要在封装函数 swizzleViewDidLoad
被调用时候才调用。

+load()

OC 中 +load()
方法会在类被装载时调用,确保需要用到的方法都是被 Swizzling 过的。Swift 中可以在 AppDelegate
init
方法中手动调用 swizzle 方法模拟 +load()
实现。

class AppDelegate: UIResponder, UIApplicationDelegate {
    override init() {
        super.init()
        UIViewController.swizzleViewDidLoad()
    }
}
复制代码

为什么要先调用 class_addMethod?

class_addMethod
这个方法是很容易被人忽视的,对于 Swizzling 一节中的代码,还有一种常见的写法:

extension NSObject {
    static func swizzlingForClass(_ forClass: AnyClass, originalSelector: Selector, swizzledSelector: Selector) {
        guard let originalMethod = class_getInstanceMethod(forClass, originalSelector),
              let swizzledMethod = class_getInstanceMethod(forClass, swizzledSelector) else {
            return
        }

        method_exchangeImplementations(originalMethod, swizzledMethod)
    }
}
复制代码

这种方式就只是简单的直接交换了 originalMethod
swizzledMethod
。乍一看貌似没有问题(其实最开始我绞尽脑汁也没想清楚到底哪里不对。。。),但是为什么各路大神都是用的第一种方式呢?网上有种说法:

要先尝试添加原 selector 是为了做一层保护,因为如果这个类没有实现原始方法"originalSel" ,但其父类实现了,那 class_getInstanceMethod 会返回父类的方法。这样 method_exchangeImplementations 替换的是父类的那个方法,这当然不是你想要的。所以我们先尝试添加 originalSel ,如果已经存在,再用 method_exchangeImplementations 把原方法的实现跟新的方法实现给交换掉。

其实这种说法已经算是比较明确问题所在了,但是愚笨的我还是没有想通到底为什就“这当然不是你想要的”了呢。

又是一番绞尽脑汁。。。终于 Biuer 的一下想通了

在举栗子前引用一段对 Selectors、Methods 和 Implementations 理解:

理解 selector, method, implementation 这三个概念之间关系的最好方式是:在运行时,类(Class)维护了一个消息分发列表来解决消息的正确发送。每一个消息列表的入口是一个方法(Method),这个方法映射了一对键值对,其中键值是这个方法的名字 selector(SEL),值是指向这个方法实现的函数指针 implementation(IMP)。 Method swizzling 修改了类的消息分发列表使得已经存在的 selector 映射了另一个实现 implementation,同时重命名了原生方法的实现为一个新的 selector。

假设父类有个方法 method
,子类未重写 method
方法,子类的中想要拿来替换的方法为 swizzledMethod

  1. 用第二种方式进行方法交换

    • 在子类的实例中调用 method
      方法时,确实按预期正常运行的
    • 在父类的实例中调用 method
      方法时,就开始崩溃了。因为方法交换后, method
      方法的 IMP
      其实和子类 swizzledMethod
      IMP
      进行了交换,此时等同于父类调用子类方法,当然会崩溃。
  2. 用第一种方式进行方法交换

    class_addMethod
    先判断了子类中是否有 method
    方法

    • 如果有,则添加失败,直接进行交换
    • 如果没有,则添加成功,将 swizzledMethod
      IMP
      赋值给 method
      这个 Selector
      ,然后在将 method
      IMP
      (其实是父类中的实现)赋值给 swizzledMethod
      这个 Selector
阅读原文...

微信扫一扫,分享到朋友圈

Method Swizzling 为什么要先调用 class_addMethod?
0
稀土掘金

react+Typescript+svg仿番茄闹钟

上一篇

Python 获取当前作用域全部参数的内置方法 locals()

下一篇

评论已经被关闭。

插入图片

热门分类

往期推荐

Method Swizzling 为什么要先调用 class_addMethod?

长按储存图像,分享给朋友