如何使用 Runtime 给现有的类添加 weak 属性

事先说好

​前不久看到 @sunnyxx 想找一个性取向正常的实习生帮他分担一点工作量，当想起他和 @ibireme 秀的亲密自拍后我就知道事情并没有这么简单→_→。但是作为刚毕业且性取向正常的我还是比较关心滴滴的招人水准，于是我便想起了之前他发的一份面试题，其中有一题就是如何使用Runtime实现weak属性。在那之后 @iOS程序犭袁 整理了一份有关这一份面试题的 参考答案，也包括了这个题目。
​看了整理的答案后觉得方法妥当，唯一不足的是不够严谨。于是我自己学习答案上的思维后修补了一些不足之处。如有大神觉得可以改进的地方望不吝指出~。

​前方高能预警！！！！老司机要开车了！！请站稳坐稳手扶好。。。

​这是一篇“我认为是一个很复杂的”文章。虽说复杂，但是涉及的都是OC基础知识。文章前面会介绍一下涉及点，后面是分析需求、定制方案以及具体实现。如果看不懂的话仅仅了解文章前面的一些基础知识点也是不错滴（也可能是我写的不够好。。。新手写文章请轻喷）。

​另外我不怎么喜欢倒序介绍，而是从零开始讲起，因为我更希望读者能够了解思考的步骤，锻炼思维，而不是仅仅看见成果。如果你不喜欢这种风格，我推荐你从后往前看。。。

还有，文中涉及的代码均在 GitHub 上

涉及点——weak & assign

​很多公司面试时候经常用这个作为面试题：weak 和 assign 有什么区别？

​这个问题在本文中非常重要，因为 runtime 中association_policy中所提供的只有 assign 却没有 weak。相比于别的修饰符，assign 是最接近 weak 的，所以我们会改造 assign 去完成 weak 的工作。

weak 和 assign 的区别

​我是个新手啊喂。。。如果有错的指出来。。。

项目	weak	assign
修饰对象	对象	基本数据类型（也可以是对象）
赋值方式	复制引用	复制数据
对对象的引用计数器的影响	无影响	无影响
对象销毁后	自动为nil	不变

​通过这个表格总结的来看，在持有对象的情况下，weak 和 assign 最大的区别，也就是本文研究方向就在于最后一条，对象销毁后如何将引用设置为nil？

这里扩充一下，对象销毁后，weak 修饰的 property 会自动设置为 nil，这个最大的好处就是之后发送的消息都不会因为对象销毁而出错；assign 修饰的 property 并不会自动变为 nil，形成野指针，所以在此之后如果没有判断对象是否销毁的话，很有可能就会对野指针发送消息导致crash。

官方来说，如果不想增加持有对象的引用计数器的话，推荐使用 weak 而不是 assign，这一点从 Apple 提供的头文件就可以看出——所有 delegate 的修饰符都是 weak。

涉及点——关联对象

​关联对象是 runtime 中的一个比较重要的技能，在此我假设你已经了解了关联对象的操作，并且你也会使用关联对象为已有的类添加属性。如果你对 runtime 的知识还不够了解的话，可以去网络上搜寻一些文章来看，或者去我的 GitHub 看我写的 runtime 系列文章。关联对象最主要的就是下面这两个c函数：

void objc_setAssociatedObject(id obj, const void *key, id value, objc_AssociationPolicy policy);
id objc_getAssociatedObject(id obj, const void *key);

​接下来介绍一下本文涉及关联对象中两个比较重要的东西。

key 的取值

​我偷偷告诉你，key 是一个坑。。。其实 key 是一个很好理解的东西，说白了就是属性名称嘛，而且又是const void *类型的，那传一个 C 字符串不就好了？于是我们可以这样写：

/** 这是 某个已有类 的分类 CategoryProperty 在 .h 文件中的一个新增的属性 */
@property (nonatomic, strong) NSObject *categoryProperty;

/** 这是 .m 文件中的 set 方法的实现 */
- (void)setCategoryProperty:(id)categoryProperty {
    objc_setAssociatedObject(self, "categoryProperty", categoryProperty, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

/** 这是 .m 文件中的 get 方法的实现 */
- (id)categoryProperty {
    return objc_getAssociatedObject(self, "categoryProperty");
}

​恩，测试了一下没毛病，但是我觉得这个关联对象可以封装成 NSObject 的一个分类，以便日后操作。于是可以这样写：

/**
 * 这是 NSObject 基于 Runtime 而增加的分类，之后如果想要给现有的类添加属性的话，可以直接调用这个分类
 **/
@implementation NSObject (Association)

/** 所有要增加的属性的 set 方法都可以调用这个方法来实现 */
- (void)setAssociatedObject:(id)obj forKey:(NSString *)key {
    objc_setAssociatedObject(self, key.UTF8String, obj, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

/** 所有要增加的属性的 get 方法都可以调用这个方法来实现 */
- (id)associatedObjectForKey:(NSString *)key {
    return objc_getAssociatedObject(self, key.UTF8String);
}

@end

​恩，测试了一下没毛病，之后我们就可以这样给已有的类添加属性了：

/** 这是 某个已有类 的分类，并基于上述的 NSObject 的 Runtime 分类而实现的增加属性 */
@property (nonatomic, strong) NSObject *categoryProperty;

/** 利用上述的分类添加属性的 set 方法的实现 */
- (void)setCategoryProperty:(id)categoryProperty {
    [self setAssociatedObject:categoryProperty forKey:@"categoryProperty"];
}

/** 利用上述的分类添加属性的 get 方法的实现 */
- (id)categoryProperty {
    return [self associatedObjectForKey:@"categoryProperty"];
}

​恩，测试了一下没毛病。诶？不知是什么力量让我想到了如果我传入的是一个 NSMutableString 或者是由程序运行期间生成的 NSString 会怎么样呢？他们的内容一样，对应的 UTF8String 的指针指向的 char * 的内容也一样，但是 UTF8String 的地址不一样，这样会影响到取值吗？程序猿要有44944的决心！

NSObject *obj = [NSObject new];     // create a object to bind new property
NSString *nameForValue = @"Xcode";  // name value
NSString *nameForCode = @"name";    // name key 1
NSString *nameForBind = [@"na" stringByAppendingString:@"me"];  // name key 2
[obj setAssociatedObject:nameForValue forKey:nameForCode];  // write value with key1 => obj.name = @"Xcode"
NSLog(@"%@", [obj associatedObjectForKey:nameForBind]); // read value with key2 => nil

​看到这里有一种“大清要完”的感觉。。。好吧，其实这个 key 并不只是读 C 字符串那么复杂，实际上仅仅是单纯地用一个地址（当然你可以理解为一个 long long long long 的整数）作为键来保存该关联关系。。要不要这么坑！！！

​ 再扩充，为什么之前在代码里写死 key 却很正常呢？set 方法和 get 方法里分别写死了两个相同的字符串，而不是使用同一个变量传入。

​ 因为在代码里写死的字符串会在程序运行期间与可执行的汇编代码被操作系统从硬盘中拷贝到内存中，并且保存在当前进程的代码区字符常量区。代码区字符常量区里的数据是无法更改的。这一点可以用下面简单的代码来验证：

char *pre = "Hello ";
char *suf = "Word!";
char *newStr = strcat(pre, suf); // => error

​ 而编译器在编译过程中会整理好所有代码里写死的字符串，并把他们统一整理到编译后的文件的某一块区域，这一点可以用 IDE IDA 来验证：

IDA反编译“设置” App 后的数据布局图

​ 图为 iOS 系统设置 App 的可执行文件内容布局图，蓝色部分是可执行的代码，灰色部分是字符串（包括所有的类名、所有方法名、所有函数名、所有框架路径等等，以及 coder 写死的字符串）。编译器在整理代码中的字符串的时候，发现两个字符串相同，于是在字符串表里只保存了一个字符串，并将两处的代码的引用指向这个表的同一个字符串中，我们也可以用简易的代码来验证一下：

NSString *str1 = @"123";
NSString *str2 = @"123";
NSString *str3 = [[NSString alloc] initWithString:str1];
NSString *str4 = [str1 stringByAppendingString:@""];
NSLog(@"%p, %p", str1, str1.UTF8String);  // 0x100001050, 0x100000f4e
NSLog(@"%p, %p", str2, str2.UTF8String);  // 0x100001050, 0x100000f4e
NSLog(@"%p, %p", str3, str3.UTF8String);  // 0x100001050, 0x100000f4e
NSLog(@"%p, %p", str4, str4.UTF8String);  // 0x100001050, 0x100000f4e
        
char *cstr1 = "321";
char *cstr2 = "321";
printf("%p\n", cstr1);    // 0x100000f60
printf("%p\n", cstr2);    // 0x100000f60

​但是，轮子一定要方便使用才能成为好轮子啊！那怎么办？！？
在什么环境下 nameForCode 和 nameForBind 是相同的呢？那就只能是容器类（集合类）！

​在 NSDictionary 中，用这两个 name 作为 key 可以取到相同的对象，所以我们可以考虑将 const char * 作为 value ，NSString * 作为 key 保存在一个 NSDictionary 中，然后利用这个字典将内容相同的 NSString 统一转为同一个 const char * 值就好了么~似乎很有道理，但是 const char * 是基本数据类型，不能保存到 OC 容器类中，所以我们需要用 NSValue 来包装一下。

​于是我们有：

/** 这是一个 NSString => NSValue< const char * >的字典 */
static NSMutableDictionary *keyBuffer;

@implementation NSObject (Association)

+ (void)load {
    keyBuffer = [NSMutableDictionary dictionary]; // 创建字典
}

/**
 * set 方法，以供以后添加属性时候给这个属性的 set 方法调用
 *
 * @param   object      要关联的对象，也就是要设置的新的属性值
 * @param   key         属性名称，传入新增属性的名称
 **/
- (void)setAssociatedObject:(id)object forKey:(NSString *)key {
    const char *cKey = [keyBuffer[key] pointerValue]; // 先获取key
    if (cKey == NULL) { // 字典中不存在就创建
        cKey = key.UTF8String;
        keyBuffer[key] = [NSValue valueWithPointer:cKey];
    }
    objc_setAssociatedObject(self, cKey, object, OBJC_ASSOCIATION_RETAIN);
}

/**
 * get 方法，以供以后添加属性时候给这个属性的 get 方法调用
 *
 * @param   key     属性名称
 */
- (id)associatedObjectForKey:(NSString *)key {
    const char *cKey = [keyBuffer[key] pointerValue];
    if (cKey == NULL) {
        return nil;
    } else {
        return objc_getAssociatedObject(self, cKey);
    }
}

@end

基于这个分类，我们就可以用 1 行 set 方法 + 1 行 get 方法即可实现关联对象。

policy 的选择

policy 就是修饰符，在 runtime 中已经提供了一些修饰符：

/**
 * Policies related to associative references.
 * These are options to objc_setAssociatedObject()
 */
typedef OBJC_ENUM(uintptr_t, objc_AssociationPolicy) {
    OBJC_ASSOCIATION_ASSIGN = 0,           /** assign */
    OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, /** retain, nonatomic */
    OBJC_ASSOCIATION_COPY_NONATOMIC = 3,   /** copy, nonatomic */
    OBJC_ASSOCIATION_RETAIN = 01401,       /** retain, atmoic */
    OBJC_ASSOCIATION_COPY = 01403          /** copy, atomic */
};// 把他们改成小写是不是一瞬间天空的星星都亮了呢 (｡・`ω´･) ~~

​然而并没有我们想要的 OBJC_ASSOCIATION_WEAK 这个选项。。(╬￣皿￣)凸，所以我们就用 assign 来改造一下吧。。。

涉及点——容器类检索方式

​这一部分是为了优化轮子而加入的，因为不能确定以后使用这个轮子时候的引用的复杂情况，很有可能某个容器过于庞大，导致检索这个容器所需要大量的时间而影响程序运行，所以会在之后的一部分的内容中使用下面的东西。。（恩，我觉得用“下面的内容”会比较好一点）

子对象的比较标准

​论一个对象是否存在一个容器中？

​对于 Foundation 所提供的容器类中，都是以 isEqual: 方法作为标准来判断是否是同一个对象。也就是说，对于不同的 Person 他们如果有相同的 identifier （身份证号码） 就可以算是同一个人，而不是看他们现在住的位置（首地址）。但是，NSObject 的 isEqual: 方法的实现却是比较对象的首地址是否相同。所以如果容器类需要操作 NSObject 的对象（非子类），则就是调用 isEqual: 后比较对象的首地址。我想说的是，我们可以在子类中重写 isEqual: 方法来达到定制“是否相同”的标准。

子对象的查找方式

​每一个容器类都有一个方法：containsObject: 是否包含某个对象。

​然而判断一个对象是否存在一个数组中，最简单的办法就是遍历，但是这也是最耗时的办法。如果你研究过算法，你可以尝试使用二分法、快速查找法等方法来检索一个项，这些仅限于有序数组。但是容器内保存的是复杂的对象，并不是一个可以比较大小的数值，所以这些算法行不通。

​苹果大大给出的方法是：在 Foundation 中，将一个对象存入容器类后，容器类会读取该对象的 hash 值，并且把这个值存入一个有序的列表中。之后，检索一个对象是否存在一个容器的时候，即可以取出 hash 值并依据当前容器中对象的个数选择最适合的算法，和这个有序的列表进行比较即可。

​hash 值是一个 NSUInteger 类型的数值，我们可以通过重写 hash 方法来定制 hash 值的计算公式。但为了严谨，我们要保证 isEqual: 返回 YES 的两个对象的 hash 值要相等。

原方案分析

​不管怎么样，在此先感谢出题者 @sunnyxx 能够让我更深入的学习到 assign 和 weak 的知识以及后面的解决方案，同时也要感谢 @iOS程序犭袁 之前整理的答案，让我有一种站在巨人的肩膀上的感觉。

​原文连接：《招聘一个靠谱的 iOS》面试题参考答案（上）

​当然，如果你喜欢看的话顺便也安利一波这篇文章所在的 repo （里面还有下篇） : GitHub:ChenYilong/iOSInterviewQuestions

​原文里提到了 weak 的底层实现，并仿照其底层实现利用 runtime 实现了 weak 变量。具体的内容可以查看原文，在此仅提取比较易懂的部分。

最单纯的实现原理

注意，本小节的解决方案以及 Demo 和原文中的参考答案不一样，本节仅仅是为了分步骤分析原文的参考答案

​要实现 weak ，说白了就是要做到两点：1、引用计数器不变；2、对象销毁后自动设置为 nil。而在 runtime 所提供的枚举中，OBJC_ASSOCIATION_ASSIGN 就已经做到了第一点，我们只需要实现第二点即可。第二点是要在对象销毁后，将 weak 引用设置为 nil ，所以我们要捕获这个对象销毁的时机，或者接收这个对象销毁的事件。在 ARC 中，对象销毁时机其实就是 dealloc 方法调用的时机，我们可以在这个方法里将这个 weak 引用设置为 nil。于是我们可以有下面的思维图：（假设 a.xxx = b，则下图中“宿主对象”就是 a，“某属性”就是 xxx，“值对象”就是 b）

初步方案思维图

​要实现上图的逻辑，我们就需要在两个时机做很多事情：

1. 建立 weak 引用时机，即执行 宿主对象.某属性 = 值对象 这段代码的时候

   我们需要告诉“值对象”，�你被一个叫“宿主对象”的对象用“某属性”弱引用了。值对象记录下 “谁” 用 “什么属性” 弱引用了自己

2. 值对象在销毁时机，即在执行 dealloc 方法的时候

   把自己记录的 “谁” 的 “什么属性” 设置为 nil。

这些就是基本的实现思路。同时，为了不影响 “宿主对象” 的生命周期，故 1 中保存 “宿主对象” 的引用也要是 weak 的。总结以上分析后，即可得到我们第一步的代码，你可以在 Demo-PlanStep-1 看到这个版本的代码。

/** 宿主类的分类 */

@implementation MUHostClass (Association)

const static char kValueObject = '0';

- (void)setValueObject:(MUValueClass *)valueObject {
    objc_setAssociatedObject(self, &kValueObject, valueObject, OBJC_ASSOCIATION_ASSIGN);
    [valueObject setWeakReference:self forWipeSEL:@selector(setValueObject:)];
}

- (MUValueClass *)valueObject {
    return objc_getAssociatedObject(self, &kValueObject);
}

@end

/** 值类的主要实现 */

@interface MUValueClass ()
{
    __weak id _hostObj;
    SEL _hostWipeSEL;
}

@end

@implementation MUValueClass

- (void)setWeakReference:(id)hostObj forWipeSEL:(SEL)wipeSEL {
    _hostObj = hostObj;
    _hostWipeSEL = wipeSEL;
}

- (void)dealloc {
    // 此处用宏取消 ARC 的警告
#pragma clang diagnostic push   // 创建取消警告域
#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
    [_hostObj performSelector:_hostWipeSEL withObject:nil];
#pragma clang diagnostic pop    // 关闭取消警告域
}

@end

​整理 Demo-PlanStep-1 的代码，即可知道初步方案的开发者角度的代码量：

1. 宿主类的分类：key定义（1行）+ set方法（2行）+ get方法（1行） = 4行
2. 值类：实例变量定义（2行）+ 传值方法（2行）+ dealloc方法（1行）= 5行 （该步骤必须在类的“主实现”里做）

总计：9 行 （其中包括改写已有类的主文件 5 行，也就是 “值类” 的一个方法和 dealloc 的代码）

​排除轮子内部实现的代码量（压根就没有），使用轮子的人用 7 行代码即可实现一个 weak 属性，这一点还是可以接受的。方案一里需求基本达到，但是这个方案需要开发者在“有可能被弱引用的对象的类”里去写代码，违背了轮子的“不入侵源代码，减少对源代码的影响”的规则，故这个方案不可行，接下来改造方案一。

参考答案中的实现

注意，本小节的解决方案以及 Demo 和原文中的参考答案基本一致，仅个别命名不同，且均有上一小节改造而成。

​在上一小节中实现的轮子，最大的问题就是对项目的源代码入侵太严重（这里指需要在 值类 的 主要实现 中添加特定的代码）。本小节将改进，降低对源代码的影响（也就是在所有类的 主要实现 中不需要添加任何代码）。

​分析第一个方案中，对源代码入侵的部分：

1. “值类”必须要有一个公开的方法，用于传递“宿主对象”和“属性”的 set 方法。

   不在“主要实现”里给一个现有的类添加一个方法以及实现，OC 中“Category”是现成的啊！！！不过仔细想想，“值类”要 weak 引用“宿主对象”，并且这个要在“Category”中实现，这不就是本文的研究内容么 = =。。这就很尴尬了。

2. “值类”销毁的时候要执行一段特殊的代码（也就是 宿主对象.某属性 = nil ）

   在 dealloc 方法里写代码主要目的还是捕获这个对象的销毁事件，要为这一部分优化的话，就必须换一个捕获方式。

经过分析，遇到了两个难点。1、在 Category 中创建一个弱引用属性；2、换一种方式捕获对象销毁时机。

​很好，第一个难点好无方向，先解决第二个难点。

​对象销毁后必然调用它的 dealloc 方法，但我们又不能去修改这个方法，不如我们可以看看在没有重写 dealloc 时，这个方法到底干了什么（ARC）？其实很简单，我们只要思考一下在 MRC 下我们应该在 dealloc 方法里写什么？

1. 对自己所有的强引用的 Ivar 发送 release 消息
2. 对自己所有的强引用的关联对象发送 release 消息
3. ...
4. 调用 [super dealloc]

前两项非常显眼（好吧，我故意的），总得来说，会释放自己所有强引用对象。第一点释放的是 Ivar 对象，Ivar是主要实现中的元素，所以我们不考虑，要不起。第二点是释放关联对象，刚好 Category 中就是用这个东西，可以考虑从他入手。

扩展：当一个对象被发送 release 消息的时候，会先判断自己的引用计数器是不是大于 1，如果是则减一，否则自毁（自毁时引用计数器为仍旧为 1）。

设想一下，假如有一个对象 a，被这个“值对象”强引用着，同时没有其余的对象强引用 a，即 a 的引用计数器在 a 活着的任何时间点都为 1，则 a 的销毁时间与“值对象”同步。即：

"值对象"的引用计数器为0
|   -> “值对象”调用 dealloc 方法
|   |   -> [a release]
|   |   |   -> a 对象的引用计数器为 0
|   |   |   |   ->a 对象调用 dealloc 方法
|   |   |   -> a 对象销毁
-> "值对象"销毁

我们可以自己创建一个 A 类，然后在“宿主对象”和“值对象”建立 weak 关系的时候，偷偷地创建一个 A 类的实例 a，绑定在 “值对象” 上。当“值对象”销毁后，这个 a 也会被销毁。而 A 类是轮子的内部类，其 dealloc 方法可以随意改造。这样就可以把 宿主对象.某属性 = nil 这段代码写在 A 类的 dealloc 方法里。由于 [a dealloc] 与 [值对象 dealloc] 是一起执行的，我们便做到在不改原有类的情况下捕获原有类的 dealloc 方法。总结来说在 Category 里我们要用关联对象的方法让“值类型”强引用 a。

​由于 a 的存在，且 A 是一个内部类，因此我们可以给 A 类创建一个弱引用属性，让他持有“宿主对象”。同时可以给 A 类创建一个 SEL 属性，让他持有 set 方法。不知不觉就把问题 1 给解决了ヾ(=^▽=)ノ。

​改造后的方案代码是 Demo-PlanStep-2 ，思维图如下：

参考答案中的思维图

整理核心代码，具体如下（涉及 block 是无参无返回值的 block）：

/** 宿主类的分类实现 */
@implementation MUHostClass (Association)

const static char kValueObject = '0';

- (void)setValueObject:(MUValueClass *)valueObject {
    objc_setAssociatedObject(self, &kValueObject, valueObject, OBJC_ASSOCIATION_ASSIGN);
    /**
     *  1. 虽然这里没有循环引用，但是还是需要把弱引用丢给 block
     *     因为 valueObj 持有 weakTask，weakTask 持有 block，block 持有 self
     *     因此 self 至少要等到 valueObj 销毁后才能销毁。严重影响到 self 的生命周期
     *
     *  2. 而使用传递 block 的方式清空属性，而不是传递 set 方法的 SEL 的方式，是为了防止形成递归
     *  3. 第2点是我瞎说的
     */
    __weak typeof(self) wself = self;
    [valueObject setWeakReferenceTask:^{
        objc_setAssociatedObject(wself, &kValueObject, nil, OBJC_ASSOCIATION_ASSIGN);
    }];
}

- (MUValueClass *)valueObject {
    return objc_getAssociatedObject(self, &kValueObject);
}

@end

/** 给所有的类添加扩展 */

@implementation NSObject (MUWeakTask)

static const char kWeakTask = '0';

- (void)setWeakReferenceTask:(TaskBlock)task {
    MUWeakTask *weakTask = [MUWeakTask taskWithTaskBlock:task];
    objc_setAssociatedObject(self, &kWeakTask, weakTask, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

@end

/** 传说中的 A 类 */
  
@implementation MUWeakTask

- (instancetype)initWithTaskBlock:(TaskBlock)taskBlock {
    self = [super init];
    if (self) {
        _taskBlock = [taskBlock copy];
    }
    return self;
}

+ (instancetype)taskWithTaskBlock:(TaskBlock)taskBlock {
    return [[self alloc] initWithTaskBlock:taskBlock];
}

- (void)dealloc {
    if (self.taskBlock) {
        self.taskBlock();
    }
}

@end

整理 Demo-PlanStep-2 的代码，即可知道改进方案的开发者角度的代码量：

1. 宿主类的分类：key定义（1行）+ set方法（4行）+ get方法（1行） = 6行
2. 值类：0 行

总计：6 行 （其中不存在改写已有类的主文件的代码）

不知你还是否记得上文中涉及点——关联对象中的key 的取值小节里，我们封装了一个关联对象的分类，仅仅用两行代码实现关联对象（同时不需要 key 的定义），所以可以利用那一小节的思维，封装好 set 方法中的代码，减少开发者的代码量，统计封装之后数据为：

1. 宿主类的分类：key定义（0行）+ set方法（1行）+ get方法（1行）= 2行
2. 值类：0 行

总计：2 行

需求分析

​通过上述分析，基本了解了这个参考答案的设计思维，已经可以满足一部分的需求。但是如果我们仔细分析实际开发的情况，就会发现有很多 bug。

情况一——宿主对象设置新的值

描述：宿主对象的 weak 属性是可以重复设置值的，当二次设置某个对象的属性后，就会覆盖之前的值，而此时之前的值对象就和宿主对象无任何关系。

问题：在参考答案中，若出现描述中的情况时，旧的“值对象”引用的 DeallocTask 对象中保存的销毁任务 block 依旧是这个宿主对象的 block 。当这个旧的值对象销毁后仍旧会执行那个 block ，这样就会导致宿主对象的新值会“无缘无故”没了。。。（新的值对象表示：宝宝躺着也中枪。。）

方案：属性设置新的值之前，把旧的值的 block 删除。

情况二——宿主对象有多个弱引用属性指向同一个值

描述：UITableView 的 dataSource 和 delegate 一般都是同一个 UIViewController

问题：若出现描述中的情况时，“值对象”只能保存一个 DeallocTask ，具体如下：

hostObj.property1 = valueObj;   // 先设置
hostObj.property2 = valueObj;   // 后设置
valueObj = nil;                 // 值对象销毁
hostObj.property1;              // 依旧是值对象
hostObj.property2;              // nil

（property1 表示：有了新欢就忘了旧爱！！！能不能有始有终！！！）

方案：具体方案请看情况三

情况三——多个宿主对象弱引用同一个值

描述：一个对象很有可能被多个变量弱引用，而当这个对象销毁后，要把所有弱引用它的变量都要设置为 nil。

问题：与情况二相同，“值对象”只能保存最后一个弱引用他的“宿主对象”的销毁 block。

方案：该方案与情况二通用，均解决多引用同对象的问题。简单来说“有几个 weak 引用自己，自己就有几个销毁 block”。
因此，“自己的 deallocTask 对象保存的 block 应该是多个，而不是单个”，也就是我们需要一个容器存放多个 block 并在 dealloc 方法中依次执行。

综上所述

1. block 可以被删除
2. block 可以被增加

所以我们需要一个 Mutable 容器来保存 block 。另外，由于很有情况出现同宿主的不同属性的情况，以及不同宿主的同属性情况，所以在这个容器里应该是建立一个 obj,key => block 的映射关系，即由一个“宿主对象”和一个“属性名称”决定一个销毁 block 。在 Foundation 中，键值映射关系由 NSDictionary 或者 NSMapTable 实现，但都是“一对一”的关系。所以我们可以建立一个新的类，用作映射关系的 键 ，这个类保存一个对象（weak）和一个字符串，并且利用 hash 和 isEqual: 方法使这个特定的 键 能够正常工作。

这“可能”是最完美的解决方案

改造 DeallocTask 类，让其拥有保存多个 block 的能力：

array 属性，保存 DeallocTaskItem 对象列表，每一个 DeallocTaskItem 保存 hostObj、property、block

add 方法，遍历 hostObj、property 都相同的 item 是否存在 array 中，存在就销毁，然后创建 item 存入 array

remove 方法，从 array 中删除相同的 hostObj、property 的那个 item

dealloc 方法，遍历 array，并执行每一个 item 中的 block

改造新的 property 的 set 方法：

获取旧的值对象，并调用这个值对象的 deallocTask 的 remove 方法

设置新的值对象

调用新的值对象的 deallocTask 的 add 方法

以上代码均在 Demo-PlanStep-Release 中实现。整理代码：

1. 宿主对象的分类：key定义（0行）+ set方法（1行）+ get方法（1行）= 2 行
2. 值对象：0 行

总计：2 行

轮子经过改造，实现 2 行代码给现有的类添加 weak 属性

写在后面

如果你很淡定地看到这里，这说明你的水平可能很高或者可能很低。。如果你是大神，也烦请支出文章中的不足，毕竟我还有很多需要学习的地方。如果你是新手，就算看不懂也可以学习学习文中的一些涉及到的基础知识，还有一些细节。

在此非常感谢你能看到这里。

最后补充一句：我的性取向非常正常。