在学习 runtime 的时候,给自己做了一个笔记,是关于消息转发机制的。先新建一个 Person 类,在 .h 文件声明一个方法: sendMessage。
// Person.h
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import
NS_ASSUME_NONNULL_BEGIN
@interface Person : NSObject
- (void)sendMessage:(NSString *)msg;
@end
NS_ASSUME_NONNULL_END
但我不在 .m 文件里实现它。
//
// Person.m
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import "Person.h"
#import
#import "SpareWheel.h"
// OC 消息转发机制
@implementation Person
@end
这个时候我去实例化这个类,然后调用声明却未实现的方法。
//
// ViewController.m
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import "ViewController.h"
#import "Person.h"
@interface ViewController ()
@end
@implementation ViewController
- (void)viewDidLoad {
[super viewDidLoad];
self.view.backgroundColor = UIColor.whiteColor;
Person *p = [[Person alloc] init];
[p sendMessage:@"hello"];
}
@end
结果可想而知,程序崩了。
那怎么解决这个问题呢?这个时候就需要使用 runtime 的消息转发来解决,声明了,不需要实现也能跑起来。
在使用 runtime 之前需要在 Person.m 文件里导入它的头文件。
//
// Person.m
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import "Person.h"
#import
OC 消息转发机制在我学习的过程中,我个人理解可以分为 3 个阶段,在这里如果有什么好的建议请及时提,大家相互学习。
第一阶段:动态方法解析
第二阶段:查找备用接收者
第三阶段:完整的消息转发。第三阶段又分两个小结: 1. 方法签名。2. 消息转发
1.先从第一个阶段,动态方法解析开始实现,需要实现 resolveInstanceMethod
这个方法,字面意思,翻译过来就是 解析实例方法
。
+ (BOOL)resolveInstanceMethod:(SEL)sel {
NSString *methodName = NSStringFromSelector(sel);
if ([methodName isEqualToString:@"sendMessage:"]) {
// 动态的添加方法实现, sel ---- IMP 的实现
return class_addMethod(self, sel, (IMP)sendMessage, "v@:@");
}
return NO;
}
这个方法需要传入一个 SEL 类型的参数,返回值为 BOOL 类型。那这个 SEL 是什么呢?就是我们的方法编号
。
解释一下这段代码:
1.获取方法名:NSString *methodName = NSStringFromSelector(sel);
。
2.判断是否有这个方法名,如果有,动态的添加方法实现。如果没有,返回 NO。
3.动态的添加方法实现要传四个参数,(self, sel, IMP,"v@:@")
。前面两个参数很好理解,第三个参数 IMP
是什么?其实这里就是填你的方法名,当然,不是你头文件声明的方法哦,而是你在 .m 文件实现的方法。
void sendMessage(id self, SEL _cmd, NSString *msg) {
NSLog(@"msg = %@", msg);
}
可以看到这个写法很明显不是 OC 的写法,是 C 的写法。
那第四个参数是什么意思呢?其实就是上面用 C 语言写的函数对应的 4 个参数。v -> 返回值类型(void), @ -> id 类型(一般是 self), ":" ->: 方法编号, @ -> id 类型(NSString)
。
到这里,第一阶段就完成了,可以跑一下项目,发现程序并没有崩,而且还打印了 msg。
到这里,第一阶段:动态方法解析
就完成了。
2. 第二阶段:查找备用接收者
在执行第二阶段之前,我们需要创建一个 SpareWheel
的OC文件,然后在 .m 文件里实现一个 sendMessage
的方法。
//
// SpareWheel.m
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import "SpareWheel.h"
@implementation SpareWheel
- (void)sendMessage:(NSString *)msg {
NSLog(@"message = %@", msg);
}
@end
然后在 Person.m 文件导入刚创建的 SpareWheel.h 文件
//
// Person.m
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import "Person.h"
#import
#import "SpareWheel.h"
然后实现forwardingTargetForSelector
这个方法。
- (id)forwardingTargetForSelector:(SEL)aSelector {
NSString *methodName = NSStringFromSelector(aSelector);
if ([methodName isEqualToString:@"sendMessage:"]) {
// 返回备用接收者
return [SpareWheel new];
}
return [super forwardingTargetForSelector:aSelector];
}
可以看到在里面的代码几乎和上面的一样。
1.获取方法名:NSString *methodName = NSStringFromSelector(sel);。
2.判断是否有这个方法名,如果有, 返回备用接收者。如果没有,返回 [super forwardingTargetForSelector:aSelector],就是它自己本身。
然后把第一阶段的代码注释调,来看一下运行结果。
可以看到,第二阶段的我们也搞定了。
3. 第三阶段:完整的消息转发
上面提到分两个步骤,现实第一个步骤,方法签名 methodSignatureForSelector
。
// 1. 方法签名
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
NSString *methodName = NSStringFromSelector(aSelector);
if ([methodName isEqualToString:@"sendMessage:"]) {
return [NSMethodSignature signatureWithObjCTypes:"v@:@"];
}
return [super methodSignatureForSelector:aSelector];
}
可以看到,实现的思路和第一第二阶段一样,"v@:@"
,这个在上面也有解释,这里就不多说了。第一步完成后,开始第二步,消息转发forwardInvocation
。
// 2. 消息转发
- (void)forwardInvocation:(NSInvocation *)anInvocation {
SEL sel = [anInvocation selector];
SpareWheel *sp = [SpareWheel new];
if ([sp respondsToSelector:sel]) {
[anInvocation invokeWithTarget:sp];
}else {
[super forwardInvocation:anInvocation];
}
}
假如在备用接收者也没有找到,它就封装这个方法(sendMessage)的一些信息,然后给它转发出去,所有的信息都放在了 NSInvocation
这个类里面。下面解释一下这段代码:
1.获得方法编号:SEL sel = [anInvocation selector];
。
2.实例化一个对象(备用接收者,假设它也去备用接收者查找),SpareWheel *sp = [SpareWheel new];
。
3.判断这个类里面有没有这个方法的实现 ,if ([sp respondsToSelector:sel]) {}else{}
。
4.如果有,就指定这个方法的接收者为这个类, [anInvocation invokeWithTarget:sp];
。
5.如果没有,就走自己的方法,[super forwardInvocation:anInvocation];
。
最后,把第二阶段的代码注释,看一下运行结果,可以看到,它一样会去找我们的备用接收者,将 message 打印出来了。
4.最后还有一个问题,那如果通过上面的三个方法都没有找到,怎么才能保证程序不会崩溃呢?
这个时候我们可以实现这个方法doesNotRecognizeSelector
。
- (void)doesNotRecognizeSelector:(SEL)aSelector {
NSLog(@"未知错误");
}
然后把上面的代码注释掉,来看一下运行结果。
可以看到,程序运行起来并没有任何问题,并且,还打印了相关的东西。到这里,关于 runtime 的消息转发机制就暂时结束了。
最后,我自己写了一个思维导图,可以根据思维导图进行对比。