Category: 深入探索其底层原理

2024-01-20 06:42:04

前言

Category 是 Objective-C 中一种强大的机制，用于扩展现有类而不修改原始源代码。它允许开发人员向现有类添加新方法、属性和实现，从而实现代码重用和灵活性。

一、Category 的基本使用

Category 使用语法如下：

@interface 类名 (分类名)
// 定义新的方法、属性或实现
@end

例如，要向 NSString 类添加一个新方法 reverseString，我们可以使用以下 Category：

@interface NSString (ReverseString)
- (NSString *)reverseString;
@end

二、Category 的底层结构

Category 实际上是类的一个扩展，它与原始类没有任何直接关联。在底层，Category 被编译为一个名为 <Category 名>.<原始类名> 的新类。

@interface NSArray (MyCategory) // MyCategory 是分类名
// 方法和属性
@end

编译后，该 Category 会生成一个名为 NSArray_MyCategory.h 的头文件和 NSArray_MyCategory.m 的实现文件。

三、Category 的加载处理流程

当一个程序执行时，Objective-C 运行时会负责加载 Category。加载过程如下：

当一个 Category 被引用时，运行时会检查它是否已经被加载。
如果还没有加载，运行时会从文件中加载 Category 类。
运行时会创建 Category 类并将其作为原始类的子类。
原始类的方法表和属性表会被更新，以包含 Category 中定义的新方法和属性。

四、Category 的 load 方法讲解

Category 中可以定义一个 load 方法，它在 Category 类加载时被调用。此方法用于执行初始化代码或注册新的选择器。

@interface NSString (ReverseString)
+ (void)load {
    // 注册新选择器
    SEL selector = @selector(reverseString);
    IMP implementation = imp_implementationWithBlock(^(id self) {
        // 实现了 reverseString 方法
    });
    class_addMethod(self, selector, implementation, "@@:");
}
@end