我们平时在使用三方时会比较喜欢根据自己的意愿去封装一下三方,一般都会继承三方创建一些子类,用于对应不同的需求。有时三方会给我们留出接口供我们使用,但有时候并没有。如果我们去给三方直接扩充这些接口是不友好的,特别是CocoaPods上的三方,所以一般我们会在子类里重写三方的方法。
三方中在.h中声明的方法比较好重写,直接在子类里重写方法,如果需要重调父类原有方法可以用super直接调用。而一些私有方法并没有在.h中声明,这时我们也可以直接重写父类方法,但是相对于想要重调父类方法,就不能用super直接调用来实现。这里super只是一个编译器修饰符,是一个指向父类标志,并不是对象的父类实例。所以你再用super调用父类的私有方法是没有作用的。这时我们有几种选择可以帮我们实现父类私有方法的调用。
首先可以用Method Swizzling(方法交换)来帮我们实现,但是我并不喜欢使用这个方法,因为此方法会引起全局的方法指针交换,多人开发中如果没有交流好很容易出现一些纰漏等问题,所以这里我就不具体介绍此方法了。
第二种方法是我们可以利用runtime的消息发送机制,为我们的对象调用其父类的私有方法。主要用到objc_msgSendSuper方法,但是直接调用是不安全的,因为你不能确定父类是否含有这个方法,因为三方也可能进行升级等操作,所以调用之前需要判断父类是否含有此方法,举个例子,代码如下:
// 判断一个父类是否包含某个方法(包含私有方法)
- (BOOL)lj_containsSuperMethod:(NSString *)methodName {
unsigned int outCount = 0;
Method *methods = class_copyMethodList([SuperClass class], &outCount);
for (int i = 0; i < outCount; i ++) {
Method method = methods[i];
SEL methodNameSEL = method_getName(method);
if ([methodName isEqualToString:NSStringFromSelector(methodNameSEL)]) {
free(methods);
return YES;
}
}
free(methods);
return NO;
}
这个方法是利用遍历父类中的所有方法来判断是否包含某一方法,这样做其实也有一定的弊端,比如当这个父类方法很多时,而你又需要频繁调用此方法时就会引起不必要的消耗。如果父类包含此方法就直接利用objc_msgSendSuper发送消息就可以了,需要声明#import <objc/message.h>。例子代码如下:
- (void)privateMethod {
if ([self lj_containsSuperMethod:@"privateMethod"]) {
struct objc_super super_obj;
super_obj.receiver = self;
super_obj.super_class = [SuperClass class];
objc_msgSendSuper(&super_obj, sel_registerName("privateMethod"));
}
}
这里调用objc_msgSendSuper方法需要将ENABLE_STRICT_OBJC_MSGSEND设置为NO。
第三种是先获取父类方法,然后利用构建的方式来调用。判断父类是否包含此方法是根据构建出来的方法是否存在来判断的,相比于上一个方法减少了遍历的过程。例子代码如下:
- (void)privateMethod {
Method method = class_getInstanceMethod([SuperClass class], sel_registerName("privateMethod"));
void (*super_func)(id,SEL) = (void *)method_getImplementation(method);
if (super_func) super_func(self, sel_registerName("privateMethod"));
}
if (super_func) 是判断父类是否包含的此方法,如果含有此方法则调用,否则不调用。