526互联

iOS：消息转发机制

发布时间 2023-04-21 13:48:26作者: dulinshun

iOS消息转发机制.png

测试类

@interface People ()

@property (nonatomic, strong) Car *car;

@end

@implementation People

- (instancetype)init
{
    self = [super init];
    if (self) {
        self.car = [[Car alloc] init];
    }
    return self;
}

void onfoot(id self, SEL _cmd) {
    NSLog(@" People 类的方法 onfoot");
}

/// 第一阶段：动态方法解析
/// 征求当前类是否允许这个方法。如果允许则添加到当前类中。然后返回YES
//+ (BOOL)resolveClassMethod:(SEL)sel {
+ (BOOL)resolveInstanceMethod:(SEL)sel {
    if ([NSStringFromSelector(sel) isEqualToString:@"onfoot"]) {
        NSLog(@" 第一阶段：动态方法解析：允许的方法，添加动态解析");
        class_addMethod([self class], sel, (IMP)onfoot, "v@:");
        return YES;
    }
    
    NSLog(@" 第一阶段：动态方法解析：不允许的方法");
    return [super resolveInstanceMethod:sel];;
}

/// 第二阶段：快速转发
/// 如果其他对象实现了该方法，则转化给其他对象
- (id)forwardingTargetForSelector:(SEL)aSelector {
    
    if ([self.car respondsToSelector:aSelector] && [NSStringFromSelector(aSelector) isEqualToString:@"travel"] ) {
        NSLog(@" 第二阶段：快速转发：找到调用者");
        return self.car;
    } else {
        NSLog(@" 第二阶段：快速转发：未找到调用者");
    }
    return [super forwardingTargetForSelector:aSelector];
}


/// 第三阶段：标准消息转发
/// 1. 获取函数的参数和返回值
///     a. 如果返回 nil，则执行 doesNotRecognizeSelector，然后 crash
///     b. 如果返回一个签名函数，则执行 forwardInvocation 给目标对象
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
    NSMethodSignature *signature = [super methodSignatureForSelector:aSelector];
    if (!signature) {
        signature = [self.car.class instanceMethodSignatureForSelector:aSelector];
        NSLog(@" 第三阶段：标准消息转发：第一步：获取方法签名");
    }
    return  signature;
}

- (void)forwardInvocation:(NSInvocation *)anInvocation {
    if ([self.car respondsToSelector:anInvocation.selector]) {
        NSLog(@" 第三阶段：标准消息转发：第二步：找到调用者");
        [anInvocation invokeWithTarget:self.car];
    } else {
        NSLog(@" 第三阶段：标准消息转发：第二步：未找到调用者");
        [self doesNotRecognizeSelector:anInvocation.selector];
    }
}

@end

@implementation Car

- (void)travel {
    NSLog(@" Car 类的方法 travel");
}

- (void)onwalk {
    NSLog(@" Car 类的方法 onwalk");
}

@end

调用

   NSLog(@"");
    [self.pp performSelector:@selector(onfoot)];
    NSLog(@"");
    [self.pp performSelector:@selector(travel)];
    NSLog(@"");
    [self.pp performSelector:@selector(onwalk)];
    NSLog(@"");

打印结果

2021-09-29 11:17:43.009181+0800 Demo[3347:88466] 
2021-09-29 11:17:43.009237+0800 Demo[3347:88466]  第一阶段：动态方法解析：允许的方法，添加动态解析
2021-09-29 11:17:43.009276+0800 Demo[3347:88466]  People 类的方法 onfoot
2021-09-29 11:17:43.009312+0800 Demo[3347:88466] 
2021-09-29 11:17:43.009347+0800 Demo[3347:88466]  第一阶段：动态方法解析：不允许的方法
2021-09-29 11:17:43.009379+0800 Demo[3347:88466]  第二阶段：快速转发：找到调用者
2021-09-29 11:17:43.009407+0800 Demo[3347:88466]  Car 类的方法 travel
2021-09-29 11:17:43.009451+0800 Demo[3347:88466] 
2021-09-29 11:17:43.009575+0800 Demo[3347:88466]  第一阶段：动态方法解析：不允许的方法
2021-09-29 11:17:43.010058+0800 Demo[3347:88466]  第二阶段：快速转发：未找到调用者
2021-09-29 11:17:43.010181+0800 Demo[3347:88466]  第一阶段：动态方法解析：不允许的方法
2021-09-29 11:17:43.010271+0800 Demo[3347:88466]  第三阶段：标准消息转发：第一步：获取方法签名
2021-09-29 11:17:43.010361+0800 Demo[3347:88466]  第一阶段：动态方法解析：不允许的方法
2021-09-29 11:17:43.010459+0800 Demo[3347:88466]  第三阶段：标准消息转发：第二步：找到调用者
2021-09-29 11:17:43.010539+0800 Demo[3347:88466]  Car 类的方法 onwalk
2021-09-29 11:17:43.010624+0800 Demo[3347:88466] 