深入探索 iOS 底层：类扩展和关联对象

IOS

2023-11-08 13:56:55

引言

在 iOS 底层原理之旅中，我们继续探究类的秘密，继分类之后，让我们深入了解类扩展和关联对象。

一、类扩展

什么是类扩展

类扩展允许我们向现有类添加方法、属性和实例变量，而无需修改原始类。这为我们提供了灵活地扩展类功能，同时保持其原始代码完整性。

工作原理

类扩展通过创建一个新的分类，该分类的名称以扩展的类名开头，后跟“Extension”。分类声明在扩展中的方法和属性。当编译时，编译器将这些方法和属性添加到原始类的元数据中，使它们看起来像是直接定义在原始类中。

用途

类扩展对于以下情况非常有用：

为现有类添加新功能，而无需修改其原始代码。
组织与现有类相关的特定功能。
在不同模块或框架之间共享扩展。

二、关联对象

什么是关联对象

关联对象是 Objective-C 中的一种机制，允许我们向对象附加额外信息，这些信息不属于对象的属性或实例变量。它为存储自定义数据和实现特殊功能提供了一种灵活的方式。

工作原理

关联对象通过 Objective-C 运行时实现。它为每个对象维护一个关联对象字典。该字典存储关联对象及其对应的值。我们可以使用 objc_setAssociatedObject 和 objc_getAssociatedObject 函数来设置和检索关联对象。

用途

关联对象对于以下情况非常有用：

向对象添加自定义数据，而无需修改其原始类。
实现私有属性或方法，这些属性或方法不会显示在类的接口中。
在不同模块或框架之间共享数据。

三、示例

以下是一个类扩展和关联对象的示例：

// Class Extension
@interface Person (Extended)

- (void)greetWithName:(NSString *)name;

@end

@implementation Person (Extended)

- (void)greetWithName:(NSString *)name {
    NSLog(@"Hello, %@!", name);
}

@end

// Associated Object
@interface Person (AssociatedObject)

- (void)setObject:(id)object forKey:(NSString *)key;
- (id)objectForKey:(NSString *)key;

@end

@implementation Person (AssociatedObject)

- (void)setObject:(id)object forKey:(NSString *)key {
    objc_setAssociatedObject(self, key, object, OBJC_ASSOCIATION_RETAIN);
}

- (id)objectForKey:(NSString *)key {
    return objc_getAssociatedObject(self, key);
}

@end