你的位置:首页 > 操作系统

[操作系统]iOS闭包循环引用精讲


前言

本篇文章精讲iOS开发中使用Block时一定要注意内存管理问题,很容易造成循环引用。本篇文章的目标是帮助大家快速掌握使用block的技巧。

我相信大家都觉得使用block给开发带来了多大的便利,但是有很多开发者对block内存管理掌握得不够好,导致经常出现循环引用的问题。对于新手来说,出现循环引用时,是很难去查找的,因此通过Leaks不一定能检测出来,更重要的还是要靠自己的分析来推断出来。

声景一:Controller之间block传值

现在,我们声明两个控制器类,一个叫ViewController,另一个叫HYBAController。其中,ViewController有一个按钮,点击时会push到HYBAController下。

先看HYBAController:

1
2
3
4
5
// 公开了一个方法
- (instancetype)initWithCallback:(HYBCallbackBlock)callback;
 
// 非公开的属性,这里放出来只是告诉大家,HYBAController会对这个属性强引用
@property (nonatomic, copy) HYBCallbackBlock callbackBlock;



下面分几种小场景来看看循环引用问题:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
@interface ViewController ()
 
// 引用按钮只是为了测试
@property (nonatomic, strong) UIButton *button;
// 只是为了测试内存问题,引用之。在开发中,有很多时候我们是
// 需要引用另一个控制器的,因此这里模拟之。
@property (nonatomic, strong) HYBAController *vc;
 
@end
 
// 点击button时
- (void)goToNext {
  HYBAController *vc = [[HYBAController alloc] initWithCallback:^{
    [self.button setTitleColor:[UIColor greenColor] forState:UIControlStateNormal];
  }];
  self.vc = vc;
  [self.navigationController pushViewController:vc animated:YES];
}



现在看ViewController这里,这里在block的地方形成了循环引用,因此vc属性得不到释放。分析其形成循环引用的原因(如下图):

image

可以简单说,这里形成了两个环:

  • ViewController->强引用了属性vc->强引用了callback->强引用了ViewController

  • ViewController->强引用了属性vc->强引用了callback->强引用了ViewController的属性button

对于此这问题,我们要解决内存循环引用问题,可以这么解:

不声明vc属性或者将vc属性声明为weak引用的类型,在callback回调处,将self.button改成weakSelf.button,也就是让callback这个block对viewcontroller改成弱引用。如就是改成如下,内存就可以正常释放了:

1
2
3
4
5
6
7
8
- (void)goToNext {
  __weak __typeof(self) weakSelf = self;
  HYBAController *vc = [[HYBAController alloc] initWithCallback:^{
    [weakSelf.button setTitleColor:[UIColor greenColor] forState:UIControlStateNormal];
  }];
//  self.vc = vc;
  [self.navigationController pushViewController:vc animated:YES];
}



笔者尝试过使用Leaks检测内存泄露,但是全是通过,一个绿色的勾,让你以为内存处理得很好了,实际上内存并得不到释放。

针对这种场景,给大家提点建议:

  • 在控制器的生命周期viewDidAppear里打印日志:

1
2
3
4
5
- (void)viewDidAppear:(BOOL)animated {
  [super viewDidAppear:animated];
 
  NSLog(@"进入控制器:%@", [[self class] description]);
}



  • 在控制器的生命周期dealloc里打印日志:

1
2
3
- (void)dealloc {
  NSLog(@"控制器被dealloc: %@", [[self class] description]);
}



这样的话,只要日志没有打印出来,说明内存得不到释放,就需要学会分析内存引用问题了。

场景二:Controller与View之间Block传值

我们先定义一个view,用于与Controller交互。当点击view的按钮时,就会通过block回调给controller,也就反馈到控制器了,并将对应的数据传给控制器以记录:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
typedef void(^HYBFeedbackBlock)(id model);
 
@interface HYBAView : UIView
 
- (instancetype)initWithBlock:(HYBFeedbackBlock)block;
 
@end
 
@interface HYBAView ()
 
@property (nonatomic, copy) HYBFeedbackBlock block;
 
@end
 
@implementation HYBAView
 
- (void)dealloc {
  NSLog(@"dealloc: %@", [[self class] description]);
}
 
- (instancetype)initWithBlock:(HYBFeedbackBlock)block {
  if (self = [super init]) {
    self.block = block;
 
    UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];
    [button setTitle:@"反馈给controller" forState:UIControlStateNormal];
    button.frame = CGRectMake(50, 200, 200, 45);
    button.backgroundColor = [UIColor redColor];
    [button setTitleColor:[UIColor yellowColor] forState:UIControlStateNormal];
    [button addTarget:self action:@selector(feedback) forControlEvents:UIControlEventTouchUpInside];
    [self addSubview:button];
  }
 
  return self;
}
 
- (void)feedback {
  if (self.block) {
    // 传模型回去,这里没有数据,假设传nil
    self.block(nil);
  }
}
 
@end



接下来看HYBAController,增加了两个属性,在viewDidLoad时,创建了aView属性:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
@interface HYBAController()
 
@property (nonatomic, copy) HYBCallbackBlock callbackBlock;
 
@property (nonatomic, strong) HYBAView *aView;
@property (nonatomic, strong) id currentModel;
 
@end
 
@implementation HYBAController
 
- (instancetype)initWithCallback:(HYBCallbackBlock)callback {
  if (self = [super init]) {
    self.callbackBlock = callback;
  }
 
  return self;
}
 
- (void)viewDidLoad {
  [super viewDidLoad];
 
  self.title = @"HYBAController";
  self.view.backgroundColor = [UIColor whiteColor];
 
  self.aView = [[HYBAView alloc] initWithBlock:^(id model) {
    // 假设要更新model
    self.currentModel = model;
  }];
  // 假设占满全屏
  self.aView.frame = self.view.bounds;
  [self.view addSubview:self.aView];
  self.aView.backgroundColor = [UIColor whiteColor];
}
 
- (void)viewDidAppear:(BOOL)animated {
  [super viewDidAppear:animated];
 
  NSLog(@"进入控制器:%@", [[self class] description]);
}
 
- (void)dealloc {
  NSLog(@"控制器被dealloc: %@", [[self class] description]);
}
 
@end



关于上一场景所讲的循环引用已经解决了,因此我们这一小节的重点就放在controller与view的引用问题上就可以了。

分析:如下图所示:

image

所形成的环有:

  • vc->aView->block->vc(self)

  • vc->aView->block->vc.currentModel

解决的办法可以是:在创建aView时,block内对currentModel的引用改成弱引用:

1
2
3
4
5
__weak __typeof(self) weakSelf = self;
self.aView = [[HYBAView alloc] initWithBlock:^(id model) {
    // 假设要更新model
    weakSelf.currentModel = model;
}];



我见过很多类似这样的代码,直接使用成员变量,而不是属性,然后他们以为这样就不会引用self,也就是控制器,从而不形成环:

1
2
3
4
self.aView = [[HYBAView alloc] initWithBlock:^(id model) {
    // 假设要更新model
    _currentModel = model;
}];



这是错误的理解,当我们引用了_currentModel时,它是控制器的成员变量,因此也就引用了控制器。要解决此问题,也是要改成弱引用:

1
2
3
4
5
__block __weak __typeof(_currentModel) weakModel = _currentModel;
self.aView = [[HYBAView alloc] initWithBlock:^(id model) {
  // 假设要更新model
  weakModel = model;
}];



这里还要加上__block哦!

模拟循环引用

假设下面如此写代码,是否出现内存得不到释放问题?(其中,controller属性都是强引用声明的)

@autoreleasepool {  A *aVC = [[A alloc] init];  B *bVC = [[B allcok] init];  aVC.controller = bVC;  bVC.controller = aVC;}

分析:

aVC->强引用了bVC->强引用了aVC,因此形成了一个环,导致内存得不到释放。

写在最后

本篇文章就讲这么多吧,写本篇文章的目的是教大家如何分析内存是否形成环,只要懂得了如何去分析内存是否循环引用了,那么在开发时一定会特别注意内存管理问题,而且查找内存相关的问题的bug时,也比较轻松。

源代码

本篇写了个小demo来测试的,如果只看文章不是很明白的话,如果下载demo来运行看一看,可以帮助您加深对block内存引用问题的理解。

下载地址:标哥的GITHUB下载地址

问啊-定制化IT教育平台,牛人一对一服务,有问必答,开发编程社交头条 官方网站:www.wenaaa.com

QQ群290551701 聚集很多互联网精英,技术总监,架构师,项目经理!开源技术研究,欢迎业内人士,大牛及新手有志于从事IT行业人员进入!