繼續Objective-C runtime的研究

最近公司專案在做使用者行為分析

於是App端在某些頁面切換,互動操作的時候需要給統計系統傳送一條訊息

在幾十個Controller 的專案裡,一個一個地加程式碼那完全是不可能的,維護起來也是吃力

但這裡需要處理的是 Controller, 可以有以下方式實現上述需求

1. 利用Objective-C 中的物件繼承

　　繼承 在面向物件開發中是非常常用的,像我們現在做的專案工程中都會有一個BaseViewController,

所有新建的ViewController都繼承BaseViewController,通過往BaseViewController中新增一些公共方法\屬性 可以被他們的子類所呼叫

這是統一我們工程中所有檢視控制器樣式的一個主要途徑

2.利用Category 和Runtime實行方法hook 

hook方案有一個好處,就是可以避免程式碼入侵,做到更加廣泛的通用性.通過swizzling我們可以將原method與自己加入的method相結合,

即不需要在原有工程中加入程式碼,又能做到全域性覆蓋

兩種方案對比:

　　通過繼承父類來實現 相對於hook來說 是較為準確的,因為需要被統計的頁面都是繼承於這個父類的控制器,而其他的如UINavigationController,系統自帶的UIAlertController等則不會誤入統計資料當中

　　上面提到 hook方案是通過hook UIViewController viewdidload/viewdidappear等方法,而這些方法實際上 每個Controller 都會呼叫,那麼就會出現不該出現的Controller 也出現在這裡(如上面說到的UINavigationController和UIAlertController).但hook方案一個比較好的特點是無程式碼入侵,在不修改專案程式碼的前提下完成工作.

考慮到 行為分析統計系統 有可能被公司其他專案中所應用,這裡採用hook方案.那麼當中必然會出現 不該統計的卻被統計 的情況,後面再作分析.

既然用到hook方案,又要用runtime 的swizzling

首先 新建一個UIViewController 的category

實現swizzling程式碼

+ (void)load{
    [super load];
    
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        // 假如要開啟controller的統計 ,則把下面這行程式碼開啟
 

        __gbh_tracer_swizzleMethod([self class], @selector(viewDidAppear:), @selector(__gbh_tracer_viewDidAppear:));
    });
}

嗯,看到這裡大家會發現 這裡呼叫的是一個C的方法,然而這個C方法是怎麼實現的呢?看下面

void __gbh_tracer_swizzleMethod(Class class, SEL originalSelector, SEL swizzledSelector){
    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);
    }
}

這是一個標準的swizzling寫法,當然了 github上面也有關於swizzling的開源庫,用起來也順手 這裡就不多說

看回第一塊程式碼,紅色的viewDidAppear是即將被我hook的方法,__gbh_tracer_viewDidAppear 則是我需要實現的方法

- (void)__gbh_tracer_viewDidAppear:(BOOL)animated{
    [self __gbh_tracer_viewDidAppear:animated];  //由於方法已經被交換,這裡呼叫的實際上是viewDidAppear:方法
    
　　 //設定不允許傳送資料的Controller
    NSArray *filter = @[@"UINavigationController",@"UITabBarController"];
    NSString *className = NSStringFromClass(self.class);
    if ([filter containsObject:className]) return ; //如果該Controller在不允許傳送log的列表裡,則不能繼續往下走
   
    if ([self.title isKindOfClass:[NSString class]] && self.title.length > 0){ //有標題的才符合我的要求
        // 這裡傳送log
    }

}

嗯,剛剛說到有部分Controller我是不發資料的,這裡有兩重判斷,一個是加入到黑名單,另一個是 判斷Controller的title屬性是否為空

以上判斷基本能滿足我這個行為分析統計系統的需求,若還需要什麼判斷還可以繼續加

以此 我只需要往工程裡面新增這個Category,這個viewDidAppear就會被hook出來,可以為所欲為..

另外 需求中還提到 需要在應用啟動的時候傳送一次init訊息

hook?可以,但我更傾向與利用category+NSNotification,因為系統中已經有 UIApplicationDidFinishLaunchingNotification

這種通知,直接用就可以

@implementation UIApplication (GBHTracer)
+ (void)load{
    [super load];
    
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{ //只執行一次就可以了
        [[NSNotificationCenter defaultCenter]addObserver:self selector:@selector(__gbh_tracer_applicationDidFinishLaunching:) name:UIApplicationDidFinishLaunchingNotification object:nil];
    });
}

+ (void)__gbh_tracer_applicationDidFinishLaunching:(NSNotification *)noti{
    //應用啟動時為所欲為!
}

@end

嗯..我們的行為分析統計系統就在原工程不Import一個頭檔案 不呼叫 任何一個方法就可以達到統計效果.

但是像什麼操作響應的時候的統計,還是需要各位看官在響應中呼叫相應的方法