Method Swizzling

http://blog.leichunfeng.com/blog/2015/06/14/objective-c-method-swizzling-best-practice/

@interface UIViewController (MRCUMAnalytics)
@end
@implementation UIViewController (MRCUMAnalytics)
+ (void)load { 
   static dispatch_once_t onceToken; 
   dispatch_once(&onceToken, ^{
   Class class = [self class]; 
   SEL originalSelector = @selector(viewWillAppear:); 
   SEL swizzledSelector = @selector(mrc_viewWillAppear:); 
   Method originalMethod = class_getInstanceMethod(class, originalSelector); 
   Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector); 
   BOOL success = class_addMethod(class, originalSelector,
                 method_getImplementation(swizzledMethod),
                 method_getTypeEncoding(swizzledMethod)); 
   if (success) { 
             class_replaceMethod(class, swizzledSelector, 
             method_getImplementation(originalMethod), 
            method_getTypeEncoding(originalMethod)); 
          } else {
           method_exchangeImplementations(originalMethod, swizzledMethod);
         } 
   });
}

#pragma mark - Method Swizzling
- (void)mrc_viewWillAppear:(BOOL)animated 
{ 
   [self mrc_viewWillAppear:animated];
   [MobClick beginLogPageView:NSStringFromClass([self class])];
}
@end

我們使用 Method Swizzling 的目的通常都是為了給程序增加功能,而不是完全地替換某個功能,所以我們一般都需要在自定義的實現中調用原始的實現。所以這里就會有兩種情況需要我們分別進行處理:
第 1 種情況:主類本身有實現需要替換的方法,也就是 class_addMethod
方法返回 NO
。這種情況的處理比較簡單,直接交換兩個方法的實現就可以了:

1.png

第 2 種情況:主類本身沒有實現需要替換的方法,而是繼承了父類的實現,即 class_addMethod
方法返回 YES
。這時使用 class_getInstanceMethod
函數獲取到的 originalSelector
指向的就是父類的方法,我們再通過執行

 class_replaceMethod(class, swizzledSelector,
method_getImplementation(originalMethod), 
method_getTypeEncoding(originalMethod));

將父類的實現替換到我們自定義的 mrc_viewWillAppear
方法中。這樣就達到了在 mrc_viewWillAppear
方法的實現中調用父類實現的目的。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容