Method Swizzling是改變一個selector的實際實現(xiàn)的技術(shù)。通過這一技術(shù)嚣镜,我們可以在運行時通過修改類的分發(fā)表中selector對應(yīng)的函數(shù)爬迟,來修改方法的實現(xiàn)。
例如祈惶,我們想跟蹤在程序中每一個view controller展示給用戶的次數(shù):當然雕旨,我們可以在每個view controller的viewDidAppear中添加跟蹤代碼扮匠;但是這太過麻煩,需要在每個view controller中寫重復(fù)的代碼凡涩。創(chuàng)建一個子類可能是一種實現(xiàn)方式棒搜,但需要同時創(chuàng)建UIViewController, UITableViewController, UINavigationController及其它UIKit中view controller的子類,這同樣會產(chǎn)生許多重復(fù)的代碼活箕。
這種情況下力麸,我們就可以使用Method Swizzling,如在代碼所示:
<pre>import <objc/runtime.h> @implementation UIViewController (Tracking) +(void)load { static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ Class class = [self class]; // When swizzling a class method, use the following: // Class class = object_getClass((id)self); SEL originalSelector = @selector(viewWillAppear:); SEL swizzledSelector = @selector(xxx_viewWillAppear:); Method originalMethod = class_getInstanceMethod(class, originalSelector); Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector); BOOL didAddMethod = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod)); if (didAddMethod) { class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod)); } else { method_exchangeImplementations(originalMethod, swizzledMethod); } }); }
</pre>
<pre>pragma mark - Method Swizzling** -(void)xxx_viewWillAppear:(BOOL)animated { [self xxx_viewWillAppear:animated]; NSLog(@"viewWillAppear: %@", self); } @end
</pre>
在這里育韩,我們通過method swizzling修改了UIViewController的@selector(viewWillAppear:)對應(yīng)的函數(shù)指針克蚂,使其實現(xiàn)指向了我們自定義的xxx_viewWillAppear的實現(xiàn)。這樣筋讨,當UIViewController及其子類的對象調(diào)用viewWillAppear時埃叭,都會打印一條日志信息。
上面的例子很好地展示了使用method swizzling來一個類中注入一些我們新的操作悉罕。當然赤屋,還有許多場景可以使用method swizzling,在此不多舉例壁袄。在此我們說說使用method swizzling需要注意的一些問題:
Swizzling應(yīng)該總是在+load中執(zhí)行
在Objective-C中类早,運行時會自動調(diào)用每個類的兩個方法。+load會在類初始加載時調(diào)用嗜逻,+initialize會在第一次調(diào)用類的類方法或?qū)嵗椒ㄖ氨徽{(diào)用涩僻。這兩個方法是可選的,且只有在實現(xiàn)了它們時才會被調(diào)用栈顷。由于method swizzling會影響到類的全局狀態(tài)逆日,因此要盡量避免在并發(fā)處理中出現(xiàn)競爭的情況。+load能保證在類的初始化過程中被加載萄凤,并保證這種改變應(yīng)用級別的行為的一致性屏富。相比之下,+initialize在其執(zhí)行時不提供這種保證–事實上蛙卤,如果在應(yīng)用中沒為給這個類發(fā)送消息,則它可能永遠不會被調(diào)用噩死。
Swizzling應(yīng)該總是在dispatch_once中執(zhí)行
與上面相同颤难,因為swizzling會改變?nèi)譅顟B(tài),所以我們需要在運行時采取一些預(yù)防措施已维。原子性就是這樣一種措施行嗤,它確保代碼只被執(zhí)行一次,不管有多少個線程垛耳。GCD的dispatch_once可以確保這種行為栅屏,我們應(yīng)該將其作為method swizzling的最佳實踐飘千。
選擇器、方法與實現(xiàn)
在Objective-C中栈雳,選擇器(selector)护奈、方法(method)和實現(xiàn)(implementation)是運行時中一個特殊點,雖然在一般情況下哥纫,這些術(shù)語更多的是用在消息發(fā)送的過程描述中霉旗。
以下是Objective-C Runtime Reference中的對這幾個術(shù)語一些描述:
Selector(typedef struct objc_selector *SEL):用于在運行時中表示一個方法的名稱。一個方法選擇器是一個C字符串蛀骇,它是在Objective-C運行時被注冊的厌秒。選擇器由編譯器生成,并且在類被加載時由運行時自動做映射操作擅憔。 <code>Method(typedef struct objc_method Method):</code>在類定義中表示方法的類型 Implementation(typedef id (IMP)(id, SEL, ...)):這是一個指針類型鸵闪,指向方法實現(xiàn)函數(shù)的開始位置。這個函數(shù)使用為當前CPU架構(gòu)實現(xiàn)的標準C調(diào)用規(guī)范暑诸。每一個參數(shù)是指向?qū)ο笞陨淼闹羔?self)蚌讼,第二個參數(shù)是方法選擇器。然后是方法的實際參數(shù)屠列。 理解這幾個術(shù)語之間的關(guān)系最好的方式是:一個類維護一個運行時可接收的消息分發(fā)表啦逆;分發(fā)表中的每個入口是一個方法(Method),其中key是一個特定名稱笛洛,即選擇器(SEL)夏志,其對應(yīng)一個實現(xiàn)(IMP),即指向底層C函數(shù)的指針苛让。
為了swizzle一個方法沟蔑,我們可以在分發(fā)表中將一個方法的現(xiàn)有的選擇器映射到不同的實現(xiàn),而將該選擇器對應(yīng)的原始實現(xiàn)關(guān)聯(lián)到一個新的選擇器中狱杰。
調(diào)用_cmd
我們回過頭來看看前面新的方法的實現(xiàn)代碼: ```
<pre> -(void)xxx_viewWillAppear:(BOOL)animated { [self xxx_viewWillAppear:animated]; NSLog(@"viewWillAppear: %@", NSStringFromClass([self class])); }
</pre>
咋看上去是會導(dǎo)致無限循環(huán)的瘦材。但令人驚奇的是,并沒有出現(xiàn)這種情況仿畸。在swizzling的過程中食棕,方法中的[self xxx_viewWillAppear:animated]已經(jīng)被重新指定到UIViewController類的-viewWillAppear:中。在這種情況下错沽,不會產(chǎn)生無限循環(huán)簿晓。不過如果我們調(diào)用的是[self viewWillAppear:animated],則會產(chǎn)生無限循環(huán)千埃,因為這個方法的實現(xiàn)在運行時已經(jīng)被重新指定為xxx_viewWillAppear:了憔儿。
注意事項
Swizzling通常被稱作是一種黑魔法,容易產(chǎn)生不可預(yù)知的行為和無法預(yù)見的后果放可。雖然它不是最安全的谒臼,但如果遵從以下幾點預(yù)防措施的話朝刊,還是比較安全的:
總是調(diào)用方法的原始實現(xiàn)(除非有更好的理由不這么做):API提供了一個輸入與輸出約定,但其內(nèi)部實現(xiàn)是一個黑盒蜈缤。Swizzle一個方法而不調(diào)用原始實現(xiàn)可能會打破私有狀態(tài)底層操作拾氓,從而影響到程序的其它部分。 避免沖突:給自定義的分類方法加前綴劫樟,從而使其與所依賴的代碼庫不會存在命名沖突痪枫。 明白是怎么回事:簡單地拷貝粘貼swizzle代碼而不理解它是如何工作的,不僅危險叠艳,而且會浪費學(xué)習(xí)Objective-C運行時的機會奶陈。閱讀Objective-C Runtime Reference和查看<objc/runtime.h>頭文件以了解事件是如何發(fā)生的。 小心操作:無論我們對Foundation, UIKit或其它內(nèi)建框架執(zhí)行Swizzle操作抱有多大信心附较,需要知道在下一版本中許多事可能會不一樣吃粒。