如何实现ARC中weak功能？
原文链接
我们都知道ARC中weak与assign或者说unsafe_unretained最大的不同就是设置weak属性后，系统会在对象被释放后自动将指向对象的指针置为nil，而assign则会产生一个悬空指针，那么系统是如何实现这一机制呢？我们能否自己模拟系统对weak的实现呢？
通过查看runtime源码中objc-accessors.h和objc-weak.h部分，我们大概可以了解系统针对weak的实现方式与strong或者copy的实现方式是不同的。
对于注册为weak的对象，系统会以weak指向的对象内存地址作为key，将之放入到一个hash表之中，当此对象的引用计数为0时会调用dealloc，之后遍历hash表中此key所对应的对象，将之置为nil，关于这部分详细内容可以自行查看源代码，在此不进行扩展。我们这里重点介绍下如何自行模拟系统这一机制。
分析
首先我们回想一下weak机制的几个步骤，其中真正与assign不相同的部分在于当对象调用dealloc之后对对象指针置空操作。那么我们很容易联想到Objective-C中关联对象的使用，关联对象也是在对象调用dealloc之后被移除，这样我们是否可以通过对对象添加一个关联对象来模拟weak的实现呢？
探究
下面我们通过代码来说明下如何自行实现一个类似weak机制的对象管理机制，本文中所用到的代码存放在https://github.com/samwei12/WeakDemo ：
unsafe_unretained和weak具体使用中的差异
- 首先，创建一个
ClassA类， 创建一个
ClassB类，并在里面添加一个测试方法print，用于等下向ClassB实例对象发送消息，确认对象是否仍在使用- (void)print { NSLog(@"Object is %@", self);
}给
ClassA添加一个ClassB对象属性，并设置为unsafe_unretained@property (nonatomic, unsafe_unretained) ClassB *objectB;在main函数中添加如下代码：
@autoreleasepool { ClassA *instanceA = [ClassA new];
ClassB *instanceB = [ClassB new];
instanceA.objectB = instanceB;
[instanceA.objectB print];
// release instanceB
instanceB = nil;
[instanceA.objectB print];
}
// prevent process be killed immediately
sleep(3);
return 0;运行一下代码，不出意外的话，程序会挂掉，因为
instanceA.objectB所指向的内存已经在instanceB = nil;时候被释放掉，instanceA.objectB仅指向一个悬空指针：2016-03-09 15:25:10.427 WeakDemo[98402:1613532] Object is 0x7fa080d0f100 2016-03-09 15:25:10.432 WeakDemo[98402:1613532] Object is 0x7fa080d0f100
Process finished with exit code 139当然，也有可能不会挂，这取决于执行
print函数时，instanceB所在的内存是否完全释放掉。如果将
ClassA中的属性改为@property (nonatomic, weak) ClassB *objectB;则不会出现crash，这就是weak的作用了，objectB对象如果被释放掉，则该指针变为nil，而向nil发送消息是不会出现问题的。
如何给unsafe_unretained添加weak功能
下面我们就使用unsafe_unretained来一步一步模拟weak修饰符：
创建一个
DeallocBlock类，代码如下：typedef void (^voidBlock)(); @interface DeallocBlock : NSObject
@property(nonatomic, copy) voidBlock block;
- (instancetype)initWithBlock:(voidBlock)block1;
@end
@implementation DeallocBlock
- (instancetype)initWithBlock:(voidBlock)block1 {
self = [super init];
if (self) {
_block = [block1 copy];
}
return self;
}
- (void)dealloc {
if (_block) {
NSLog(@"DeallocBlock dealloc!");
_block();
}
}
@end覆盖它的
dealloc函数，这样就实现了该对象被释放时执行外部传入的block功能。添加一个
NSObject的category，如下：#import#import "DeallocBlock.h"
#import
@interface NSObject (deallocBlock)
- (void)runBlockOnDealloc:(voidBlock)block;
@end
@implementation NSObject (deallocBlock)
- (void)runBlockOnDealloc:(voidBlock)block {
if (block) {
DeallocBlock *deallocBlock = [[DeallocBlock alloc] initWithBlock:block];
objc_setAssociatedObject(self, _cmd, deallocBlock, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
}
@end这里主要实现的功能是，给所有的
NSObject对象添加一个传入block的方法，这个方法的作用就是在其中根据传入的block生成一个DeallocBlock实例，并使用关联对象将这个实例关联到自身。之后当这个对象被释放时候，会自动移除所有的关联对象，也就会触发DeallocBlock实例的dealloc方法，执行我们传入的block。接下来，我们重载
ClassA的setObjectB方法：- (void)setObjectB:(ClassB *)objectB { _objectB = objectB;
// 仅对objectB != nil case做处理
if (_objectB) {
[_objectB runBlockOnDealloc:^{
NSLog(@"_objectB dealloc");
_objectB = nil;
}];
}
}这段代码主要作用就是给
_objectB添加一个deallocBlock,这样，如果_objectB所指向的内存已经被释放掉，则会调用我们传递进去的block来将此悬空指针置为空再次运行程序：
2016-03-09 15:53:36.881 WeakDemo[99270:1639867] Object is 0x7fad79c0f310 2016-03-09 15:53:36.882 WeakDemo[99270:1639867] DeallocBlock dealloc!
2016-03-09 15:53:36.882 WeakDemo[99270:1639867] _objectB dealloc
Process finished with exit code 0由此可知，在
main函数中执行instanceB = nil;之后，开始执行deallocBlock，之后由于instanceA.objectB已经为nil，print方法不执行操作。
总结
至此，我们已经实现了自己模拟系统weak机制的运行，给一个本会产生悬空指针的unsafe_unretained修饰符加上了weak的功能。大家可以自行尝试https://github.com/samwei12/WeakDemo,如果有任何问题，欢迎指正。
参考文档
- https://www.opensource.apple.com/source/objc4/
- Fun With the Objective-C Runtime: Run Code at Deallocation of Any Object