Method Swizzling 问题
网上关于使用这个方法的教程很多,但是跟着教程走一遍发现一些问题,下面我来说说我对这个方法的理解.
void swizzleMethod(Class class, SEL originalSelector, SEL swizzledSelector)
{
// the method might not exist in the class, but in its superclass
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
// class_addMethod will fail if original method already exists
BOOL didAddMethod = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
// the method doesn’t exist and we just added one
if (didAddMethod) {
class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
}
else {
method_exchangeImplementations(originalMethod, swizzledMethod);
}
}
上面是核心代码,目的是交换两个方法实现.
SEL originalSelector 理解相当于源方法的Name
SEL swizzledSelector 可以理解为调换的方法Name
BOOL didAddMethod = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
要先尝试添加originalSelector是为了做一层保护，因为如果这个类没有实现originalSelector ，但其父类实现了，那 class_getInstanceMethod 会返回父类的方法.这时候违背了我们交换方法的意愿.
现在具体的代码讲解完了,实践一下。举一个老生常谈的Logging方法,埋点问题。主要是探索用户习惯,对 App 的用户行为进行追踪和分析。
建一个UIViewController的类别UIViewController+Logging
+ (void)load{
swizzleMethod([self class], @selector(viewWillAppear:), @selector(new_viewWillAppear:));
}
void swizzleMethod(Class class, SEL originalSelector, SEL swizzledSelector)
{
// the method might not exist in the class, but in its superclass
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
// class_addMethod will fail if original method already exists
BOOL didAddMethod = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
// the method doesn’t exist and we just added one
if (didAddMethod) {
class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
}
else {
method_exchangeImplementations(originalMethod, swizzledMethod);
}
}
- (void)new_viewWillAppear:(BOOL)animated{
}
+load: 是个特例，当一个类被读到内存的时候， runtime 会给这个类及它的每一个类别都发送一个 +load: 消息。
注意: new_viewWillAppear里并没有写方法,正常应该是:
- (void)new_viewWillAppear:(BOOL)animated{
[self new_viewWillAppear:animated];
}
这时候调用自身方法实际是调用的原viewWillAppear的IMP,因为上面已经将两个方法的IMP交换了。
跑一下代码,发现会先走的是子类的viewWillAppear方法, 然后才会走你类别的new_viewWillAppear方法.你发现你晕了,我已经交换了IMP啊,为啥还会走子类的viewWillAppear,问题出现在:
+ (void)load{
swizzleMethod([self class], @selector(viewWillAppear:), @selector(new_viewWillAppear:));
}
中的[self class],打印它,它是:UIViewController。
原来我们hook了父类的viewWillAppear,交换的实际是UIViewController 中的viewWillAppear方法,将子类viewWillAppear中的[super viewWillAppear:animated]删除,你会发现,此时类别中的- (void)new_viewWillAppear:(BOOL)animated不会走了。
现在你会问:既然父类的方法被类别中的替换了,那么子类也应该也被替换.
答案:
方法并不会重写,只会在method list里面插入的位置靠前,category写一个跟类里面名字 方法签名一模一样的方法 还可以有办法可以调用到类原来的这个方法。
注意:在交换方法的时候一定要注意,hook 的是哪个类,这个类是不是你想要交换的方法的那个类。
欢迎各位同学指正错误,我会努力及时改正!
