不想废话,直接上代码进入主题,从基本的循环引用说起:
在ViewController中声明一个block属性,如下:
@interface ViewController ()
@property (nonatomic, assign) NSInteger times;
@property (nonatomic, copy) void(^block)();
@end
在viewdidLoad中,我们来模拟一下循环引用:
- (void)viewDidLoad {
[super viewDidLoad];
self.times = 0;
self.block = ^{
self.times += 3;
NSLog(@"%@---%ld", self, self.times);
});
self.block();
}
- (void)dealloc
{
NSLog(@"成功销毁");
}
上面这段代码便造成了循环引用:self对block属性有一条强引用,block中又要捕获_times实例变量,所以必须必须得保留self,即编译器自动对self的引用计数+1,这就形成了self —> block -> self
的"保留环"。即便pop出当前VC,因为该保留环的存在,我们可以看到dealloc方法不会被调用,这块内存也不会被销毁,这就造成了内存泄漏。
这种情况下,一般我们的解决方案是,把该保留环的一条强指针弱化,一般是:在block前加上:__weak type(self) weakSelf = self
即viewDidLoad里面的代码变成:
- (void)viewDidLoad {
[super viewDidLoad];
self.times = 0;
__weak type(self) weakSelf = self;
self.block = ^{
weakSelf.times += 3;
NSLog(@"%@---%ld", weakSelf, weakSelf.times);
});
self.block();
}
一般情况下,这么做完全没问题,但是在下面这种情况下,就出问题了:
__weak typeof(self) weakSelf = self;
self.block = ^{
// 延迟执行Block里面的内容
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(3.0 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
weakSelf.times += 3;
NSLog(@"%@---%ld", weakSelf, weakSelf.times);
});
};
self.block();
这段代码表示:我block里面的代码异步延迟执行;
我们在刚push进VC的时候3s内又pop出去,发现虽然dealloc方法被调用了,程序也没有崩溃,但是发现weakSelf变为了null。对_times的+3操作无效,依然为0;
这种情形下问题就出现了,也就是我们Block里面的内容还没执行完毕的时候,当前控制器已经被pop,因为self是弱指针,block就不能保留self,self的引用计数不在+1。那么self不被强指针引用,当然会变成nil。这就造成了问题,那么怎么办呢?
这种情况下,就应该在block里面让block生成一个自动变量保持对这个弱self的强引用,让其不会被销毁,即:
__weak typeof(self) weakSelf = self;
self.block = ^{
__strong typeof(self) strongSelf = weakSelf;
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(3.0 * NSEC_PER_SEC)), dispatch_get_global_queue(0, 0), ^{
strongSelf.times += 3;
NSLog(@"%@", strongSelf);
});
};
self.block();
这样,即使在3s内当前控制器被pop,那么block依然会执行,_times的值也顺利被+3,问题也就解决了。有人可能就不理解了,你又把self变回强指针,那么又会回到原来的循环引用了吗?其实不然,这里的strongSelf和self已经不是一个东西了,它对block是没有引用的。这个strongSelf是在block内部声明的局部变量,只有block对他有引用,当block被销毁时,它也会跟着被销毁。所以,这就完美的解决了循环引用的问题。
当然开发中为了图省事儿,一般都把该两句代码定义为宏,方便直接使用:
#define weakly(objc, weakObjc) __weak typeof(objc) weakObjc = objc;
#define strongly(objc, strongObjc) __strong typeof(objc) strongObjc = objc;
weakly(self, weakSelf);
self.block = ^{
strongly(weakSelf, strongSelf);
[strongSelf doSomething];
});
或者你也可以在libextobjc这个开源库中,使用
#import "EXTScope.h"
@weakify(self)
self.block = ^{
@strongify(self)
[self doSomething];
});