黑魔法！聊聊iOS中的Method Swizzling！

IOS

2023-12-08 15:28:14

缘起：需求的出现

就在前不久，我们的项目迎来了一个新的需求：我们需要在所有页面上添加统计功能，即每当用户进入某个页面，就对其进行一次统计。

最直接的解决方案是：在每个控制器中加入统计代码。然而，这种方法存在几个显而易见的问题：

代码重复：在每个控制器中复制粘贴统计代码，这不仅繁琐，而且容易出错。
维护困难：当统计需求发生变化时，我们需要逐个修改每个控制器中的代码，这将是一项巨大的工作量。

拨云见日：Method Swizzling的登场

为了解决上述问题，我们需要一种更加优雅、灵活、可维护的方法。这时，Method Swizzling闪亮登场了。

Method Swizzling是一种非常强大的技术，它允许我们动态地交换两个方法的实现。换句话说，我们可以让一个方法调用另一个方法的实现，而无需修改任何源代码。

揭秘：Method Swizzling的原理

要理解Method Swizzling的原理，我们需要首先了解Objective-C中的方法调用机制。

在Objective-C中，方法调用实际上是通过消息传递来实现的。当一个对象向另一个对象发送消息时，系统会根据消息选择器的值找到相应的方法并执行它。

Method Swizzling正是利用了这一机制。它通过修改方法选择器的值，将一个方法的消息指向另一个方法的实现，从而实现了方法调用的交换。

示例：统计功能的实现

为了更好地理解Method Swizzling的用法，我们通过一个具体示例来实现统计功能。

首先，我们需要创建一个名为"StatisticsManager"的类，该类负责统计功能的实现。

@interface StatisticsManager : NSObject

+ (void)trackPageView;

@end

然后，我们在该类中实现"trackPageView"方法，该方法负责统计页面访问量。

@implementation StatisticsManager

+ (void)trackPageView {
    //统计代码
}

@end

接下来，我们需要使用Method Swizzling将"viewWillAppear"方法的实现替换为"trackPageView"方法的实现。

Class cls = [UIViewController class];
SEL originalSelector = @selector(viewWillAppear:);
SEL swizzledSelector = @selector(swizzled_viewWillAppear:);
Method originalMethod = class_getInstanceMethod(cls, originalSelector);
Method swizzledMethod = class_getInstanceMethod(cls, swizzledSelector);
BOOL didAddMethod = class_addMethod(cls, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
if (didAddMethod) {
    class_replaceMethod(cls, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
} else {
    method_exchangeImplementations(originalMethod, swizzledMethod);
}