NSNotification引起的内存泄漏和循环引用

0.358字数 751阅读 1831

在上一篇《OC循环引用》的文章中,介绍了NSNotification会导致循环引用,我们先来看一下那个例子。

@implementation SecondViewController
- (void)addObserver {
     [[NSNotificationCenter defaultCenter] addObserverForName:@"noticycle" object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification * _Nonnull note) {
        NSLog(@"%s, %@", __FUNCTION__, self);
    }];
}
- (void)postNotification {
    [[NSNotificationCenter defaultCenter] postNotificationName:@"noticycle" object:nil];
    NSLog(@"%s, %@", __FUNCTION__, self);
}
@end

//调用代码
@implementation FirstViewController
- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    //创建两个SecondViewController对象,VC1做观察者,VC2做通知发送者
    SecondViewController *VC1 = [[SecondViewController alloc] init];
    VC1.title = @"VC1";
    [VC1 addObserver];
    
    SecondViewController *VC2 = [[SecondViewController alloc] init];
    VC2.title = @"VC2";
    [VC2 postNotification];    
}
@end

运行结果:

2017-09-06 12:31:17.710 RetainCycleDemo[58071:30501179] -[SecondViewController addObserver]_block_invoke, VC1
2017-09-06 12:31:17.710 RetainCycleDemo[58071:30501179] -[SecondViewController postNotification], VC2
2017-09-06 12:31:17.712 RetainCycleDemo[58071:30501179] -[SecondViewController dealloc], VC2

当时看到这个运行结果,便果断的判断了是循环引用导致的问题,但却不知道为什么。后来在学习了AFN的循环引用问题处理方式之后,突然想到,NSNotificationCenter和NSURLSession可能类似,是由于NSNotificationCenter没有释放,一直还持有block,所以导致VC1没有释放。用图来表示:

NSNotificationCenter的block中使用self,导致self不销毁

到底是不是这个原因呢,我们从头入手

查看- (id<NSObject>)addObserverForName:(NSNotificationName)name object:(id)obj queue:(NSOperationQueue *)queue usingBlock:(void (^)(NSNotification *note))block;方法的头文件中的说明:

NSNotification.h

文档中指出,返回值被系统强引用,并且调用者需要持有该对象以用来移除通知。于是,我们修改一下代码,如下:

@interface SecondViewController ()
@property (nonatomic, strong) id observer; //持有注册通知后返回的对象
@end

@implementation SecondViewController

- (void)addObserver {
    void(^myBlock)(NSNotification * _Nonnull note) = ^(NSNotification * _Nonnull note) {
        NSLog(@"%s, %@", __FUNCTION__, self.title);
    };
    self.observer = [[NSNotificationCenter defaultCenter] addObserverForName:@"wangb" object:nil queue:[NSOperationQueue mainQueue] usingBlock:myBlock];
    NSLog(@"myBlock:%@", myBlock);  //打印myBlock的地址
}

- (void)postNotification {
    [[NSNotificationCenter defaultCenter] postNotificationName:@"wangb" object:nil];
    NSLog(@"%s, %@", __FUNCTION__, self.title);
}

- (void)removeObserver {
    [[NSNotificationCenter defaultCenter] removeObserver:self.observer name:@"wangb" object:nil];
    self.observer = nil;
}

- (void)dealloc {
    NSLog(@"%s, %@", __FUNCTION__, self.title);
}
@end

//调用代码
    SecondViewController *VC1 = [[SecondViewController alloc] init];
    VC1.title = @"VC1";
    [VC1 addObserver];
    
    SecondViewController *VC2 = [[SecondViewController alloc] init];
    VC2.title = @"VC2";
    [VC2 postNotification];

    [VC1 removeObserver];

运行如下:

通过调试发现,addObserver后返回的对象是一个__NSObserver类型,类中存储着通知的一些相关信息:

  • observer对象中存储的nc,即[NSNotificationCenter defaultCenter]
  • observer对象中存储的block,即addObserver时传入的block

这个__NSObserver特别像AFN里面的AFURLSessionManagerTaskDelegate,通过__NSObserver管理通知信息及回调接口。NSNotificationCenter中存储observer。所以,从调用addObserver,到断点的位置,运行流程及对象之间的关系猜测如下:

__NSObserver<--> NSNotificationCenter<-->self

图解:

  1. 调用[[NSNotificationCenter defaultCenter] addObserverForName:xxx之后,创建__NSObserver对象,并赋值:
__NSObserver *observer = [__NSObserver alloc] init];
observer->nc = [NSNotificationCenter defaultCenter];  //见图中①
observer->block = myBlock;  //见图中②
  1. NSNotificationCenter保存observer对象,用于通知分发,见图中③
  2. NSNotificationCenter把observer返回,被self持有,见图中④
  3. 当postNotification后,NSNotificationCenter通过observer找到相关信息,通过block回调,block中调用self。
  4. 到此为止,表示了运行到断点的过程。

下面,再回过头来看最开始的例子,self在没有持有NSNotificationCenter的情况下,只是在block中单方面调用了self,就导致了VC1不能释放,原因是由于observer单方面一直强引用VC1且observer没有释放,才导致的VC1没有释放。

那么,怎样才是正确使用NSNotificationCenter的姿势呢?

方法一:切断上图中的⑤和③:

正确使用NSNotificationCenter的姿势

方法二:如果在block中不是用weakSelf,那么必须要先销毁observer,才能解除对self的强引用。需要注意的是,这种情况下,不要尝试在dealloc方法中做任何操作,因为在observer解除对self的强引用前,self是释放不了的,所以都不会调用到dealloc方法。

方法二中,如果observer属性用weak修饰,则在removeObserver中可以不用写self.observer = nil;
在开发中,可以按照方法一来就可以了,最常规,最靠谱!

推荐阅读更多精彩内容