說明 | 時間 |
---|---|
首次發布 | 2017年08月09日 |
最近更新 | 2019年05月04日 |
消息機制可以認為是objc_msgSend的執行流程,包括消息發送、動態方法解析和消息轉發。
一、消息發送
消息發送
二、動態方法解析
- 檢查是否曾經有動態解析
- 有:消息轉發
- 沒有:調用
+ (BOOL)resolveInstanceMethod:(SEL)sel
或+ (BOOL)resolveClassMethod:(SEL)sel
來動態解析,并將標記置為YES。
注意:在動態解析后,會重新回到 消息發送 的流程。
示例:
//MZPerson.h
#import <Foundation/Foundation.h>
@interface MZPerson : NSObject
- (void)sayHello;
- (void)sayWorld;
+ (void)unimplementationClassMethod;
@end
//MZPerson.m
#import "MZPerson.h"
#import <objc/runtime.h>
@implementation MZPerson
/**
動態添加方法之成員方法
*/
- (void)addByMethod {
NSLog(@"%@--%s--%s", self, sel_getName(_cmd), __func__);
}
void addByFunction(id self, SEL _cmd) {
NSLog(@"%@--%s--%s", self, sel_getName(_cmd), __func__);
}
+ (BOOL)resolveInstanceMethod:(SEL)sel {
//通過C函數實現
if (sel == @selector(sayHello)) {
//types:類型編碼, v表示void, @表示id或NSObject*, :表示SEL. 這些都可以用@encode()查看
class_addMethod(self, sel, (IMP)addByFunction, "v@:");
//return YES/NO都是無所謂的,只是方便底層打印而已.但最好還是按規范來,YES表示添加了方法到接受者
return YES;
}
//通過OC方法實現
if (sel == @selector(sayWorld)) {
Method method = class_getInstanceMethod(self, @selector(addByMethod));
class_addMethod(self, sel, method_getImplementation(method), method_getTypeEncoding(method));
return YES;
}
return [super resolveInstanceMethod:sel];
}
+ (BOOL)resolveClassMethod:(SEL)sel {
//通過C函數實現
if (sel == @selector(unimplementationClassMethod)) {
//需要特別注意的是,類方法存放在元類里,所以這里要把方法添加到object_getClass(self)
class_addMethod(object_getClass(self), sel, (IMP)addByFunction, "v@:");
return YES;
}
return [super resolveClassMethod:sel];
}
@end
三、消息轉發
消息轉發分為兩個階段:
第一階段:- (id)forwardingTargetForSelector:(SEL)aSelector
;
第二階段:- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
和- (void)forwardInvocation:(NSInvocation *)anInvocation
;
當階段一沒有實現,或者返回nil的時候,則進行階段二;如果消息轉發處理的是成員方法,那么三個階段都使用成員方法,否則就使用類方法。如代碼里所示。
消息轉發圖示
//消息轉發第一階段
//底層是通過`id cls = object_getClass(receiver);`拿到receiver
- (id)forwardingTargetForSelector:(SEL)aSelector {
if (aSelector == @selector(sayHello)) {
// //sayHello是對象方法,則返回的必須是一個對象,并且該對象中有成員方法sayHello
// //返回nil,則直接進入消息轉發的第二個階段
return [GoodGuy new];
}
return [super forwardingTargetForSelector:aSelector];
}
//消息轉發第二階段(該部分是針對成員方法)
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
if (aSelector == @selector(sayHello)) {
return [NSMethodSignature signatureWithObjCTypes:"v@:@q"];
}
return [super methodSignatureForSelector:aSelector];
}
- (void)forwardInvocation:(NSInvocation *)anInvocation {
//可以用anInvocation里的selector; 也可以自己直接指定一個方法
anInvocation.selector = @selector(offerHelp:age:);
NSString *name = @"小紅";
NSInteger age = 18;
//每一個方法會默認隱藏兩個參數:self、_cmd,所以自定義參數要從下標2開始
[anInvocation setArgument:&name atIndex:2];
[anInvocation setArgument:&age atIndex:3];
GoodGuy *goodguy = [GoodGuy new];
//做個判斷,保證指定的receiver會響應該selector
if ([goodguy respondsToSelector:anInvocation.selector]) {
[anInvocation invokeWithTarget:goodguy];
}
}
//以下是針對于類方法的消息轉發(區別只是將這兩個方法換成`+`即可)
//消息轉發第一階段
+ (id)forwardingTargetForSelector:(SEL)aSelector {
if (aSelector == @selector(unimplementationClassMethod)) {
//返回nil,則直接進入消息轉發的第二個階段
//消息轉發類方法,這里需要傳入一個類對象
return [GoodGuy class];
}
return [super forwardingTargetForSelector:aSelector];
}
+ (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
if (aSelector == @selector(unimplementationClassMethod)) {
//如果需要傳參,則ObjCTypes一定要寫寫清,不然在傳參時會報越界
//如果需要進行消息轉發給一個`無參無返回值`的方法,無論被轉發的selector是什么類型的,通通可以用@"v@:"來表示
return [NSMethodSignature signatureWithObjCTypes:"v@:@q"];
}
return [super methodSignatureForSelector:aSelector];
}
+ (void)forwardInvocation:(NSInvocation *)anInvocation {
//可以用anInvocation里的selector; 也可以自己直接指定一個方法
anInvocation.selector = @selector(offerHelp:age:);
NSString *name = @"小紅";
NSInteger age = 18;
//每一個方法會默認隱藏兩個參數:self、_cmd,所以自定義參數要從下標2開始
[anInvocation setArgument:&name atIndex:2];
[anInvocation setArgument:&age atIndex:3];
GoodGuy *goodguy = [GoodGuy new];
//做個判斷,保證指定的receiver會響應該selector
if ([goodguy respondsToSelector:anInvocation.selector]) {
[anInvocation invokeWithTarget:goodguy];
}
}
拓展
消息轉發還可以用于降低 unrecognized selector
的崩潰率。
示例:
#import <Foundation/Foundation.h>
@interface MZPerson : NSObject
- (void)printPerson:(NSString *)name age:(NSInteger)age;
@end
@implementation MZPerson
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
// 能響應的直接走原先的邏輯
if ([self respondsToSelector:aSelector]) {
return [super methodSignatureForSelector:aSelector];
}
// 找不到的方法
return [NSMethodSignature signatureWithObjCTypes:"v@:"];
}
// 找不到的方法,都會來到這里
- (void)forwardInvocation:(NSInvocation *)anInvocation {
NSLog(@"找不到%@方法", NSStringFromSelector(anInvocation.selector));
}
@end
int main(int argc, const char * argv[]) {
@autoreleasepool {
MZPerson *person = [[MZPerson alloc] init];
//MZPerson里并沒有實現`printPerson:age:`的方法
[person printPerson:@"LynnZhang" age:12];
}
return 0;
}