首先来看下什么是方法的交换,也是很多人口中的runtime黑魔法。方法交换,就是把Method的IMP指针给交换了。关于这两Method和IMP到底是啥?请先看这里
本文的demo可以下载调试,不过代码量很少,建议自己动手撸一遍。参考链接
先直接上代码吧:
- (void)textSwizzLing {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
Method originMethod = class_getInstanceMethod([self class],
@selector(orignMethod));
Method swizzMthod = class_getInstanceMethod([self class], @selector(swizzLingMethod));
BOOL didAdd = class_addMethod([self class], @selector(orignMethod),
method_getImplementation(swizzMthod), method_getTypeEncoding(swizzMthod));
//,把swiz的实现方法,添加给originmethod,如果没有实现origin方法,则添加成功,否则添加失败。
如果直接exchange的话,那么就会覆盖掉父类的方法。,
if (didAdd) {
class_replaceMethod([self class],@selector(swizzLingMethod),
method_getImplementation(originMethod),method_getTypeEncoding(originMethod));
}else {
method_exchangeImplementations(originMethod, swizzMthod);
//如果origin的方法,本类中已经实现了,直接交换就好了。
}
});
[self performSelector:@selector(orignMethod) withObject:nil];
}
整个逻辑是这样的:把swizz的实现,添加给origin,如果origin的方法,已经在本类中实现了,那么直接exchange就可以了。 这是一种情况。因为addMethod的官方文档说,如果origin在本类有实现,那么会添加失败。
看了这一段代码,其实很多教程教你这么写,可是你知道为什么这么写么?
如果didAdd失败会怎样?
下面分析一下第二种情况,请看代码(or 自己写一下):
Class cls = [UILabel class];
SEL originalSelector = @selector(willMoveToSuperview:);
SEL swizzledSelector = @selector(myWillMoveToSuperview:);
Method originalM = class_getInstanceMethod(cls, originalSelector);
Method swizzedM = class_getInstanceMethod(cls, swizzledSelector);
NSLog(@"%p",method_getImplementation(swizzedM));
NSLog(@"%p",method_getImplementation(originalM));
BOOL add = class_addMethod(cls, originalSelector, method_getImplementation(swizzedM), method_getTypeEncoding(swizzedM));
NSLog(@"%p",method_getImplementation(swizzedM));
NSLog(@"%p",method_getImplementation(originalM));
if (add) {
class_replaceMethod(cls, swizzledSelector, method_getImplementation(originalM), method_getTypeEncoding(originalM));
NSLog(@"%p",method_getImplementation(swizzedM));
NSLog(@"%p",method_getImplementation(originalM));
}else {
method_exchangeImplementations(originalM, swizzedM);
}
这个willMoveToSuperview:是父类中实现的。UILabel中并没有实现这个函数,所以,Add为YES,那么,通过Add把swizz的实现传给了willMoveToSuperview。那么, class_replaceMethod(cls,swizzledSelector, method_getImplementation(originalM), method_getTypeEncoding(originalM));就是把origin的实现传给swizzled这个函数名了?
打印一下我发现,可是这两个函数都指向的同一个实现方法啊。意味着,origin确实指向了swizz的实现,可是swizz的实现还是swizz的实现?郁闷呢,怎么会这样?不应该是交换么,说好的SwizzLing呢,那岂不是调用两个都会执行同一个函数。既然网上的资料说不清,那我还是看官方文档吧
终于找到了罪魁祸首,注意看最后一句话,大概意思就是去这个类的 父类中去找这个方法。
现在明白了,class_getInstanceMethod原来拿到的是父类中的方法,后面这句
class_replaceMethod(cls,swizzledSelector, method_getImplementation(originalM), method_getTypeEncoding(originalM));
是把父类的实现,覆盖了swizz的实现,所以两个打印出来的地址会是一样.所以在打印 NSLog(@"%p",method_getImplementation(originalM));
的时候,还是打印的父类中的实现,并不是现在新加的实现。
那么现在稍微多了解一点,class_getInstanceMethod会先在子类找还是先在父类找?(是想知道,如果我在子类中重写了父类中的方法,那么这个函数是拿到的子类中的还是父类中的)
SEL originalSelector = @selector(willMoveToSuperview:);
Method originalM = class_getInstanceMethod(cls, originalSelector);
Method supperM = class_getInstanceMethod([UIView class], originalSelector);
NSLog(@"%p",method_getImplementation(originalM));
NSLog(@"%p",method_getImplementation(supperM));
并且实现这个方法:
- (void)willMoveToSuperview:(UIView *)view {
NSLog(@"1");
}
然后打印发现,两个的地址不一样,所以这个方法class_getInstanceMethod,是先在本类中找实现,找不到再去父类找。证明,在子类中找到了,就不会去父类中找了。
现在理解了为什么要这样写,那结合关联对象试着给label写个属性。
先介绍三个方法
1,objc_setAssociatedObject(self, @selector(isSystemColor), @(isSystemColor),
OBJC_ASSOCIATION_RETAIN_NONATOMIC);
第一个参数是要关联的对象。第二个是一个能够唯一辨识的key(建议用@selector(get方法)),第三个参数是要关联的值,第四个是关联的方案。这个根据枚举的定义,也表达的很清楚了,基本上和@property的属性类型对应。
2, objc_getAssociatedObject(self, _cmd);
第一个参数要拿哪个对象的关联对象
第二个参数,是这个关联参数的key值,看了这里的都知道,每个都有两个隐藏参数,一个是self,一个是_cmd,也就是@selector()
3,objc_removeAssociatedObjects,基本不用,会移除self的所有关。如果要移除,直接传nil就可以移除了。
这里,我理解关联对象的意思,就是把这个数值以一种字典的形式存起来了,需要的时候再用key值取出来。
而存储方案,就是关联方案,不过我也没有测到到底这几种在ARC下的不同,以及对象的移除时机,所以这里先不做介绍。
demo中用的是给label改变背景颜色,一般项目中会用到改变字体,不过背景颜色来得明显一些。这里结合关联对象,是因为实际项目中,有的label不需要改变字体,所以我们需要传入一个值判断是否设为统一的字符,并且又改最少的代码,这种少数的label,就给这个对象设为NO就可以了。
具体代码可以参看demo
最后不要忘记移除关联对象
objc_setAssociatedObject(self, @selector(isSystemColor), nil,
OBJC_ASSOCIATION_RETAIN_NONATOMIC);