在学习 runtime 的时候，给自己做了一个笔记，是关于消息转发机制的。先新建一个 Person 类，在 .h 文件声明一个方法: sendMessage。
// Person.h
// 01-RuntimeSendMessage
//
// Created by Mac on 2019/10/30.
// Copyright © 2019 Mac. All rights reserved.
//
#import <Foundation/Foundation.h>
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 <objc/runtime.h>
#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 <objc/runtime.h>
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 <objc/runtime.h>
#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 的消息转发机制就暂时结束了。
最后，我自己写了一个思维导图，可以根据思维导图进行对比。
