iOS - 关联对象

详细总结 Objective-C 的关联对象功能：
1. 基本利用

1. // 1. 设置关联对象
2. objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy);
4. // 2. 获取关联对象
5. id objc_getAssociatedObject(id object, const void *key);
7. // 3. 移除关联对象
8. objc_removeAssociatedObjects(id object);

复制代码

2. 关联策略

1. typedef OBJC_ENUM(uintptr_t, objc_AssociationPolicy) {
2.     OBJC_ASSOCIATION_ASSIGN = 0,           // 弱引用
3.     OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, // 强引用，非原子
4.     OBJC_ASSOCIATION_COPY_NONATOMIC = 3,   // 复制，非原子
5.     OBJC_ASSOCIATION_RETAIN = 01401,       // 强引用，原子
6.     OBJC_ASSOCIATION_COPY = 01403          // 复制，原子
7. };

复制代码

3. 实现示例

3.1 分类中添加属性

1. // UIView+Extension.h
2. @interface UIView (Extension)
3. @property (nonatomic, strong) NSString *identifier;
4. @end
6. // UIView+Extension.m
7. @implementation UIView (Extension)
9. static void *IdentifierKey = &IdentifierKey;
11. - (void)setIdentifier:(NSString *)identifier {
12.     objc_setAssociatedObject(self,
13.                             IdentifierKey,
14.                             identifier,
15.                             OBJC_ASSOCIATION_COPY_NONATOMIC);
16. }
18. - (NSString *)identifier {
19.     return objc_getAssociatedObject(self, IdentifierKey);
20. }
22. @end

复制代码

3.2 利用宏简化

1. // 定义关联属性的宏
2. #define OBJC_ASSOCIATION_PROPERTY_STRONG(_getter_, _setter_, _association_) \
3. - (void)_setter_:(id)object { \
4.     objc_setAssociatedObject(self, @selector(_getter_), object, _association_); \
5. } \
6. - (id)_getter_ { \
7.     return objc_getAssociatedObject(self, @selector(_getter_)); \
8. }
10. // 使用宏
11. @implementation UIView (Extension)
12. OBJC_ASSOCIATION_PROPERTY_STRONG(identifier, setIdentifier, OBJC_ASSOCIATION_COPY_NONATOMIC)
13. @end

复制代码

4. 关键点

4.1 Key 的选择

1. // 1. 静态变量地址
2. static char kAssociatedObjectKey;
3. objc_setAssociatedObject(obj, &kAssociatedObjectKey, value, policy);
5. // 2. 选择器
6. objc_setAssociatedObject(obj, @selector(propertyName), value, policy);
8. // 3. 字符串常量
9. static NSString *const kAssociatedKey = @"kAssociatedKey";
10. objc_setAssociatedObject(obj, (__bridge void *)kAssociatedKey, value, policy);

复制代码

4.2 内存管理

1. // 强引用示例
2. objc_setAssociatedObject(self,
3.                         @selector(strongProperty),
4.                         object,
5.                         OBJC_ASSOCIATION_RETAIN_NONATOMIC);
7. // 弱引用示例
8. objc_setAssociatedObject(self,
9.                         @selector(weakProperty),
10.                         object,
11.                         OBJC_ASSOCIATION_ASSIGN);

复制代码

5. 线程安全

1. // 原子操作
2. objc_setAssociatedObject(self,
3.                         key,
4.                         value,
5.                         OBJC_ASSOCIATION_RETAIN); // 使用带原子性的关联策略
7. // 手动加锁
8. @synchronized(self) {
9.     objc_setAssociatedObject(self,
10.                             key,
11.                             value,
12.                             OBJC_ASSOCIATION_RETAIN_NONATOMIC);
13. }

复制代码

6. 最佳实践

6.1 属性封装

1. @interface NSObject (Associated)
3. @property (nonatomic, strong) id associatedObject;
4. @property (nonatomic, weak) id weakAssociatedObject;
5. @property (nonatomic, copy) NSString *copyAssociatedObject;
7. @end

复制代码

6.2 清理处置惩罚

1. - (void)dealloc {
2.     // 移除特定关联对象
3.     objc_setAssociatedObject(self, @selector(associatedObject), nil, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
4.    
5.     // 或移除所有关联对象
6.     objc_removeAssociatedObjects(self);
7. }

复制代码

7. 留意事项

1. // 1. 避免循环引用
2. @implementation MyClass
3. - (void)setupBlock {
4.     __weak typeof(self) weakSelf = self;
5.     self.associatedBlock = ^{
6.         __strong typeof(weakSelf) strongSelf = weakSelf;
7.         [strongSelf doSomething];
8.     };
9. }
11. // 2. 合理选择关联策略
12. - (void)setDelegate:(id<MyDelegate>)delegate {
13.     objc_setAssociatedObject(self,
14.                             @selector(delegate),
15.                             delegate,
16.                             OBJC_ASSOCIATION_ASSIGN); // 代理使用 ASSIGN
17. }

复制代码

关联对象的优点：

• 无需修改原类即可添加属性
  
• 支持各种属性特性（strong/weak/copy）
  
• 线程安全选项
  
• 运行时动态性
  

留意事项：

• 合理利用关联策略
  
• 留意内存管理
  
• 制止过度利用
  
• 思量性能影响
  
• 留意命名冲突
  

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。