ObjectiveC关联属性原理分析

一 引子

Objective-C 是一种动态性很强的语言，所谓动态能力，也可以理解为运行时能力。对于 Objective-C 开发者来说，动态性所带来的编程便利无处不在。例如通过 Category 类别来扩展已有类的功能。可以使已有类拥有新的方法和属性。但是，如果你有使用 Category 来扩展类的属性，你一定了解并非简单的使用 @property 进行声明即可。例如下面的代码：

#import 
@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 

@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 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 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 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();
}

可以看到，此函数会进行自动释放池，关联属性等逻辑的初始化。

作者：珲少