接上篇iOS Runtime之消息傳遞

老規(guī)矩，先上經(jīng)典圖

msg.png

消息在繼承樹上沒有找到實現(xiàn)還有消息轉(zhuǎn)發(fā)提供三次機會，不至于直接報doesNotRecognizeSelector錯誤崩潰，當然這也給線上熱修復(fù)提供了支持。

@interface ViewController ()

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    [self performSelector:@selector(drink)];
}

繼承樹上沒有drink方法，則進入消息轉(zhuǎn)發(fā)階段

一、動態(tài)方法解析 resolveInstanceMethod

+ (BOOL)resolveInstanceMethod:(SEL)sel
{
    if ([NSStringFromSelector(sel) isEqualToString:@"drink"])
    {
        NSLog(@"drink 動態(tài)方法解析");
        class_addMethod([self class], sel, (IMP)drinkMethod, nil);
        return YES;
    }
    return [super resolveInstanceMethod:sel];
}

void drinkMethod()
{
    NSLog(@"動態(tài)方法添加 drink");
}

通過runtime動態(tài)添加drinkMethod方法來實現(xiàn)，這個類方法只要動態(tài)增加了方法，不管return YES或者NO都不會繼續(xù)往下走。

二、備用接受者
如果resolveInstanceMethod沒有動態(tài)添加方法就會繼續(xù)往下走forwardingTargetForSelector，將消息轉(zhuǎn)發(fā)給其他對象來實現(xiàn)。

- (id)forwardingTargetForSelector:(SEL)aSelector
{
    if ([NSStringFromSelector(aSelector) isEqualToString:@"drink"])
    {
        NSLog(@"備用接收者");
        return [Son new];
    }
    return [super forwardingTargetForSelector:aSelector];
}

如果Son類中有drink方法，則會執(zhí)行，如果沒有drink則會報doesNotRecognizeSelector崩潰。如果forwardingTargetForSelector方法返回nil或者沒有實現(xiàn)，則會進入最后一步完整消息轉(zhuǎn)發(fā)。

3、完整消息轉(zhuǎn)發(fā)
如果前兩步都未處理，則會進入最后的機會，先調(diào)用methodSignatureForSelector獲取方法簽名，每個字符都代表一種參數(shù)，具體對應(yīng)表格可參考官方文檔方法簽名,也可以直接使用類型編碼 @encode(type) 獲取表示該類型的字符串，而不必硬編碼。然后調(diào)用forwardInvocation處理，這一步的處理可以直接轉(zhuǎn)發(fā)給其它對象，即和第二步的效果一樣，沒有必要，故在這一步，會改變消息的內(nèi)容，比如增加參數(shù)，改變方法名稱。

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
{
    if ([NSStringFromSelector(aSelector) isEqualToString:@"drink"])
    {
        NSLog(@"簽名 進入forward");
        return [NSMethodSignature signatureWithObjCTypes:"v@:*"];
    }
    return [super methodSignatureForSelector:aSelector];
}

- (void)forwardInvocation:(NSInvocation *)anInvocation
{
 SEL sel = @selector(water:);
    NSString *str = [NSString stringWithFormat:@"%s%s%s%s",@encode(void),@encode(id),@encode(SEL),@encode(char *)];
    const char * cStr = [str cStringUsingEncoding:NSUTF8StringEncoding];//等同于"v@:*"
    NSMethodSignature *signature = [NSMethodSignature signatureWithObjCTypes:cStr];
    anInvocation = [NSInvocation invocationWithMethodSignature:signature];
    [anInvocation setTarget:self];
    [anInvocation setSelector:@selector(water:)];
    NSString *water = @"喝水";
    [anInvocation setArgument:&water atIndex:2];//第三個參數(shù)是字符串
    
    if ([self respondsToSelector:sel])
    {
        [anInvocation invokeWithTarget:self];
        return;
    }else
    {
        Son *s = [Son new];
        if ([s respondsToSelector:sel])
        {
            [anInvocation invokeWithTarget:s];
            return;
        }
    }
    [super forwardInvocation:anInvocation];
}

- (void)water:(NSString *)str
{
    NSLog(@"完整消息轉(zhuǎn)發(fā) %@",str);
}

通過第三步完整消息轉(zhuǎn)發(fā)我們發(fā)現(xiàn)drink方法已經(jīng)轉(zhuǎn)發(fā)到了water：方法里，其中"v@:"里,v代表返回值是void，@代表響應(yīng)者是一個ID對象 ，：代表選擇子，代表字符串，我們這里water：方法改變了方法名稱，增加了字符串參數(shù)。如果還是找不到對應(yīng)方法，則會沿著繼承樹查找，最終還是沒找到就會報doesNotRecognizeSelector錯誤程序掛掉。