寻找方法在类中最初的实现

我们知道,如果在分类(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] ,但是这样的设计并不好,分类的实现莫名就不起作用了;还有就是如果多个分类重写的一个方法,如何保证调用的一定是类原有的实现。

你可能感兴趣的:(寻找方法在类中最初的实现)