OC 原理——Category 关联对象

IOS

2024-02-17 20:30:14

OC 原理——Category 关联对象

Objective-C (OC) 中的 Category 是一个强大的特性，它允许我们在不修改原始类的源代码的情况下向类添加方法和属性。这在扩展现有类以添加新功能时非常有用，特别是当我们无法访问原始源代码时。

OC 中的 Category 实际上是一种关联对象，它与原始类动态关联。当我们创建一个 Category 时，我们实际上是在向关联的对象添加方法和属性。这些方法和属性在运行时与原始类相关联，就好像它们是原始类本身的一部分一样。

在 Category 中声明属性

虽然 Category 允许我们添加方法，但我们不能直接在 Category 中声明属性。这是因为属性通常与实例变量相关联，而 Category 无法修改原始类的实例变量。

使用关联对象解决问题

为了解决这个问题，我们可以使用关联对象。关联对象是一种机制，它允许我们动态地向任何对象（包括 Category）添加额外的数据。

要向 Category 添加属性，我们可以使用以下步骤：

定义一个关联键，这是一个唯一的字符串，用于标识我们的属性。
使用 objc_setAssociatedObject 函数将属性值与关联键和 Category 相关联。
使用 objc_getAssociatedObject 函数检索关联的值。

代码示例

以下是一个代码示例，演示如何在 Category 中使用关联对象实现属性：

// 定义关联键
static const char *MyKey = "MyPropertyKey";

@interface MyClass (MyCategory)

@property (nonatomic, strong) NSString *myProperty;

@end

@implementation MyClass (MyCategory)

- (void)setMyProperty:(NSString *)myProperty {
    objc_setAssociatedObject(self, MyKey, myProperty, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (NSString *)myProperty {
    return objc_getAssociatedObject(self, MyKey);
}

@end