我們看到在很多Runtime相關(guān)文章中都有介紹IMP指針的概念激挪，那么IMP究竟有什么實(shí)際的作用呢？我們先從一個(gè)函數(shù)看起來(lái)留特。
Method Swizzling
如果對(duì)Runtime有一定了解的話研叫，一定聽(tīng)說(shuō)過(guò)或者用過(guò)這個(gè)函數(shù)：
void method_exchangeImplementations(Method m1, Method m2)
它通常叫做method swizzling，算是ObjC的"黑魔法"了，作用就是在程序運(yùn)行期間動(dòng)態(tài)的給兩個(gè)方法互換實(shí)現(xiàn)大诸，比如有這樣一種使用場(chǎng)景：
我們的程序中有許多個(gè)ViewController捅厂，我想在對(duì)項(xiàng)目改動(dòng)最小的情況下，在當(dāng)每個(gè)Controller執(zhí)行完ViewDidLoad以后就在控制臺(tái)把自己的名字打印出來(lái)资柔，方便我去做調(diào)試或者了解項(xiàng)目結(jié)構(gòu)焙贷。
有許多朋友會(huì)這樣說(shuō)，讓所有控制器都繼承一個(gè)BaseController不就可以了嗎贿堰？我在這里要解釋一下這樣做的缺點(diǎn)：假如你的項(xiàng)目里有許多Controller的話辙芍，你就需要把項(xiàng)目里凡是沒(méi)有繼承自BaseController的每個(gè)Controller都做一次修改了，而且隨意更改層級(jí)結(jié)構(gòu)會(huì)發(fā)生意想不到的錯(cuò)誤羹与。
其實(shí)我們的目的就是重寫(xiě)ViewDidLoad的方法故硅，并在他的方法最后加上幾句Log，所以我們需要給UIViewController建立一個(gè)category纵搁，因?yàn)槲覀冎莱孕疲绻贑atagory中重寫(xiě)一個(gè)方法，就會(huì)覆蓋它的原有方法實(shí)現(xiàn)腾誉，但是徘层，這樣做以后就沒(méi)有辦法調(diào)用系統(tǒng)原有的方法峻呕，因?yàn)樵谝粋€(gè)方法里調(diào)用自己的方法會(huì)是一個(gè)死循環(huán)。所以我們的解決辦法就是趣效，另外寫(xiě)一個(gè)方法來(lái)和viewDidLoad“交換”瘦癌，這樣外部調(diào)用viewDidLoad就會(huì)調(diào)到新建的這個(gè)方法中，同樣跷敬，我們調(diào)用新建的方法就會(huì)調(diào)用到系統(tǒng)的viewDidLoad中了讯私。
+ (void)load
{
//保證方法只執(zhí)行一次
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
// 獲取到這個(gè)類(lèi)的viewDidLoad方法，它的類(lèi)型是一個(gè)objc_method結(jié)構(gòu)體指針
Method viewDidLoad = class_getInstanceMethod(self, @selector(viewDidLoad));
// 獲取自己新創(chuàng)建的方法
Method viewDidLoaded = class_getInstanceMethod(self, @selector(viewDidLoaded));
// 交換兩個(gè)方法實(shí)現(xiàn)
method_exchangeImplementations(viewDidLoad, viewDidLoaded);
});
}
- (void)viewDidLoaded
{
// 調(diào)用自己原有的方法
[self viewDidLoad];
NSLog(@"%@ did load",self);
}
IMP指針
其實(shí)干花，還有一種更加簡(jiǎn)單的方法可以讓我們辦到相同的目的妄帘，運(yùn)用IMP指針，IMP就是Implementation的縮寫(xiě)池凄，顧名思義抡驼，它是指向一個(gè)方法實(shí)現(xiàn)的指針，每一個(gè)方法都有一個(gè)對(duì)應(yīng)的IMP肿仑，所以致盟，我們可以直接調(diào)用方法的IMP指針，來(lái)避免方法調(diào)用死循環(huán)的問(wèn)題尤慰。
調(diào)用一個(gè)IMP的方式和調(diào)用普通C函數(shù)相同馏锡，比如：
id returnObjc = someIMP(objc,SEL,params...);
不過(guò)如果你的項(xiàng)目沒(méi)有做其他配置的話這樣調(diào)用編譯器是不會(huì)通過(guò)的，我們來(lái)看一下先它的定義：
if !OBJC_OLD_DISPATCH_PROTOTYPES
typedef void (*IMP)(void /* id, SEL, ... */ );
else
typedef id (*IMP)(id, SEL, ...);
endif
在默認(rèn)情況下你的工程是打開(kāi)這個(gè)配置的：
這種情況下IMP被定義為無(wú)參數(shù)無(wú)返回值的函數(shù)伟端。所以你需要到工程里搜索到這個(gè)選項(xiàng)并把它關(guān)閉杯道。這樣的麻煩就是，每次使用责蝠，你都需要修改工程配置党巾，所以這里我再介紹另外一種辦法：重新定義一個(gè)和有參數(shù)的IMP指針相同的指針類(lèi)型，在獲取IMP時(shí)把它強(qiáng)轉(zhuǎn)為此類(lèi)型霜医。這樣運(yùn)用IMP指針后齿拂，就不需要額外的給ViewController寫(xiě)新的方法：
#import "UIViewController+ViewDidLoad.h"
typedef id (*_IMP)(id ,SEL, ...);
@implementation UIViewController (ViewDidLoad)
+ (void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
// 獲取原始方法
Method viewDidLoad = class_getInstanceMethod(self, @selector(viewDidLoad));
// 獲取方法實(shí)現(xiàn)
_IMP viewDidLoad_IMP = (_IMP)method_getImplementation(viewDidLoad);
//重新設(shè)置方法實(shí)現(xiàn)
method_setImplementation(viewDidLoad, imp_implementationWithBlock(^(id target,SEL action){
//獲取方法實(shí)現(xiàn)
viewDidLoad_IMP(target,@selector(viewDidLoad));
//新增方法
NSLog(@"%@ did load",target);
}));
});
}
@end
還有一個(gè)地方我們需要注意，如果這樣直接調(diào)用IMP的話就會(huì)發(fā)生經(jīng)典的EXC_BAD_ACCESS錯(cuò)誤肴敛，我們定義的IMP指針是一個(gè)有返回值的類(lèi)型署海，而其實(shí)我們獲取的viewDidLoad這個(gè)方法是沒(méi)有返回值的，所以我們需要新定義一個(gè)和IMP相同類(lèi)型的函數(shù)指針比如VIMP医男，把他的返回值定位Void砸狞，這樣如果你修改的方法有返回值就用IMP，沒(méi)有返回值就用VIMP镀梭。
#import "UIViewController+ViewDidLoad.h"
typedef id (*_IMP)(id ,SEL, ...);
typedef void (*_VIMP)(id ,SEL, ...);
@implementation UIViewController (ViewDidLoad)
+ (void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
// 獲取原始方法
Method viewDidLoad = class_getInstanceMethod(self, @selector(viewDidLoad));
// 獲取方法實(shí)現(xiàn)
_VIMP viewDidLoad_IMP = (_VIMP)method_getImplementation(viewDidLoad);
//重新設(shè)置方法實(shí)現(xiàn)
method_setImplementation(viewDidLoad, imp_implementationWithBlock(^(id target,SEL action){
//獲取方法實(shí)現(xiàn)
viewDidLoad_IMP(target,@selector(viewDidLoad));
//新增方法
NSLog(@"%@ did load",target);
}));
});
}
@end
值得注意的是趾代，如果你重寫(xiě)的方法有返回值，不要忘記在最后做return丰辣。
總結(jié)
實(shí)際上直接調(diào)用一個(gè)方法的IMP指針的效率是高于調(diào)用方法本身的撒强，所以禽捆，如果你有一個(gè)合適的時(shí)機(jī)獲取到方法的IMP的話，你可以試著調(diào)用它飘哨。
這是只是IMP使用的場(chǎng)景之一胚想，它還有許多作用，希望大家多多發(fā)現(xiàn)芽隆。
