揭秘iOS底层原理：类扩展与关联对象详解

2023-11-17 21:42:08

类扩展与关联对象是iOS开发中非常重要的两个概念。它们允许您在不修改原始类源代码的情况下向类添加新的功能或数据。这使得代码维护和重用变得更加容易。

类扩展

类扩展允许您向现有类添加新的属性、方法和实现。这与使用类别(类别)非常相似，但类扩展有一些关键区别。

类扩展是使用Objective-C的@interface语法定义的，而类别是使用@interface {}语法定义的。
类扩展只能在实现文件中使用，而类别可以在头文件或实现文件中使用。
类扩展可以访问父类的私有成员，而类别不能。

以下是一个使用类扩展的示例：

@interface Person ()

@property (nonatomic, strong) NSString *phoneNumber;

@end

@implementation Person

- (void)setPhoneNumber:(NSString *)phoneNumber {
  if (_phoneNumber != phoneNumber) {
    _phoneNumber = phoneNumber;
    [[NSNotificationCenter defaultCenter] postNotificationName:@"PhoneNumberDidChangeNotification" object:self];
  }
}

@end

在这个示例中，我们使用类扩展向Person类添加了一个名为phoneNumber的新属性。我们还添加了一个setPhoneNumber:方法来设置phoneNumber属性，并发布一个通知以通知其他对象phoneNumber已更改。

关联对象

关联对象允许您将任意数据与特定对象关联。这对于存储与对象相关但不属于对象本身的数据非常有用。例如，您可以将用户首选项与视图控制器关联，或者将图像缓存与图像视图关联。

要使用关联对象，您需要使用objc_setAssociatedObject()和objc_getAssociatedObject()函数。以下是一个使用关联对象的示例：

@interface Person ()

@property (nonatomic, strong) NSString *phoneNumber;

@end

@implementation Person

- (void)setPhoneNumber:(NSString *)phoneNumber {
  if (_phoneNumber != phoneNumber) {
    _phoneNumber = phoneNumber;
    [[NSNotificationCenter defaultCenter] postNotificationName:@"PhoneNumberDidChangeNotification" object:self];
  }
}

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

- (void)setPhoneNumber:(NSString *)phoneNumber {
  objc_setAssociatedObject(self, @selector(phoneNumber), phoneNumber, OBJC_ASSOCIATION_RETAIN);
}

@end

在这个示例中，我们使用关联对象来存储phoneNumber属性的值。我们使用objc_setAssociatedObject()函数将phoneNumber属性的值与self对象关联，并使用objc_getAssociatedObject()函数来获取phoneNumber属性的值。