MK
摩柯社区 - 一个极简的技术知识社区
AI 面试

Objective-C 类别（Category）的原理与运用

2024-10-105.4k 阅读

Objective-C 类别（Category）的原理

类别在 Objective-C 中的地位与作用

在 Objective-C 编程中，类别（Category）是一种强大的语言特性，它允许开发者在不继承类或修改类的原始代码的情况下，为现有的类添加新的方法。这一特性在很多场景下都非常有用，例如将一个大型类的方法拆分成多个文件，方便代码管理和维护；为系统类添加自定义方法，扩展系统类的功能等。

从设计模式的角度来看，类别有点类似装饰者模式，它在不改变类的继承结构的前提下，为对象添加新的职责。与继承不同，继承是一种 “is - a” 关系，子类继承父类的属性和方法并可能进行扩展或重写；而类别只是为类添加方法，不涉及属性的添加（虽然可以通过关联对象间接实现属性添加，这在后面会详细介绍）。

类别数据结构剖析

Objective-C 的类别在底层有其特定的数据结构来支撑其功能。在 Objective - C 的 runtime 源码中，类别是通过 category_t 结构体来表示的。以下是简化后的 category_t 结构体定义（实际源码中有更多成员和细节处理）：

struct category_t {
    const char *name;
    classref_t cls;
    struct method_list_t *instanceMethods;
    struct method_list_t *classMethods;
    struct protocol_list_t *protocols;
    struct property_list_t *instanceProperties;
    // Fields below this point are not always present on disk.
    struct property_list_t *_classProperties;

    method_list_t *methodsForMeta(bool isMeta) {
        if (isMeta) return classMethods;
        else return instanceMethods;
    }

    property_list_t *propertiesForMeta(bool isMeta, struct header_info *hi);
};
  • name：类别名称，是一个 C 字符串，用于标识这个类别。例如，如果我们为 NSString 类创建一个名为 MyCategory 的类别，那么这里的 name 就是 "MyCategory"。
  • cls：指向被扩展类的指针，明确这个类别是为哪个类添加方法。
  • instanceMethods：指向实例方法列表的指针，存储类别中定义的实例方法。
  • classMethods：指向类方法列表的指针，用于存储类别中定义的类方法。
  • protocols：指向协议列表的指针，如果类别声明遵循了某些协议，这里会列出相关协议。
  • instanceProperties：指向实例属性列表的指针，虽然类别不能直接添加实例变量，但可以声明属性。

当编译器处理类别代码时，会生成对应的 category_t 结构体实例，并在运行时将其注册到 runtime 系统中。runtime 系统会将类别中的方法整合到类的方法列表中，使得类能够调用这些新添加的方法。

类别方法的加载与整合机制

在程序启动阶段，runtime 会进行一系列初始化操作，其中就包括类别方法的加载。当一个类被加载到内存时，runtime 会先处理类本身的方法列表、属性列表等信息。然后，runtime 会查找与该类相关的所有类别，并将类别中的方法整合到类的方法列表中。

具体的整合过程如下：

  1. 实例方法整合：runtime 会将类别中的实例方法添加到类的实例方法列表的前面。这意味着如果类别中定义的方法与类本身的方法同名，类别中的方法会优先被调用。例如：
@interface MyClass : NSObject
- (void)originalMethod;
@end

@implementation MyClass
- (void)originalMethod {
    NSLog(@"Original method in MyClass");
}
@end

@interface MyClass (MyCategory)
- (void)originalMethod;
@end

@implementation MyClass (MyCategory)
- (void)originalMethod {
    NSLog(@"Method in MyCategory");
}
@end

int main(int argc, const char * argv[]) {
    @autoreleasepool {
        MyClass *obj = [[MyClass alloc] init];
        [obj originalMethod];
    }
    return 0;
}

在上述代码中，当调用 [obj originalMethod] 时，输出的是 "Method in MyCategory"，因为类别中的方法在实例方法列表的前面，优先被调用。

  1. 类方法整合：对于类别中的类方法，runtime 会将其添加到类的元类（meta - class）的方法列表前面。类方法实际上是定义在元类中的，所以整合过程与实例方法类似，只是针对的是元类的方法列表。

这种方法整合机制使得类别能够灵活地为类添加新的方法，并且在方法调用时能够按照预期的顺序找到合适的方法。然而，由于类别方法会覆盖同名的原类方法，这也需要开发者在使用类别时谨慎处理方法命名，避免出现意外的行为。

Objective - C 类别（Category）的运用

为系统类扩展功能

为系统类扩展功能是类别最常见的应用场景之一。例如，NSString 类是 iOS 开发中经常使用的字符串类，但有时系统提供的方法不能满足特定的业务需求。我们可以通过类别为 NSString 添加自定义方法。

假设我们需要一个方法来判断字符串是否是有效的邮箱地址，我们可以这样定义类别：

@interface NSString (EmailValidation)
- (BOOL)isValidEmail;
@end

@implementation NSString (EmailValidation)
- (BOOL)isValidEmail {
    NSString *emailRegex = @"[A-Z0-9a-z._%+-]+@[A-Za-z0-9.-]+\\.[A-Za-z]{2,6}";
    NSPredicate *emailTest = [NSPredicate predicateWithFormat:@"SELF MATCHES %@", emailRegex];
    return [emailTest evaluateWithObject:self];
}
@end

在其他代码中，我们就可以直接使用这个新方法：

NSString *email = @"test@example.com";
BOOL isValid = [email isValidEmail];
if (isValid) {
    NSLog(@"Valid email");
} else {
    NSLog(@"Invalid email");
}

通过这种方式，我们无需继承 NSString 类，就为其添加了一个实用的方法，方便在项目中进行邮箱地址验证。

代码模块化与组织

当一个类的代码量非常大时，将所有方法都写在一个 .m 文件中会导致代码难以维护和阅读。类别可以将类的方法拆分到多个文件中，提高代码的模块化程度。

例如，在一个游戏开发项目中，有一个 GameCharacter 类，包含了移动、攻击、防御、技能释放等多种功能。我们可以将不同功能的方法分别放在不同的类别中：

// GameCharacter+Movement.m
@interface GameCharacter (Movement)
- (void)moveForward;
- (void)moveBackward;
- (void)turnLeft;
- (void)turnRight;
@end

@implementation GameCharacter (Movement)
- (void)moveForward {
    // 实现向前移动的逻辑
    NSLog(@"Moving forward");
}

- (void)moveBackward {
    // 实现向后移动的逻辑
    NSLog(@"Moving backward");
}

- (void)turnLeft {
    // 实现向左转的逻辑
    NSLog(@"Turning left");
}

- (void)turnRight {
    // 实现向右转的逻辑
    NSLog(@"Turning right");
}
@end

// GameCharacter+Combat.m
@interface GameCharacter (Combat)
- (void)attack;
- (void)defend;
@end

@implementation GameCharacter (Combat)
- (void)attack {
    // 实现攻击逻辑
    NSLog(@"Attacking");
}

- (void)defend {
    // 实现防御逻辑
    NSLog(@"Defending");
}
@end

这样，GameCharacter 类的代码被清晰地划分成不同的模块，每个模块专注于一个特定的功能领域，使得代码的维护和理解更加容易。在使用 GameCharacter 类时，开发者可以根据需要导入相应的类别头文件，只关注与当前功能相关的方法。

实现类似多继承的效果

在 Objective - C 中，一个类只能有一个直接父类，不支持多继承。然而，通过类别可以在一定程度上模拟多继承的效果。

假设我们有两个独立的功能模块，一个是 Printable，提供打印相关的方法；另一个是 Serializable，提供序列化相关的方法。我们可以为需要这两个功能的类分别创建类别：

// PrintableCategory.h
@interface NSObject (Printable)
- (void)printInfo;
@end

// PrintableCategory.m
@implementation NSObject (Printable)
- (void)printInfo {
    NSLog(@"This is a printable object");
}
@end

// SerializableCategory.h
@interface NSObject (Serializable)
- (NSData *)serialize;
@end

// SerializableCategory.m
@implementation NSObject (Serializable)
- (NSData *)serialize {
    // 简单示例，实际序列化逻辑会更复杂
    NSString *string = NSStringFromClass([self class]);
    return [string dataUsingEncoding:NSUTF8StringEncoding];
}
@end

然后，对于某个类，比如 MyDataClass，我们可以让它使用这两个类别来获得相应的功能：

@interface MyDataClass : NSObject
// 类本身的属性和方法
@end

@implementation MyDataClass
// 类本身的实现
@end

// 使用类别功能
MyDataClass *data = [[MyDataClass alloc] init];
[data printInfo];
NSData *serializedData = [data serialize];

通过这种方式，MyDataClass 类虽然没有真正继承多个父类，但却拥有了多个不同功能模块的方法，类似于实现了多继承的效果。不过需要注意的是，这种方式与真正的多继承还是有区别的，例如类别不能添加实例变量，并且方法冲突的处理也与多继承有所不同。

类别在 iOS 框架中的应用实例

在 iOS 开发中，很多系统框架都使用了类别来扩展类的功能。例如，UIViewController 类在不同的系统版本中通过类别添加了新的方法。在 iOS 13 中，UIViewController 通过类别添加了 overrideUserInterfaceStyle 属性，用于设置视图控制器的用户界面风格。

@interface UIViewController (UIContentAppearance)
@property (nonatomic) UIUserInterfaceStyle overrideUserInterfaceStyle API_AVAILABLE(ios(13.0));
@end

开发者在自己的视图控制器子类中可以直接使用这个属性来控制界面风格，而无需关心底层是如何通过类别实现的。这体现了类别在系统框架中对现有类进行功能扩展的便利性和实用性，同时也为开发者提供了更丰富的开发接口。

类别与关联对象实现 “伪属性”

虽然类别不能直接添加实例变量，但可以通过关联对象（Associated Objects）来模拟添加属性。关联对象是 Objective - C runtime 提供的一种机制，它允许开发者为对象动态地添加额外的属性。

例如，我们为 UIButton 类创建一个类别，添加一个自定义的 identifier 属性：

#import <objc/runtime.h>

@interface UIButton (CustomIdentifier)
@property (nonatomic, copy) NSString *identifier;
@end

@implementation UIButton (CustomIdentifier)
- (void)setIdentifier:(NSString *)identifier {
    objc_setAssociatedObject(self, @selector(identifier), identifier, OBJC_ASSOCIATION_COPY_NONATOMIC);
}

- (NSString *)identifier {
    return objc_getAssociatedObject(self, @selector(identifier));
}
@end

在上述代码中，通过 objc_setAssociatedObject 函数来设置关联对象，通过 objc_getAssociatedObject 函数来获取关联对象。这样，我们就为 UIButton 类添加了一个类似属性的功能。在实际使用中：

UIButton *button = [UIButton buttonWithType:UIButtonTypeSystem];
button.identifier = @"myButton";
NSString *idValue = button.identifier;
NSLog(@"Button identifier: %@", idValue);

这种通过类别和关联对象实现 “伪属性” 的方式，在一些需要为系统类或现有类添加额外属性的场景中非常有用，但在使用时需要注意内存管理，因为关联对象的内存管理方式取决于设置时指定的策略（如 OBJC_ASSOCIATION_COPY_NONATOMIC、OBJC_ASSOCIATION_RETAIN_NONATOMIC 等）。

类别使用中的注意事项

  1. 方法命名冲突：由于类别方法会覆盖同名的原类方法，如果不小心在类别中定义了与原类或其他类别同名的方法，可能会导致不可预期的行为。因此，在命名类别方法时，应该尽量使用唯一的命名前缀，避免冲突。例如，为 NSString 类的类别方法命名为 myString_categoryMethod。
  2. 属性声明与实现：虽然类别可以声明属性，但不能直接为其合成实例变量和存取方法。如果需要实现属性功能，通常需要使用关联对象。同时，在使用关联对象实现属性时，要注意内存管理和性能问题。
  3. 类别加载顺序：类别方法的加载顺序可能会影响方法的调用优先级。一般来说，最后加载的类别中的方法会在方法列表的最前面，优先被调用。在复杂的项目中，要注意类别加载顺序对方法调用的影响，确保代码的正确性和稳定性。

通过深入理解 Objective - C 类别（Category）的原理和运用技巧，开发者可以更加灵活高效地进行编程，充分利用这一强大的语言特性来优化代码结构、扩展类的功能，从而提升项目的开发效率和质量。无论是为系统类添加自定义功能，还是对大型类进行模块化拆分，类别都能发挥重要的作用。但在使用过程中，务必注意各种潜在的问题，遵循良好的编程规范，以确保代码的健壮性和可维护性。