Block的内存管理，看这里就够了

最近发现很多开发者对block的理解并不是很深，很多项目当中使用的时候多多少会有些问题，今天给大家详细讲讲block的内存管理, 主要从以下几个方面来讲：

• 根据内存划分block的类型
• block内存管理
• 防止循环引用

Block类型

根据Block在内存中的位置，系统把Block分为3类：NSGlobalBlock，NSStackBlock, NSMallocBlock;

• NSGlobalBlock：位于内存全局区
• NSStackBlock：位于内存栈区
• NSMallocBlock：位于内存堆区

我们通过block引用不同的变量来

全局区block(NSGlobalBlock)

没有引用局部变量的block叫做NSGlobalBlock，如下实例:

//类型1：没有使用任何外部变量
-(void)test
{
    void (^gBlock1)(int , int ) =^(int a, int b){
        NSLog(@"a + b = %d", a+b);
    };

    NSLog(@"%@", gBlock1);
    //打印结果为:
    //<__NSGlobalBlock__: 0x1025e8110>
}

//类型2：使用全局变量

//全局变量
int a = 10;

-(void)test
{
    void (^gBlock)() = ^(){
        NSLog(@"%d", a);
    };

    NSLog(@"%@", gBlock);
    //输出结果为：
    //<__NSGlobalBlock__: 0x103676110>
}

栈区block(NSStackBlock)

引用了局部变量的block叫做NSStackBlock， 实例如下：

-(void)test
{
    //局部变量
     NSArray *arr = @[@"zhangsan", @"lisi"];

    void (^sBlock)() = ^(){
        NSLog(@"arr = %@", arr);
    };

    NSLog(@"%@", sBlock);
    //输出结果为：
    //<__NSStackBlock__: 0x7fff5bbf1a58>
}

PS：栈区block在方法返回后就会被释放，所以只能在方法内部使用，如果将他赋值给其他对象或者存储起来，后面使用时将会出现错误.

堆区Block(NSMallocBlock)

在非ARC下，我们一般不手动创建NSMallocBlock，我们把从栈区复制(copy)过来的block称为堆区block。实例如下：

-(void)test
{

    NSArray *arr = @[@"zhangsan", @"lisi"];

    //栈区block
    void (^sBlock)() = ^(){
        NSLog(@"arr = %@", arr);
    };
    NSLog(@"%@", sBlock);

    //堆区block
    void (^mBlock)() = [sBlock copy];
    NSLog(@"%@", mBlock);

    //输出结果为：
    //<__NSStackBlock__: 0x7fff59bf9a38>
    //<__NSMallocBlock__: 0x7fc173f0dd80>

}

Block内存管理

对block自身内存的管理

对于block，有两个内存管理方法：Block_copy, Block_release;Block_copy与copy等效， Block_release与release等效；

不管是对block进行retian,copy,release,block的引用计数都不会增加，始终为1；

• NSGlobalBlock:使用retain,copy, release都无效，block依旧存在全局区，且没有释放, 使用copy和retian只是返回block的指针；

• NSStackBlock:使用retain,release操作无效；栈区block会在方法返回后将block空间回收； 使用copy将栈区block复制到堆区，可以长久保留block的空间，以供后面的程序使用；

• NSMallocBlock:支持retian,release，虽然block的引用计数始终为1，但内存中还是会对引用进行管理，使用retain引用+1， release引用-1； 对于NSMallocBlock使用copy之后不会产生新的block，只是增加了一次引用，类似于使用retian;

对引用变量的内存管理

在block中经常会用到外部变量/对象，如果这个block是存储在堆区，或者被复制到堆区，则对象对应的实例引用+1，当block释放后block的引用-1；

-(void)test
{

    NSArray *arr = @[@"zhangsan", @"lisi"];
    NSLog(@"arr.retianCount = %ld", arr.retainCount);

    //栈区block
    void (^sBlock)() = ^(){
        NSLog(@"arr = %@", arr);
    };
    //栈区block不会对引用的变量引用计数+1
    NSLog(@"arr.retianCount = %ld", arr.retainCount);


    //堆区block
    void (^mBlock)() = [sBlock copy];
    //复制到堆区后，引用计数+1
    NSLog(@"arr.retianCount = %ld", arr.retainCount);
}

循环引用

因为block中会对引用的对象进行持有(引用计数+1),从而导致相互持有引起循环引用；解决这种问题的方式是对引用变量使用修饰词__block或者__weak;

• __block：在非ARC中使用，NSMallocBlock类型的block不会对__block修饰的的变量引用计数+1，从而消除循环引用；在ARC中使用__block无效
• __weak：在ARC中使用，作用和__block一样，从而消除循环引用；在非ARC中不可以使用__weak;

防止循环引用案例：

//TestClass.h

@interface testClass : NSObject

@property (nonatomic, copy)void (^myBlock)(void);

@end

//TestClass.m

#define TestClassExample3 1

@implementation TestClass


-(void)dealloc
{
    NSLog(@"测试对象 被释放了。。。");

    [super dealloc];
}


-(instancetype)init
{
    self = [super init];
    if (self) {

#if TestClassExample1

        //会引起循环应用，当前对象无法被释放
        self.myBlock = ^(){
            //增加自己本身的引用计数
            [self doSomething];
        };

#elif TestClassExample2

        //在非ARC下有效，防止循环引用
        //在ARC下无效，会产生循环引用
        __block TestClass *weakSelf = self;
        self.myBlock = ^(){

            //在非ARC下不会增加self的引用计数
            [weakSelf doSomething];
        };

#elif TestClassExample3

        //在非ARC下无效，会产生循环引用
        //在ARC下有效，防止循环应用
        __weak TestClass *weakSelf = self;
        self.myBlock = ^(){

            //在非ARC下不会增加self的引用计数
            [weakSelf doSomething];
        };


#endif
    }

    return self;
}

-(void)doSomething
{
    NSLog(@"测试程序");
}


@end

//main.h

int main(int argc, char * argv[]) {
    @autoreleasepool {

        TestClass *tc = [[TestClass alloc] init];
        [tc release];
        tc = nil;
    }
}