block中如何避免循环引用

时间：2021-12-18 05:11:56

使用 weak–strong dance 技术

block 可以直接引用 self，但是要非常小心地在 block 中引用 self。因为在 block 引用 self，可能会导致循环引用。如下例所示：

  1. @interface KSViewController ()
  2. {
  3. id _observer;
  4. }
  5. @end
  6. @implementation KSViewController
  7. - (void)viewDidLoad
  8. {
  9. [super viewDidLoad];
  10. // Do any additional setup after loading the view, typically from a nib.
  11. KSTester * tester = [[KSTester alloc] init];
  12. [tester run];
  13. _observer = [[NSNotificationCenter defaultCenter]
  14. addObserverForName:@"TestNotificationKey"
  15. object:nil queue:nil usingBlock:^(NSNotification *n) {
  16. NSLog(@"%@", self);
  17. }];
  18. }
  19. - (void)dealloc
  20. {
  21. if (_observer) {
  22. [[NSNotificationCenter defaultCenter] removeObserver:_observer];
  23. }
  24. }

在上面代码中，我们添加向通知中心注册了一个观察者，然后在 dealloc 时解除该注册，一切看起来正常。但这里有两个问题：
    1.) 在消息通知 block 中引用到了 self，在这里 self 对象被 block retain，而 _observer 又
retain 该 block的一份拷贝，通知中心又持有 _observer。因此只要 _observer 对象还没有被解除注册，block
就会一直被通知中心持有，从而 self 就不会被释放，其 dealloc 就不会被调用。而我们却又期望在 dealloc 中通过
removeObserver 来解除注册以消除通知中心对 _observer/block 的 retain。
    2.) 同时，_observer 是在 self 所在类中定义赋值，因此是被 self retain 的，这样就形成了循环引用。
上面的过程 1) 值得深入分析一下：
苹果官方文档中对 addObserverForName:object:queue:usingBlock: 中的 block 变量说明如下：

The block is copied by the notification center and (the copy) held until the observer registration is removed.
因此，通知中心会拷贝 block 并持有该拷贝直到解除 _observer 的注册。在 ARC 中，在被拷贝的 block 中无论是直接引用 self 还是通过引用 self 的成员变量间接引用 self，该 block 都会 retain self。
这两个问题，可以用 weak–strong dance 技术来解决。该技术在 WWDC 中介绍过：2011 WWDC Session #322 (Objective-C Advancements in Depth）

  1. __weak KSViewController * wself = self;
  2. _observer = [[NSNotificationCenter defaultCenter]
  3. addObserverForName:@"TestNotificationKey"
  4. object:nil queue:nil usingBlock:^(NSNotification *n) {
  5. KSViewController * sself = wself;
  6. if (sself) {
  7. NSLog(@"%@", sself);
  8. }
  9. else {
  10. NSLog(@"<self> dealloc before we could run this code.");
  11. }
  12. }];

下面来分析为什么该手法能够起作用。
    首先，在 block 之前定义对 self 的一个弱引用 wself，因为是弱引用，所以当 self 被释放时 wself 会变为 nil；然后在 block 中引用该弱应用，考虑到多线程情况，通过使用强引用 self 来引用该弱引用，这时如果 self 不为 nil 就会 retain self，以防止在后面的使用过程中 self 被释放；然后在之后的 block 块中使用该强引用 self，注意在使用前要对 self 进行了 nil
检测，因为多线程环境下在用弱引用 wself 对强引用 sself 赋值时，弱引用 wself 可能已经为 nil 了。
    通过这种手法，block 就不会持有 self 的引用，从而打破了循环引用

相关文章