前言

​ 我们先了解 UIViewController 生命周期相关的内容和 iOS 的“黑魔法” Method Swizzling。然后再了解页面浏览事件（$AppViewScreen）全埋点的实现原理

一、UIViewController 生命周期

​ 众所周知，每一个 UIViewController 都管理着一个由多个视图组成的树形结构，其中根视图保存在 UIViewController 的 view 属性中。UIViewController 会懒加载它所管理的视图集，直到第一次访问 view 属性时，才会去加载或者创建 UIViewController 的视图集。

有以下几种常用的方式加载或者创建 UIViewController 的视图集：

• 使用 Storyboard
• 使用 Nib 文件
• 使用代码，即重写 - loadView

​ 以上这些方法，最终都会创建出合适的根视图并保存在 UIViewController 的 view 属性中，这是 UIViewController 生命周期的第一步。当 UIViewController 的根视图需要展示在页面上时，会调用 - viewDidLoad 方法。在这个方法中，我们可以做一些对象初始化相关的工作。

​ 需要注意的是：此时，视图的 bounds 还没有确定。对于使用代码创建视图，- viewDidLoad 方法会在 -loadView 方法调用结束之后运行；如果使用的是 Stroyboard 或者 Nib 文件创建视图，- viewDidLoad 方法则会在 - awakeFromNib 方法之后调用。

​ 当 UIViewController 的视图在屏幕上的显示状态发生变化时，UIViewController 会自动回调一些方法，确保子类能够响应到这些变化。如下图所示，它展示了 UIViewController 在不同的显示状态时会回调不同的方法。

​ 在 UIViewController 被销毁之前，还会回调 - dealloc 方法，我们一般通过重写这个方法来主动释放不能被 ARC 自动释放的资源。

​ 我们现在对 UIViewController 的整个生命周期有了一些基本了解。那么，我们如何去实现页面浏览事件（ $AppViewScreen 事件）的全埋点呢？

​ 通过 UIViewController 的生命周期可知，当执行到 - viewDidAppear: 方法时，表示视图已经在屏幕上渲染完成，也即页面已经显示出来了，正等待用户进行下一步操作。因此，- viewDidAppear: 方法就是我们触发页面浏览事件的最佳时机。如果想要实现页面浏览事件的全埋点，需要使用 iOS 的“黑魔法” Method Swizzling 相关的技术。

二、Method Swizzling 黑魔法

​ Method Swizzling，顾名思义，就是交换两个方法的实现。简单的来说，就是利用 Objective-C runtime 的动态绑定特性，把一个方法的实现与另一个方法的实现进行交换。

2.1 Method Swizzling 基础

​ 在 Objective-C 的 runtime 中，一个类是用一个名为 objc_class 的结构体表示的，它的定义如下：

struct objc_class {
Class _Nonnull isa OBJC_ISA_AVAILABILITY;

#if !__OBJC2__
Class _Nullable super_class OBJC2_UNAVAILABLE;
const char * _Nonnull name OBJC2_UNAVAILABLE;
long version OBJC2_UNAVAILABLE;
long info OBJC2_UNAVAILABLE;
long instance_size OBJC2_UNAVAILABLE;
struct objc_ivar_list * _Nullable ivars OBJC2_UNAVAILABLE;
struct objc_method_list * _Nullable * _Nullable methodLists OBJC2_UNAVAILABLE;
struct objc_cache * _Nonnull cache OBJC2_UNAVAILABLE;
struct objc_protocol_list * _Nullable protocols OBJC2_UNAVAILABLE;
#endif

} OBJC2_UNAVAILABLE;

​ 在上面的结构体中，虽然有很多字段在 OBJC2 中已经废弃了（OBJC2_UNAVAILABLE），但是了解这个结构体还是有助于我们理解 Method Swizzling 的底层原理。我们从上述结构体中可以发现，有一个 objc_method_list 指针，它保存着当前类的所有方法列表。同时，objc_method_list 也是一个结构体，它的定义如下：

struct objc_method_list {
struct objc_method_list * _Nullable obsolete OBJC2_UNAVAILABLE;

int method_count OBJC2_UNAVAILABLE;
#ifdef __LP64__
int space OBJC2_UNAVAILABLE;
#endif
/* variable length structure */
struct objc_method method_list[1] OBJC2_UNAVAILABLE;
}

​ 在上面的结构体中，有一个 objc_method 字段，我们再来看看 objc_method 这个结构体：

struct objc_method {
SEL _Nonnull method_name OBJC2_UNAVAILABLE;
char * _Nullable method_types OBJC2_UNAVAILABLE;
IMP _Nonnull method_imp OBJC2_UNAVAILABLE;
}

​ 从上面的结构体中可以看出，一个方法由下面三个部分组成：

• method_name：方法名
• method_types：方法类型
• method_imp：方法实现

使用 Method Swizzling 交换方法，其实就是修改了 objc_method 结构体中的 method_imp，也即改变了 method_name 和 method_imp 的映射关系，如下图所示。

那我们如何改变 method_name 和 method_imp 的映射关系呢？在 Objective-C 的 runtime 中，提供了很多非常方便使用的函数，让我们可以很简单的就能实现 Method Swizzling，即改变 method_name 和 method_imp 的映射关系，从而达到交换方法的效果。

2.2 实现 Method Swizzling 的相关函数

1. Method class_getInstanceMethod

   // 返回目标类 aClass、方法名为 aSelector 的实例方法
   // aClass :目标类
   // aSelector: 方法名
   OBJC_EXPORT Method _Nullable
   class_getInstanceMethod(Class _Nullable cls, SEL _Nonnull name)
       OBJC_AVAILABLE(10.0, 2.0, 9.0, 1.0, 2.0);
   

2. BOOL class_addMethod

   // 给目标类 aClass 添加一个新的方法，同时包括方法的实现
   // aClass: 目标类
   // aSelector: 要添加方法的方法名
   // imp: 要添加方法的方法实现
   // types: 方法实现的编码类型
   OBJC_EXPORT BOOL
   class_addMethod(Class _Nullable cls, SEL _Nonnull name, IMP _Nonnull imp, 
                   const char * _Nullable types) 
       OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);
   

3. IMP method_getImplementation

   // 返回方法实现的指针
   // 目标方法
   OBJC_EXPORT IMP _Nonnull
   method_getImplementation(Method _Nonnull m) 
       OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);
   

4. IMP class_replaceMethod

   // 替换目标类 aClass 的 aSelector 方法指针
   // aClass: 目标类
   // aSelector: 目前方法的方法名
   // imp：新方法的方法实现
   // types: 方法实现的编码类型
   OBJC_EXPORT IMP _Nullable
   class_replaceMethod(Class _Nullable cls, SEL _Nonnull name, IMP _Nonnull imp, 
                       const char * _Nullable types) 
       OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);
   

5. void method_exchangeImplementations

   // 交换2个方法的实现指针
   // m1: 交换方法1
   // m2: 交换方法2
   OBJC_EXPORT void
   method_exchangeImplementations(Method _Nonnull m1, Method _Nonnull m2) 
       OBJC_AVAILABLE(10.5, 2.0, 9.0, 1.0, 2.0);
   

2.3 实现 Method Swizzling

第一步 创建 NSObject 的分类 NSObject+SASwizzler

第二步 在 NSObject+SASwizzler.h 声明方法交换方法

/// 交换方法名为 originalSEL 和方法名为 alternateSEL 两个方法实现
/// @param originalSEL 原始的方法名称
/// @param alternateSEL 要交换的方法名称
+ (BOOL)sensorsdata_swizzleMethod:(SEL)originalSEL withMethod:(SEL)alternateSEL;

第三步 在 NSObject+SASwizzler.m 实现方法的交换

+ (BOOL)sensorsdata_swizzleMethod:(SEL)originalSEL withMethod:(SEL)alternateSEL {
   
    // 获取原始方法
    Method originalMethod = class_getInstanceMethod(self, originalSEL);
    // 当原始的方法不存在时，返回NO，表示 Swizzler 失败
    if (!originalMethod) {
        return NO;
    }
    
    // 获取要交换的方法
    Method alternateMethod = class_getInstanceMethod(self, alternateSEL);
    // 当交换的方法不存在时，返回NO，表示 Swizzler 失败
    if (!alternateMethod) {
        return NO;
    }
    
    // 交换两个方法的实现
    method_exchangeImplementations(originalMethod, alternateMethod);
    return YES;
}

三、实现界面预览事件全埋点

​ 利用方法交换，来交换 UIViewController 的 -viewDidAppear: 方法，然后在方法交换中触发 $AppViewScreen 事件，来实现界面预览的全埋点。

3.1 实现步骤

第一步：在 SensorsSDK 项目中，新增一个 UIViewController 类别 UIViewController+SensorsData

第二步：在 UIViewController+SensorsData.m 类别新增交换方法 - sensorsdata_viewDidAppear：，然后再交换方法中调用原始方法，并触发 $AppViewScreen 事件

- (void)sensorsdata_viewDidAppear:(BOOL)animated {
    // 调用原始方法, 即 - viewDidAppear
    [self sensorsdata_viewDidAppear:animated];
    
    // 触发 $AppViewScreen 事件
    NSMutableDictionary *properties = [NSMutableDictionary dictionary];
    [properties setValue:NSStringFromClass([self class]) forKey:@"$screen_name"];
    [properties setValue:self.navigationItem.title forKey:@"$title"];
    [[SensorsAnalyticsSDK sharedInstance] track:@"$AppViewScreen" properties:properties];
}

第三步: 在 UIViewController+SensorsData.m 中重写 + load 类方法，并在 + load 类方法中调用 NSObject+SASwizzler 的类方法交换

+ (void)load {
    [UIViewController sensorsdata_swizzleMethod:@selector(viewDidAppear:) withMethod:@selector(sensorsdata_viewDidAppear:)];
}

第四步 : 测试验证

{
  "event" : "$AppViewScreen",
  "time" : 1648626597682,
  "propeerties" : {
    "$model" : "x86_64",
    "$manufacturer" : "Apple",
    "$lib_version" : "1.0.0",
    "$os" : "iOS",
    "$app_version" : "1.0",
    "$screen_name" : "ViewController",
    "$os_version" : "15.2",
    "$lib" : "iOS"
  }
}

3.2 优化

问题：在应用程序启动过程中，会触发多余的 $AppViewScreen ，我们可以引入黑名单的机制，即在黑名单里配置那些 UIViewController 及子类不触发 $AppViewScreen 事件。

第一步 创建一个 sensorsdata_black_list.plist 文件，并把 root 类型改成 Array，该文件就是黑名单文件，然后在黑名单文件中添加控制器，如图所示：

第二步 在 UIViewController+SensorsData.m 文件中新增 - shouldTrackAppViewScreen 方法，用来判断当前控制器是否在黑名单中。

static NSString * const kSensorsDataBlackListFileName = @"sensorsdata_black_list";

// 黑名单
- (BOOL)shouldTrackAppViewScreen {
    static NSSet *blackList = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        NSString *path = [[NSBundle bundleForClass:SensorsAnalyticsSDK.class] pathForResource:kSensorsDataBlackListFileName ofType:@"plist"];
        NSArray *classNames = [NSArray arrayWithContentsOfFile:path];
        NSMutableSet *set = [NSMutableSet setWithCapacity:classNames.count];
        for (NSString *className in classNames) {
            [set addObject:NSClassFromString(className)];
        }
        blackList = [set copy];
    });
    for (Class cla in blackList) {
        if ([self isKindOfClass:cla]) {
            return  NO;
        }
    }
    return YES;
}

第三步 在触发 $AppViewScreen 事件之前，判断是否在黑名单中

- (void)sensorsdata_viewDidAppear:(BOOL)animated {
    // 调用原始方法, 即 - viewDidAppear
    [self sensorsdata_viewDidAppear:animated];
    
    // 触发 $AppViewScreen 事件
    if ([self shouldTrackAppViewScreen]) {
        NSMutableDictionary *properties = [NSMutableDictionary dictionary];
        [properties setValue:NSStringFromClass([self class]) forKey:@"$screen_name"];
        [properties setValue:self.navigationItem.title forKey:@"$title"];
        [[SensorsAnalyticsSDK sharedInstance] track:@"$AppViewScreen" properties:properties];
    }
}

第四步 测试验证

​ 运行Demo，所添加到黑名单中的 controller 不会发送 $AppViewScreen 事件。

3.4 遗留问题

​ 按照目前的方案实现 $AppViewScreen 事件的全埋点，会有2个问题：

应用程序热启动是（从后台恢复），第一个界面没有触发 $AppViewScreen 事件。原因是这个界面没有再次执行 - viewDidAppear: 方法

要求 UIViewController 的子类不重写 -viewDidAppear：方法，一旦重写，必须调用[super viewDidAppear:animated], 否则不会触发 $AppViewScreen 事件。原因是直接交换了 UIViewController 的 - viewDidAppear: 方法。