揭秘 Associated Object 内部实现原理，赋能动态对象行为

2023-10-28 06:17:56

Associated Object 的原理

在 Objective-C 中，Associated Object 是一种将任意值与对象关联的机制，允许我们动态地为对象添加新的行为。它通过运行时 API 实现，为对象分配额外的内存空间来存储这些关联值，并且可以通过关联键来访问它们。

属性访问

Associated Object 最常见的用途之一是为对象添加新的属性。我们可以使用 objc_setAssociatedObject() 函数将一个值与对象关联，并使用 objc_getAssociatedObject() 函数来检索它。这种机制允许我们动态地为对象添加属性，而无需修改类的实现。

方法交换

Associated Object 的另一个重要用途是方法交换。我们可以使用 objc_setAssociatedObject() 函数将一个方法与对象关联，并使用 objc_msgSend() 函数来调用它。这种机制允许我们动态地替换对象的方法实现，而无需修改类的实现。

Associated Object 的示例代码

以下是一个示例代码，展示如何使用 Associated Object 为对象添加一个新的属性和方法：

#import <objc/runtime.h>

@interface Person : NSObject

@end

@implementation Person

- (void)sayHello {
  NSLog(@"Hello, world!");
}

@end

int main() {
  Person *person = [[Person alloc] init];

  // 为对象添加一个新的属性
  objc_setAssociatedObject(person, @"name", @"John Doe", OBJC_ASSOCIATION_RETAIN);

  // 从对象中获取属性值
  NSString *name = objc_getAssociatedObject(person, @"name");

  NSLog(@"Name: %@", name);

  // 为对象添加一个新的方法
  objc_setAssociatedObject(person, @"sayGoodbye", [self sayGoodbye], OBJC_ASSOCIATION_COPY);

  // 调用对象的新方法
  objc_msgSend(person, @selector(sayGoodbye));

  return 0;
}

- (void)sayGoodbye {
  NSLog(@"Goodbye, world!");
}

在这个示例中，我们使用 Associated Object 为 Person 对象添加了一个新的属性 "name" 和一个新的方法 "sayGoodbye"。我们使用 objc_setAssociatedObject() 函数将值与对象关联，并使用 objc_getAssociatedObject() 函数来检索值。我们还使用 objc_msgSend() 函数来调用对象的新方法。

Associated Object 的优点

Associated Object 具有以下优点：