Block循环引用详解
阅读原文时间:2023年07月13日阅读:1

在项目中经常用到block,使用不当就很容易因为循环引用而造成内存泄漏。本文分析了block循环引用形成原因以及处理办法,如果有什么不对或者疑问请留言。

block为了保证代码块内部对象不被提前释放,会对block中的对象进行强引用,就相当于持有了其中的对象,而如果此时block中的对象又持有了该block,就会造成循环引用。

1.所有block都会造成循环引用

在block中,并不是所有的block都会循造成环引用,比如UIView动画block、Masonry添加约束block、AFN网络请求回调block等。UIView动画block不会造成循环引用是因为这是类方法,不可能强引用一个类,所以不会造成循环引用。Masonry约束block不会造成循环引用是因为Masonry中设置布局的方法中的block对象并没有被外部View所引用,超出当前作用域后,block会被释放,所以我们使用Masonry的时候不需要担心循环引用。

  • Masonry内部代码

    • (NSArray *)mas_makeConstraints:(void(^)(MASConstraintMaker *))block {
          self.translatesAutoresizingMaskIntoConstraints = NO;
          MASConstraintMaker *constraintMaker = [[MASConstraintMaker alloc] initWithView:self];
      //这里并不是self.block
          block(constraintMaker);
          return [constraintMaker install];
      }

AFN请求回调block不会造成循环引用是因为在内部做了处理。block先是被AFURLSessionManagerTaskDelegate对象持有。而AFURLSessionManagerTaskDelegate对象被mutableTaskDelegatesKeyedByTaskIdentifier字典持有,在block执行完成后,mutableTaskDelegatesKeyedByTaskIdentifier字典会移除AFURLSessionManagerTaskDelegate对象,这样对象就被释放了,所以不会造成循环引用。

  • AFN内部代码

    #pragma mark - 添加代理

    • (void)addDelegateForDataTask:(NSURLSessionDataTask *)dataTask
                      uploadProgress:(nullable void (^)(NSProgress *uploadProgress)) uploadProgressBlock
                    downloadProgress:(nullable void (^)(NSProgress *downloadProgress)) downloadProgressBlock
                   completionHandler:(void (^)(NSURLResponse *response, id responseObject, NSError *error))completionHandler
      {
          AFURLSessionManagerTaskDelegate *delegate = [[AFURLSessionManagerTaskDelegate alloc] init];
          delegate.manager = self;
           //block被代理引用
          delegate.completionHandler = completionHandler;
          dataTask.taskDescription = self.taskDescriptionForSessionTasks;
          //设置代理
          [self setDelegate:delegate forTask:dataTask];

        delegate.uploadProgressBlock = uploadProgressBlock;
        delegate.downloadProgressBlock = downloadProgressBlock;
    }
    #pragma mark - 设置代理

    • (void)setDelegate:(AFURLSessionManagerTaskDelegate *)delegate
                  forTask:(NSURLSessionTask *)task
      {
          NSParameterAssert(task);
          NSParameterAssert(delegate);

        [self.lock lock];
        //代理被mutableTaskDelegatesKeyedByTaskIdentifier字典引用
        self.mutableTaskDelegatesKeyedByTaskIdentifier[@(task.taskIdentifier)] = delegate;
        [delegate setupProgressForTask:task];
        [self addNotificationObserverForTask:task];
        [self.lock unlock];
    }
    #pragma mark - 任务完成

    • (void)URLSession:(NSURLSession *)session
                dataTask:(NSURLSessionDataTask *)dataTask
      didBecomeDownloadTask:(NSURLSessionDownloadTask *)downloadTask
      {
          AFURLSessionManagerTaskDelegate *delegate = [self delegateForTask:dataTask];
          if (delegate) {
              //任务完成,移除
              [self removeDelegateForTask:dataTask];
              [self setDelegate:delegate forTask:downloadTask];
          }

        if (self.dataTaskDidBecomeDownloadTask) {
            self.dataTaskDidBecomeDownloadTask(session, dataTask, downloadTask);
        }
    }
    #pragma mark - 移除任务代理

    • (void)removeDelegateForTask:(NSURLSessionTask *)task {
          NSParameterAssert(task);

        AFURLSessionManagerTaskDelegate *delegate = [self delegateForTask:task];
        [self.lock lock];
        [delegate cleanUpProgressForTask:task];
        [self removeNotificationObserverForTask:task];
        //移除
        [self.mutableTaskDelegatesKeyedByTaskIdentifier removeObjectForKey:@(task.taskIdentifier)];
        [self.lock unlock];
    }

2.block中只有self会造成循环引用

在block中并不只是self会造成循环引用,用下划线调用属性(如_name)也会出现循环引用,效果和使用self是一样的。

//会造成循环引用
_person1 = [[Person alloc] init];
_person2 = [[Person alloc] init];
_person2.name = @"张三";
    
_person1.block = ^{
    NSLog(@"%@",_person2.name)
};

3.weakSelf可以解决所有block造成的循环引用

大部分情况下,这样使用是可以解决block循环引用,但是有些情况下这样使用会造成一些问题,比如在block中延迟执行一些代码,在还没有执行的时候,控制器被销毁了,这样控制器中的对象也会被释放,__weak对象就会变成null。所以会输出null。

//在延迟执行期间,控制器被释放了,打印出来的会是**(null)**
_person1 = [[Person alloc] init];
_person2 = [[Person alloc] init];
_person2.name = @"张三";
__weak __typeof(self) weakSelf = self;
_person1.block = ^{
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(5 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        NSLog(@"%@",weakSelf.person2.name);
    });
};

4.用self调用带有block的方法会引起循环引用

并不是所有通过self调用带有block的方法会引起循环引用,需要看方法内部有没有持有self。

//不会引起循环引用
[self dismissViewControllerAnimated:YES completion:^{
    NSLog(@"%@",self.string);
}];

5.类方法不会造成循环引用

这是一种错误的说法,类方法我们需要知道其内部是否持有对象,如果内部持有对象,并且在block执行后未释放持有的对象,那么也会造成循环引用。

1.weakSelf、strongSelf结合使用

使用weakSelf结合strongSelf的情况下,能够避免循环引用,也不会造成提前释放导致block内部代码无效。

_person1 = [[Person alloc] init];
_person2 = [[Person alloc] init];
_person2.name = @"张三";
__weak __typeof(self) weakSelf = self;
_person1.block = ^{
    __typeof(&*weakSelf) strongSelf = weakSelf;
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(5 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        NSLog(@"%@",strongSelf.person2.name);
    });
};

2.block的外部对象使用week

外部对象通过week修饰,使用全局弱指针指向一个局部强引用对象,这样局部变量在超出其作用域后也不会被销毁,因为是弱指针,所以不会造成循环引用。

@interface CLViewController ()
//弱引用指针
@property (nonatomic,weak) Person *person1;
@property (nonatomic,strong) Person *person2;

@end

@implementation CLViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    self.view.backgroundColor = [UIColor redColor];
    //局部强引用对象
    Person *person1 = [[Person alloc] init];
    _person1 = person1;
    
    
    _person2 = [[Person alloc] init];
    _person2.name = @"张三";
    _person1.block = ^{
            NSLog(@"%@",self.person2.name);
    };

}

3.将对象置为nil

使用完对象之后就没有必要再保留该对象了,将对象置为nil。在ARC中,被置为nil的对象会被销毁。虽然这样也可以达到破除循环引用的效果,但是这样使用起来很不方便,如果一个对象有多个block,在前面将对象置空,后面的block就不会执行,所以使用这种方法来防止循环引用,需要注意在合适的位置将对象置空。

_person1 = [[Person alloc] init];
_person2 = [[Person alloc] init];
_person2.name = @"张三";
_person1.block = ^{
    NSLog(@"%@",self.person2.name);
    //置空,避免循环引用
    _person1 = nil;
};
//由于上面已经将对象置空,所以这里block里边的代码不会执行
_person1.block = ^{
    NSLog(@"%@",self.person2.name);
};

虽然这种方式使用起来不是很友好,但是在封装block的时候,可以考虑使用完马上置空当前使用的block,这样使用的时候就不需要考虑循环引用的问题。

- (void)back
{
    if (self.BackBlock)
    {
        self.BackBlock(button);
    }
//使用完,马上置空当前block
    self.BackBlock = nil;
}

循环引用的本质是当前对象对这个block进行了强持有,并且block内部对这个对象进行了强持有。使用block的时候,我们首先需要做的就是判断当前block是否会引起循环引用,如果会引起循环引用,再考虑采取哪种方式来避免循环引用。