追根问底:Objective-C关联属性原理分析
一 引子
Objective-C是一种动态性很强的语言,所谓动态能力,也可以理解为运行时能力。对于Objective-C开发者来说,动态性所带来的编程便利无处不在。例如通过Category类别来扩展已有类的功能。可以使已有类拥有新的方法和属性。但是,如果你有使用Category来扩展类的属性,你一定了解并非简单的使用@property进行声明即可。例如下面的代码:#import <Foundation/Foundation.h>
@interface MyObject : NSObject
@end
@implementation MyObject
@end
@interface MyObject (Property)
@property (nonatomic, copy) NSString *addProperty;
@end
@implementation MyObject (Property)
@end
int main(int argc, const char * argv[]) {
@autoreleasepool {
MyObject *object = [[MyObject alloc] init];
object.addProperty = @"HelloWorld";
NSLog(@"%@", object.addProperty);
}
return 0;
} 代码在编译时不会有任何问题,但是如果运行,就会出现未定义的方法异常。因此如果要扩展类的属性,我们通常会这样实现:#import <objc/runtime.h>
@implementation MyObject (Property)
static NSString *kAddPropertyKey = @"kAddPropertyKey";
- (void)setAddProperty:(NSString *)addProperty {
objc_setAssociatedObject(self, kAddPropertyKey, addProperty, OBJC_ASSOCIATION_COPY);
}
- (NSString *)addProperty {
return objc_getAssociatedObject(self, kAddPropertyKey);
}
@end 再次运行,就可以正常的对addProperty属性进行存取值了。这里其实就使用到了Objective-C运行时的特性,在Objective-C中,类对象在创建时其所占用的内存空间就已经确定,那么你有没有想过,通过objc_setAssociatedObject这个运行时方法所存储的属性值是如何与当前对象关联起来的,这些数据又是存在哪里的?幸运的时,从objc源码可以清楚的了解关联属性的实现逻辑,这也是我们本篇文章要讨论的重点,了解这里的原理可能不能对你使用关联属性提供多大的帮助,但是这种设计思路定会使你在日常开发中受益匪浅。
二 objc_setAssociatedObject方法的核心原理
通过objc的runtime源码,我们可以看到objc_setAssociatedObject的方法实现如下:void
objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
{
_object_set_associative_reference(object, key, value, policy);
} 这一步无需过多解释,只是调用了一个内部函数,_object_set_associative_reference内部函数是关联属性实现的核心,此函数解析如下:void
_object_set_associative_reference(id object, const void *key, id value, uintptr_t policy)
{
// 1.安全性检查,要关联的对象和要关联的值不能为空
if (!object && !value) return;
// 2.检查当前对象是否不允许关联属性,某些类不允许其实例有关联属性
if (object->getIsa()->forbidsAssociatedObjects())
_objc_fatal("objc_setAssociatedObject called on instance (%p) of class %s which does not allow associated objects", object, object_getClassName(object));
// 3.创建一个包装对象指针的结构对象,存储要关联的对象指针
DisguisedPtr<objc_object> disguised{(objc_object *)object};
// 4.创建一个包装关联策略和被关联值的对象
ObjcAssociation association{policy, value};
// 5.根据关联策略来对值进行引用(retain或copy)
association.acquireValue();
bool isFirstAssociation = false;
{
// 6.获取关联管理器及其中的关联表对象
AssociationsManager manager;
AssociationsHashMap &associations(manager.get());
// 7.值如果存在,则进行关联
if (value) {
// 8.尝试向表中插入当前要关联的对象和值,如果已经存在,则什么都不做
auto refs_result = associations.try_emplace(disguised, ObjectAssociationMap{});
// 9. 检查try_emplace方法的插入结果,如果做了插入操作,则标记首次关联为true
if (refs_result.second) {
isFirstAssociation = true;
}
// 10.将表中存储的值与key进行关联
auto &refs = refs_result.first->second;
auto result = refs.try_emplace(key, std::move(association));
// 11.如果key已经存在,则进行关联策略和关联值的交换
if (!result.second) {
association.swap(result.first->second);
}
// 12. 要关联的值为nil,则为清除操作
} else {
// 13. 查找到关联到此对象的属性对
auto refs_it = associations.find(disguised);
if (refs_it != associations.end()) {
// 14.有关联属性,获取存储key的表
auto &refs = refs_it->second;
// 15.查找对应key是否存在
auto it = refs.find(key);
if (it != refs.end()) {
// 16.存在则进行关联数据的替换,包括关联策略和值,此时实际上是将值清空了
association.swap(it->second);
// 17.相关擦除操作
refs.erase(it);
if (refs.size() == 0) {
associations.erase(refs_it);
}
}
}
}
}
// 18.判断是否为此类实例对象的第一次关联,如果是,则修改标记位,标明已经有关联属性
if (isFirstAssociation)
object->setHasAssociatedObjects();
// 19.将旧的值进行release,如果需要的话
association.releaseHeldValue();
} 可以看到,整个关联属性的过程非常清晰,对于新值是否需要retain以及旧值是否需要release,是由关联策略决定的:enum {
OBJC_ASSOCIATION_SETTER_ASSIGN = 0, // assgin属性
OBJC_ASSOCIATION_SETTER_RETAIN = 1, // 设置值的时候需要retain
OBJC_ASSOCIATION_SETTER_COPY = 3, // 设置值的时候需要copy
OBJC_ASSOCIATION_GETTER_READ = (0 << 8), // readonly的属性
OBJC_ASSOCIATION_GETTER_RETAIN = (1 << 8), // 获取值的时候需要retain
OBJC_ASSOCIATION_GETTER_AUTORELEASE = (2 << 8), // 获取值的时候需要autorelease
OBJC_ASSOCIATION_SYSTEM_OBJECT = _OBJC_ASSOCIATION_SYSTEM_OBJECT, // 1 << 16
}; acquireValue方法实现如下,其只是判断是否需要retain和copy,之后调用对应的函数:inline void acquireValue() {
if (_value) {
switch (_policy & 0xFF) {
case OBJC_ASSOCIATION_SETTER_RETAIN:
_value = objc_retain(_value);
break;
case OBJC_ASSOCIATION_SETTER_COPY:
_value = ((id(*)(id, SEL))objc_msgSend)(_value, @selector(copy));
break;
}
}
} 在上面第8步中,有调用try_emplace方法来将数据插入到表结构中,此函数插入时会判断要插入的数据是否存在,其返回值会告知调用者是否产生了插入操作,如果已经存在,则此函数会什么都不做。
三 获取和移除关联属性的原理
现在,我们已经基本清楚了关联属性是如何设置和存储的,再来理解如果获取和移除就非常容易了。
获取关联属性的值是使用objc_getAssociatedObject运行时方法实现的,此方法实现如下:id
objc_getAssociatedObject(id object, const void *key)
{
return _object_get_associative_reference(object, key);
} 我们还是主要来解析下其调用的_object_get_associative_reference内部方法:id
_object_get_associative_reference(id object, const void *key)
{
// 1.创建关联对象结构
ObjcAssociation association{};
{
// 2.获取关联管理器及全局的Hash表
AssociationsManager manager;
AssociationsHashMap &associations(manager.get());
// 3.尝试查找当前传入对象的关联属性
AssociationsHashMap::iterator i = associations.find((objc_object *)object);
if (i != associations.end()) {
// 4.如果当前对象有关联属性,尝试查找存储key的列表中是否存在传入的key
ObjectAssociationMap &refs = i->second;
ObjectAssociationMap::iterator j = refs.find(key);
if (j != refs.end()) {
// 5.如果可以查到,对association进行赋值
association = j->second;
// 6.根据关联策略来决定是否对返回的值进行retain
association.retainReturnedValue();
}
}
}
// 7.根据返回策略来决定是否需要autorelease,如果没有查到,会返回nil值
return association.autoreleaseReturnedValue();
} 对于已经关联了属性的对象,我们也可以调用objc_removeAssociatedObjects方法来将关联的所有属性进行移除,此方法实现如下:void objc_removeAssociatedObjects(id object)
{
// 对象存在,并且已经标记过关联属性
if (object && object->hasAssociatedObjects()) {
_object_remove_assocations(object, /*deallocating*/false);
}
} _object_remove_assocation内部函数的实现也不复杂,解析如下:void
_object_remove_assocations(id object, bool deallocating)
{
// 1.创建关联对象
ObjectAssociationMap refs{};
{
// 2.获取关联管理器及Hash表
AssociationsManager manager;
AssociationsHashMap &associations(manager.get());
// 3.查找传入对象的关联属性
AssociationsHashMap::iterator i = associations.find((objc_object *)object);
if (i != associations.end()) {
// 4.查找到后,用空值进行交换
refs.swap(i->second);
// 5.如果是系统对象,则需要保留关联
bool didReInsert = false;
if (!deallocating) {
for (auto &ref: refs) {
if (ref.second.policy() & OBJC_ASSOCIATION_SYSTEM_OBJECT) {
i->second.insert(ref);
didReInsert = true;
}
}
}
// 6.无需保留关联,则直接清除数据
if (!didReInsert)
associations.erase(i);
}
}
// 7.对旧值进行release操作
SmallVector<ObjcAssociation *, 4> laterRefs;
for (auto &i: refs) {
if (i.second.policy() & OBJC_ASSOCIATION_SYSTEM_OBJECT) {
// If we are not deallocating, then RELEASE_LATER associations don't get released.
if (deallocating)
laterRefs.append(&i.second);
} else {
i.second.releaseHeldValue();
}
}
for (auto *later: laterRefs) {
later->releaseHeldValue();
}
} 四 关联属性如何进行内存管理?
通过前面的介绍,我们知道在关联属性时,可以通过关联策略来设置一些和内存管理相关的选项,在设置关联属性时,如果需要的话,其内部会根据内存管理策略对旧值进行release操作,但是你是否有想过,当对象正常的生命周期结束后,这些关联属性占用的的内存是如何回收的?这就需要我们从系统的dealloc方法中寻找答案了。
系统对象在销毁时,dealloc方法最终会执行到一个名为objc_destructInstance的内部函数,此函数实现如下:void *objc_destructInstance(id obj)
{
if (obj) {
bool cxx = obj->hasCxxDtor();
// 通过标记获取此对象是否有关联属性
bool assoc = obj->hasAssociatedObjects();
if (cxx) object_cxxDestruct(obj);
// 移除掉此对象的关联属性
if (assoc) _object_remove_assocations(obj, /*deallocating*/true);
obj->clearDeallocating();
}
return obj;
} 其中会判断要销毁的对象是否有关联属性,如果有,又会调用到_object_remove_assocation函数来进行关联属性的移除,这个函数前面介绍过,内部会处理内存管理问题。
五.关联管理器与表的创建时机
在整个关联属性实现方案中,还有一点我们没有闭环介绍,即全局的关联管理器和Hash表是怎么创建的,何时创建的。我们目前只看到,当要设置或获取关联属性时,直接拿到管理器和Hash表进行使用,并无初始化。其实,这些全局数据结构的创建在runtime初始化时就已经完成,流程路径如下:
1. 调用runtime入口函数_objc_init
2. 通知调用map_images函数
3. 调用map_images_nolock函数
4. map_images_nolock其中会调用arr_init函数,此函数实现如下:void arr_init(void)
{
AutoreleasePoolPage::init();
SideTablesMap.init();
_objc_associations_init();
if (DebugScanWeakTables)
startWeakTableScan();
} 可以看到,此函数会进行自动释放池,关联属性等逻辑的初始化。专注技术,热爱生活,交流技术,也做朋友。
—— 珲少 QQ:316045346
同时,如果本篇文章让你觉得有用,欢迎分享给更多朋友,请标明出处。
|