我们知道,如果在分类(category)中重写了类的方法,那么在调用对应方法时,被调用的代码是分类的实现代码。其中的原因并不是“分类的实现覆盖了原类方法的实现”,而是在加载分类方法列表的时候,追加在类方法列表前面。类原有的方法实现还在方法列表中,当在查找时,会先找到分类的实现(如果不太熟悉,还请看深入理解Objective-C:Category)。
知道了分类方法“覆盖”原类方法的实现原理之后,我们是否能够做到,保证我们调用的实现代码是原类的实现,而不是分类的实现?利用 Runtime ,我们可以轻易做到,这里我提供了一种实现方法,仅仅当做实验,可能实际开发没有使用场景。
在此之前需要熟悉一下 Method (类的方法)的结构,到 runtime.h 页面我们能够找到如下代码:
/// An opaque type that represents a method in a class definition.
typedef struct objc_method *Method;
接下来到苹果的 runtime 开源代码中找到 Method 对应的结构体是
struct method_t {
SEL name;///方法的名字,标识符
const char *types;///方法签名
IMP imp;///方法的实现地址
/// 省略
};
我的想法是:新增一个 Method ,当调用该方法时需要改变调用的方式, [obj sel]
变为 [obj pi_performSeleter:sel]
。实现上面有两个问题,
1、如何找到类原有方法的 IMP ?
2、 pi_performSeleter 如何保证通用性?
这里,第二问题 YYKit 作者提供了较好的实现,可以拿来使用(谢谢大神)。接下来看看如何解决第一个问题。
查看 Runtime 提供的 API ,我们能够找到 IMP _Nullable class_getMethodImplementation(Class _Nullable cls, SEL _Nonnull name)
,可以拿来试试,结果会让你失望的,因为这个方法返回的 IMP 是分类的实现(具体源码分析见后续文章),这样的结果也是合情合理的,因为这个 API 在寻找 IMP 的路径就是查找方法的路径,找到的 IMP 就是分类的实现。
如果你知道 Class 的结构,方法存放在 class->getdata()->methods
数组中,寻找方法的实现时,是顺着方法列表找的,最初的实现 IMP 在分类同名方法 IMP 后面,我们可以使用 SEL 倒序寻找,找到的第一个 IMP 就是类原有的实现。接着就是新增一个 Method(新的 SEL ,寻找到的 IMP ,原方法的签名)。
具体的代码如下:
static NSString *const s_pi_selector_prefix = @"pi_origin_";
static SEL pi_aliasForSelector(SEL selector) {
NSCParameterAssert(selector);
return NSSelectorFromString([NSString stringWithFormat:@"%@_%@",s_pi_selector_prefix,NSStringFromSelector(selector)]);
}
static void pi_addAliasMethodBaseClassAndSelector(Class cls, SEL sel){
NSCParameterAssert(cls);
NSCParameterAssert(sel);
unsigned int count;
IMP targetIMP = nil; Method targetMethod =nil;
NSString* needSelName = NSStringFromSelector(sel);
while (cls && (targetMethod == nil)) {
Method *methods = class_copyMethodList(cls, &count);
for (int i = count-1; i >= 0; i--) {
Method method = methods[i];
NSString *methodName = NSStringFromSelector(method_getName(method));
if ([methodName isEqualToString:needSelName]) {/// find
targetIMP = ((struct pi_method_t*)method)->imp;
targetMethod = method;
break;
}
}
free(methods);
if (targetMethod) { /// break
break;
}
cls = class_getSuperclass(cls);
}
SEL newSEL = pi_aliasForSelector(sel);
if (targetMethod && targetIMP && !class_respondsToSelector(cls, newSEL)) {
class_addMethod(cls, newSEL, targetIMP, method_getTypeEncoding(targetMethod));
}
}
Demo地址:PrimalImplement
为了调用类原有的实现,我们还可以怎么做呢?这里还有两种思路:
1、还可以利用
method_exchangeImplementations
交换分类方法的 IMP 和原有实现的 IMP
2、利用class_replaceMethod
替换分类的 Method
上面两种实现后可以不用修改调用方法的一般方式 [obj sel]
,但是这样的设计并不好,分类的实现莫名就不起作用了;还有就是如果多个分类重写的一个方法,如何保证调用的一定是类原有的实现。