iOS - runtime-01 消息转发机制。

在学习 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

结果可想而知,程序崩了。


WeChatd3bf8d4598bdb790c6a64fcac873c6a3.png

那怎么解决这个问题呢?这个时候就需要使用 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。

WechatIMG20.jpeg

到这里,第一阶段:动态方法解析就完成了。
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],就是它自己本身。
然后把第一阶段的代码注释调,来看一下运行结果。


WeChat6e550872c0224e2d2d52886198ab9bc5.png

可以看到,第二阶段的我们也搞定了。

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 打印出来了。

WechatIMG51.png

4.最后还有一个问题,那如果通过上面的三个方法都没有找到,怎么才能保证程序不会崩溃呢?
这个时候我们可以实现这个方法doesNotRecognizeSelector

- (void)doesNotRecognizeSelector:(SEL)aSelector {
    NSLog(@"未知错误");
}

然后把上面的代码注释掉,来看一下运行结果。


WechatIMG53.png

可以看到,程序运行起来并没有任何问题,并且,还打印了相关的东西。到这里,关于 runtime 的消息转发机制就暂时结束了。
最后,我自己写了一个思维导图,可以根据思维导图进行对比。


WechatIMG25.png

你可能感兴趣的:(iOS - runtime-01 消息转发机制。)