iOS架构师_消息转发机制

1.动态方法解析

对象在收到无法处理的消息时，会调用下面的方法，前者是调用类方法时会调用，后者是调用对象方法时会调用

//未实现类方法调用
+ (BOOL)resolveClassMethod:(SEL)sel
//未实现实例方法调用
+ (BOOL)resolveInstanceMethod:(SEL)sel

在该方法中，需要给对象所属类动态的添加一个方法，并返回YES，表明可以处理

+ (BOOL)resolveInstanceMethod:(SEL)sel
{
    NSString *method = NSStringFromSelector(sel);
    if ([@"playPiano" isEqualToString:method]) {
        /**
         添加方法

         @param self 调用该方法的对象
         @param sel 选择子
         @param IMP 新添加的方法，是c语言实现的
         @param 新添加的方法的类型，包含函数的返回值以及参数内容类型，eg：void xxx(NSString *name, int size)，类型为：v@i
         */
        class_addMethod(self, sel, (IMP)playPiano, "v");
        return YES;
    }
    return NO;
}

使用这种办法的前提是：相关方法代码已经实现，只是在运行时将改方法动态添加到目标类中。例如CoreData中使用的@dynamic属性，就是在运行时动态添加方法实现。

2.备援接收者

经历了第一步后，如果该消息还是无法处理，那么就会调用下面的方法，查询是否有其它对象能够处理该消息。

- (id)forwardingTargetForSelector:(SEL)aSelector

在这个方法里，我们需要返回一个能够处理该消息的对象

- (id)forwardingTargetForSelector:(SEL)aSelector
{
    NSString *seletorString = NSStringFromSelector(aSelector);
    if ([@"playPiano" isEqualToString:seletorString]) {
        Student *s = [[Student alloc] init];
        return s;
    }
    // 继续转发
    return [super forwardingTargetForSelector:aSelector];
}

3.完整的消息转发

经历了前两步，还是无法处理消息，那么就会做最后的尝试，先调用methodSignatureForSelector:获取方法签名，然后再调用forwardInvocation:进行处理，这一步的处理可以直接转发给其它对象，即和第二步的效果等效，但是很少有人这么干，因为消息处理越靠后，就表示处理消息的成本越大，性能的开销就越大。所以，在这种方式下，会改变消息内容，比如增加参数，改变选择子等等。

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
- (void)forwardInvocation:(NSInvocation *)anInvocation

下面是改变选择子的例子，比如我们直接调用的是playPiano方法，最后转发给了traval:方法。

// 完整的消息转发
- (void)travel:(NSString*)city
{
    NSLog(@"Teacher travel：%@", city);
}

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
{
    NSString *method = NSStringFromSelector(aSelector);
    if ([@"playPiano" isEqualToString:method]) {

        NSMethodSignature *signature = [NSMethodSignature signatureWithObjCTypes:"v@:@"];
        return signature;
    }
    return nil;
}

- (void)forwardInvocation:(NSInvocation *)anInvocation
{
    SEL sel = @selector(travel:);
    NSMethodSignature *signature = [NSMethodSignature signatureWithObjCTypes:"v@:@"];
    anInvocation = [NSInvocation invocationWithMethodSignature:signature];
    [anInvocation setTarget:self];
    [anInvocation setSelector:@selector(travel:)];
    NSString *city = @"北京";
    // 消息的第一个参数是self，第二个参数是选择子，所以"北京"是第三个参数
    [anInvocation setArgument:&city atIndex:2];

    if ([self respondsToSelector:sel]) {
        [anInvocation invokeWithTarget:self];
        return;
    } else {
        Student *s = [[Student alloc] init];
        if ([s respondsToSelector:sel]) {
            [anInvocation invokeWithTarget:s];
            return;
        }
    }

    // 从继承树中查找
    [super forwardInvocation:anInvocation];
}

+ (BOOL)resolveClassMethod:(SEL)sel OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);//类方法
 + (BOOL)resolveInstanceMethod:(SEL)sel OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);//对象方法

 - (id)forwardingTargetForSelector:(SEL)aSelector OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);//快速消息转发

 //标准消息的转发
 - (void)forwardInvocation:(NSInvocation *)anInvocation OBJC_SWIFT_UNAVAILABLE("");
 - (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector OBJC_SWIFT_UNAVAILABLE("");

消息转发机制代码示例：

新建Person类进行演示

Person.h

#import <Foundation/Foundation.h>

@interface Person : NSObject
- (void)run;
@end

Person.m

#import "Person.h"
#import <objc/message.h>

@implementation Person

// 1.动态方法
+ (BOOL)resolveInstanceMethod:(SEL)sel {
    NSLog(@"sel = %@", NSStringFromSelector(sel));

    // 1.判断没实现方法, 就动态添加方法
//    if (sel == @selector(run:)) {
//        // 动态添加方法
//        class_addMethod(self, sel, (IMP)newRun, "v@:@");
//        return YES;
//    }

    //2.动态添加的方法没有使用，就给super
    return [super resolveInstanceMethod:sel];
}

void newRun(id self,SEL sel, NSString *str) {
    NSLog(@"---run起来了没有 %@--",str);
}

// 二.消息转发重定向
- (id)forwardingTargetForSelector:(SEL)aSelector {
    NSLog(@"%@",NSStringFromSelector(aSelector));
    return [super forwardingTargetForSelector:aSelector];
//    return [[Animation alloc] init];
}


// 三.生成方法签名.
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
    // 1. 转化字符串
    NSString *sel = NSStringFromSelector(aSelector);

    // 2.进行判断 手动生成签名
    if ([sel isEqualToString:@"run"]) {
        return [NSMethodSignature signatureWithObjCTypes:"v@:"];
    } else {
    return [super methodSignatureForSelector:aSelector];
    }
}

// 四.拿到方法签名配发消息
- (void)forwardInvocation:(NSInvocation *)anInvocation {
    NSLog(@"-----%@",anInvocation);
    // 1.拿到这个消息
    SEL selector = [anInvocation selector];
    // 2.转发消息.
    Animation *anm = [[Animation alloc] init];

    if ([anm respondsToSelector:selector]) {
        // 调用这个对象,进行转发
        [anInvocation invokeWithTarget:anm];

    } else {
        [super forwardInvocation:anInvocation];
    }
}


// 五.抛出异常
- (void)doesNotRecognizeSelector:(SEL)aSelector {
    NSString *selStr = NSStringFromSelector(aSelector);
    NSLog(@"这个-----%@---方法不存在", selStr);
}

@end