OC runtime学习笔记之关联对象

前言

objective-c 不能动态的添加一些属性到对象上，和其他的一些原生支持这点的语言不一样。所以之前你都不得不努力为未来要增加的变量预留好空间。在 mac os x 10.6 中，objective-c 的 runtime 已经原生的支持这个功能了。

oc runtime 中关联对象的方法相信大家都使用过，但是其原理和内部实现就很多人不知道。这篇文章就来讲解一下 oc runtime 之关联对象，下面直接步入正题。

runtime 关联对象原理

runtime 中提供的 api 如下:

/// 对象关联策略
typedef objc_enum(uintptr_t, objc_associationpolicy) {
 objc_association_assign = 0,  /**< 关联一个弱引用 */
 objc_association_retain_nonatomic = 1, /**< 指定关联一个强引用. 非原子性. */
 objc_association_copy_nonatomic = 3, /**< 指定关联一个拷贝引用. 非原子性. */
 objc_association_retain = 01401, /**< 指定关联一个强引用.原子性. */
 objc_association_copy = 01403  /**< 指定关联一个拷贝引用.原子性. */
};

/** 
 * 用给定的 key 和关联策略 policy 为指定的 object 设置关联关系
 */
void objc_setassociatedobject(id object, const void * key, id value, objc_associationpolicy policy);

/** 
 * 通过给定的 object 和 key 拿到关联的对象
 */
id objc_getassociatedobject(id object, const void * key);

/** 
 * 移除给定 object 所有关联的关系
 */
void objc_removeassociatedobjects(id object)；

runtime 的 api 很简单，实际上工作中使用 runtime 的关联对象也很方便，如下：

static const void * _nonnull titlekey = @"titlekey";

- (void)settitle:(nsstring *)title
{
 objc_setassociatedobject(self, titlekey, key, objc_association_retain);
}

- (nsstring *)title
{
 return objc_getassociatedobject(self, titlekey);
}

如此简单强大的功能，如果想要观察系统的实现，可以找关联对象的源码，我们可以从objc 源码中拿到，

从源码中，可以找到以上三个 api 的实现如下：

以 objc_setassociatedobject 方法为例，方法内部调用的一个内部方法，参数同上，进入内部方法

/// 方法为我加过注释的代码。
void _object_set_associative_reference(id object, void *key, id value, uintptr_t policy) {
 
 // 取原来的绑定关系，新绑定关系原来关系为空
 objcassociation old_association(0, nil);
 // 验证参数
 id new_value = value ? acquirevalue(value, policy) : nil;
 {
 // 内部绑定关系管理器.内部管理一个全局 associationshashmap
 associationsmanager manager;
 // 管理器中的绑定关系map
 associationshashmap &associations(manager.associations());
 // 生成伪装地址。处理参数 object 地址
 disguised_ptr_t disguised_object = disguise(object);
 if (new_value) {
  // 打破原来存在的绑定关系
  associationshashmap::iterator i = associations.find(disguised_object); // 以伪装地址为key 在associationshashmap找objc 对应的二级 map
  if (i != associations.end()) {
  // 二级map存在。并取二级map： objectassociationmap
  objectassociationmap *refs = i->second;
  // 同样的方法找到 objectassociationmap 中绑定关系
  objectassociationmap::iterator j = refs->find(key); // 通过key 在objectassociationmap 中找objcassociation
  if (j != refs->end()) {
   // 找到原来 objcassociation 关系，直接赋值old_association
   old_association = j->second;
   j->second = objcassociation(policy, new_value);
  } else {
   // 不存在 -> 重新创建一次
   (*refs)[key] = objcassociation(policy, new_value);
  }
  } else {
  // 不存在二级map 就直接从新创建
  objectassociationmap *refs = new objectassociationmap;
  associations[disguised_object] = refs;
  (*refs)[key] = objcassociation(policy, new_value);
  object->sethasassociatedobjects();
  }
 } else {
  // 未传 value ，直接设置关联关系为nil。移除原来的绑定关系
  associationshashmap::iterator i = associations.find(disguised_object);
  if (i != associations.end()) {
  objectassociationmap *refs = i->second;
  objectassociationmap::iterator j = refs->find(key);
  if (j != refs->end()) {
   old_association = j->second;
   refs->erase(j);
  }
  }
 }
 }
 // 释放 old value.
 if (old_association.hasvalue()) releasevalue()(old_association);
}

从源码中可以看到维护对象关联的对象主要有四个

associationsmanager  // 维护关联关系的全局管理者 ---> 保存associationshashmap
associationshashmap  // 保存关联关系的hashmap ---> 保存对象对应的关联关系的map
objectassociationmap // 对象关联关系的map，二级map
objcassociation      // 关联关系

它们的关系用图像表示如下：

runtime

相当于从我们的对象外部重新创建并管理了一套对象关联机制，通过双层 hashmap 来维护对象之间的关联关系。其中 associationshashmap(外层map)是存放在 associationsmanager 中的全局静态 map。通过要关联对象的内存地址处理得到和对象唯一绑定的 key 在 map 中保存和对象相关的objectassociationmap，此 map 又维护一个一个的 objcassociation (关联策略和值)。

以上以 objc_setassociatedobject 为例讲解了关联对象的原理，objc_getassociatedobject和objc_removeassociatedobjects 原理也是一样。

思维拓展

runtime 中对象关联的原理如上，学习了苹果源码中的思想。可以发现工作中很多类似的数据存储恰好能借鉴一下。

恰好下午研究 runtime 对象关联原理的时候，朋友问我一个多用户系统的实现逻辑。

大意是：

• 用户通过手机号登录后会在内存中生成用户信息表，表中可以保存用户的诸如：个人基本信息、联系人信息、钱包信息……
• 用户退出登录后，信息不会丢失。
• 其他用户登录后，会同样创建对应的用户信息表，并保存
• 之前用户登录后，依然能查看之前自己的用户信息

看到这个问题，我直接感觉 runtime 中对象关联的实现恰好就是这个问题的极好实现。

通过用户登录id为 key 创建用户信息map。确定每个用户id对应唯一一个用户信息表，用户信息表中想保存什么信息就自定义啦。

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。