26·iOS 面试题·什么是 Method Swizzling?

2018-12-05 本文已影响22人 pengxuyuan

前言

Objective-C 是一门动态语言，在运行时期才能真正具体确定调用哪个函数。运行时，方法的大概调用逻辑如下：

在运行时，类（Class）维护了一个消息分发列表来解决消息的正确发送。每一个消息列表的入口是一个方法（Method），这个方法映射了一对键值对，其中键值是这个方法的名字 selector（SEL），值是指向这个方法实现的函数指针 implementation（IMP）。

Method swizzling 修改了类的消息分发列表使得已经存在的 selector 映射了另一个实现 implementation，同时重命名了原生方法的实现为一个新的 selector。

摘自：Method Swizzling

简单来说 Method Swizzling 就是替换方法列表中的 selector 对应的 IMP，达到方法交互的目的。

Method Swizzling 的实现方案

1、使用 class_addMethod、class_replaceMethod、method_exchangeImplementations

可以使用系统提供的方法来操作对象的方法列表，具体实现代码如下：

    SEL originalSeletor = originalSelector;
    SEL swizzledSeletor = swizzledSelector;
    
    Method originalMethod = class_getInstanceMethod(originalClass, originalSeletor);
    Method swizzledMethod = class_getInstanceMethod(swizzledClass, swizzledSeletor);
    if (!isInstanceMethod) {
        originalMethod = class_getClassMethod(originalClass, originalSeletor);
        swizzledMethod = class_getClassMethod(swizzledClass, swizzledSeletor);
    }
    
        //先尝试給源SEL添加IMP，这里是为了避免源SEL没有实现IMP的情况
    BOOL didAddMethod = class_addMethod(originalClass, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
    if (didAddMethod) {
        // class_addMethod 添加成功则说明：originalClass 没有实现 originalSelector 对应的方法，并且将此时已经将 originalSelector 对应实现改成 swizzledMethod 的实现。
        //添加成功：说明源SEL没有实现IMP，将源SEL的IMP替换到交换SEL的IMP
        class_replaceMethod(swizzledClass, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
    } else {
        // class_addMethod 添加失败则说明：originalClass 已经实现 originalSelector 对应的方法，不能再加了，这里直接交换实现就可以了。
        //添加失败：说明源SEL已经有IMP，直接将两个SEL的IMP交换即可
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }