对Method Swizzling 的理解。(更新了swift3.0 中MethodSwizzling 的用法)
首先,在OC中调一个方法,其实就是向对象发一条消息,根据selector进行查找,那么利用Runtime的特性,可以在运行时把selector对应方法的实现给换掉,Method Swizzling正是利用了这个原理
下面是示例
首先建立UIViewController的一个分类UIViewController+Swizzling
在.m文件中重写以下方法
+(void)load{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
//这里使用dispatch_once 是因为虽然load方法只会在初始化的时候被调用一次,但不能确保别的程序员调用你load的方法啊,所以这里用dispatch_once来保证以下。
Class class = [self class];
SEL originalSelector = @selector(viewDidLoad);
SEL swizzledSelector = @selector(my_viewDidLoad);
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
//给viewController增加一个叫viewDidload的方法 具体实现的内容为swizzMehthod的实现,class_addMethod 这个方法若干在本类没有找到的话会去父类那里找,实在找不到了再给它增加。
BOOL success = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
if (success) {
//如果添加成功的话,证明本身没有viewDidload这个方法,那viewDidLoad的这个方法对应的实现是my_viewDidLoad 的实现。那名为my_viewDidLoad的方法的实现实际就变成是viewDidLoad的实现。
class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
//这里的method_getTypeEncoding 方法之后得到的是 const char* 类型,字符指针, 如果写死可以写成”v@:@”()
"v@:@",解释v-返回值void类型,@-self指针id类型,:-SEL指针SEL类型,@-函数第一个参数为id类型
} else {
NSLog(@"fail");
//本身就存在ViewDidLoad这个方法
method_exchangeImplementations(originalMethod, swizzledMethod);
}
});
}
-(void)my_viewDidLoad{
//父类的实现
[self my_viewDidLoad];
//这里看是循环调用了, 但由于方法交换,viewController在调my_viewDidLoad的时候实际是进到这个方法里面来了,而上面又掉了my_viewDidLoad实际掉的是原来viewdidload的方法,所以这样就相当于在所有的UIViewController的viewDidload方法下增加了下面的逻辑,适合比如说在所有的页面增加统计的API ,就不用每个页面逐一的去写了
//下面统一需要添加给系统的逻辑
NSLog(@"self class%@",self.class);
}
===========================================================
import UIKit
extension UIViewController {
open override static func initialize() {
struct Static {
static var token = NSUUID().uuidString
}
DispatchQueue.once(token: Static.token) {
let originalSelector = #selector(UIViewController.viewDidLoad)
let swizzledSelector = #selector(UIViewController.newViewDidLoad)
let originalMethod = class_getInstanceMethod(self, originalSelector)
let swizzledMethod = class_getInstanceMethod(self, swizzledSelector)
let didAddMethod = class_addMethod(self, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod))
if didAddMethod {
class_replaceMethod(self, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod))
} else {
method_exchangeImplementations(originalMethod, swizzledMethod);
}
}
}
// MARK: - Method Swizzling
func newViewDidLoad() {
self.newViewDidLoad()
//这里写要对ViewController里面viewDidLoad 要增加的方法
print(“add something to all viewControllers”)
}
}
}
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()
}
}