1. 程式人生 > >__weak和__block探究與區別

__weak和__block探究與區別

準備工作

首先我定義了一個類 MyObject 繼承 NSObject,並添加了一個屬性 text,重寫了description方法,返回 text 的值。這個主要是因為編譯器本身對 NSString 是有優化的,建立的 string 物件有可能是靜態儲存區永不釋放的,為了避免使用 NSString 引起一些問題,還是建立一個 NSObject 物件比較合適。

另外我自定義了一個 TLog 方法輸出物件相關值,定義如下:

#define TLog(prefix,Obj) {NSLog(@"變數記憶體地址:%p, 變數值:%p, 指向物件值:%@, --> %@",&Obj,Obj,Obj,prefix);}

__weak


我們測試下面一段程式碼

MyObject *obj = [[MyObject alloc]init];
obj.text = @"111111111111";
TLog(@"obj", obj);

__weak MyObject *weakObj = obj;
TLog(@"weakObj", weakObj);

void(^testBlock)() = ^(){
    TLog(@"weakObj - block", weakObj);
};
testBlock();
obj = nil;
testBlock();


輸出:

變數記憶體地址:0x7fff58c8a9f0, 變數值:0x7f8e0307f1d0, 指向物件值:111111111111, --> obj
變數記憶體地址:0x7fff58c8a9e8, 變數值:0x7f8e0307f1d0, 指向物件值:111111111111, --> weakObj 變數記憶體地址:0x7f8e030804c0, 變數值:0x7f8e0307f1d0, 指向物件值:111111111111, --> weakObj - block 變數記憶體地址:0x7f8e030804c0, 變數值:0x0, 指向物件值:(null), --> weakObj - block

從上面的結果可以看到

  • block 內的 weakObj 和外部的 weakObj 並不是同一個變數
  • block 捕獲了 weakObj 同時也是對 obj 進行了弱引用,當我在 block 外把 obj 釋放了之後,block 內也讀不到這個變量了
  • 當 obj 賦值 nil 時,block 內部的 weakObj 也為 nil 了,也就是說 obj 實際上是被釋放了,可見 __weak 是可以避免迴圈引用問題的

接下來我們再看第二段程式碼

MyObject *obj = [[MyObject alloc]init];
obj.text = @"111111111111";
TLog(@"obj", obj);

__weak MyObject *weakObj = obj;
TLog(@"weakObj-0", weakObj);
dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
    __strong MyObject *strongObj = weakObj;
    sleep(3);
    TLog(@"weakObj - block", weakObj);
    TLog(@"strongObj - block", strongObj);
});
sleep(1);
obj = nil;
TLog(@"weakObj-1", weakObj);
sleep(4);
TLog(@"weakObj-2", weakObj);

輸出

變數記憶體地址:0x7fff572a9ab8, 變數值:0x7f9562f76210, 指向物件值:111111111111, --> obj
變數記憶體地址:0x7fff572a9ab0, 變數值:0x7f9562f76210, 指向物件值:111111111111, --> weakObj-0
變數記憶體地址:0x7fff572a9ab0, 變數值:0x7f9562f76210, 指向物件值:111111111111, --> weakObj-1
變數記憶體地址:0x7f9562d143a0, 變數值:0x7f9562f76210, 指向物件值:111111111111, --> weakObj - block
變數記憶體地址:0x116ccbe08, 變數值:0x7f9562f76210, 指向物件值:111111111111, --> strongObj - block
變數記憶體地址:0x7fff572a9ab0, 變數值:0x0, 指向物件值:(null), --> weakObj-2

如果你看過 AFNetworking 的原始碼,會發現 AFN 中作者會把變數在 block 外面先用 __weak 宣告,在 block 內把前面 weak 宣告的變數賦值給 __strong 修飾的變數。這種寫法的好處就是可以讓變數在 block 內部安全可用,即使外部釋放了,也會在 block 的生命週期內保留該變數。這種寫法非常巧妙,既避免了迴圈引用的問題,又可以在 block 內部持有該變數。

__block

先上程式碼

MyObject *obj = [[MyObject alloc]init];
obj.text = @"11111111111111";
TLog(@"obj",obj);

__block MyObject *blockObj = obj;
obj = nil;
TLog(@"blockObj -1",blockObj);

void(^testBlock)() = ^(){
    TLog(@"blockObj - block",blockObj);
    MyObject *obj2 = [[MyObject alloc]init];
    obj2.text = @"222222222222";
    TLog(@"obj2",obj2);
    blockObj = obj2;
    TLog(@"blockObj - block",blockObj);
};
NSLog(@"%@",testBlock);
TLog(@"blockObj -2",blockObj);
testBlock();
TLog(@"blockObj -3",blockObj);

結果

變數記憶體地址:0x7fff5021a9f0, 變數值:0x7ff6b48d8cd0, 指向物件值:11111111111111, --> obj
變數記憶體地址:0x7fff5021a9e8, 變數值:0x7ff6b48d8cd0, 指向物件值:11111111111111, --> blockObj -1
<__NSMallocBlock__: 0x7ff6b48d8c20>
變數記憶體地址:0x7ff6b48da518, 變數值:0x7ff6b48d8cd0, 指向物件值:11111111111111, --> blockObj -2
變數記憶體地址:0x7ff6b48da518, 變數值:0x7ff6b48d8cd0, 指向物件值:11111111111111, --> blockObj - block
變數記憶體地址:0x7fff5021a7f8, 變數值:0x7ff6b48d9960, 指向物件值:222222222222, --> obj2
變數記憶體地址:0x7ff6b48da518, 變數值:0x7ff6b48d9960, 指向物件值:222222222222, --> blockObj - block
變數記憶體地址:0x7ff6b48da518, 變數值:0x7ff6b48d9960, 指向物件值:222222222222, --> blockObj -3

我對 __block 的理解是其實際上是把變數的作用域給改變了,應該是提升了變數的作用域,使得在 block 內部和外部所訪問的是同一個變數。類似於宣告一個 static 和 global 變數的意思。

之所以程式碼中列印了一個 testBlock 的型別,是因為可以看到在 testBlock 宣告前,blockObj 的地址和聲明後的地址是有變化的,也就是說 blockObj 應該是在 block 內部提升的作用域。

再來看看 __block 能不能避免迴圈引用的問題

MyObject *obj = [[MyObject alloc]init];
obj.text = @"11111111111111";
TLog(@"obj",obj);

__block MyObject *blockObj = obj;
obj = nil;
void(^testBlock)() = ^(){
    TLog(@"blockObj - block",blockObj);
};
obj = nil;
testBlock();
TLog(@"blockObj",blockObj);

輸出

變數記憶體地址:0x7fff57eef9f0, 變數值:0x7ff86a55a160, 指向物件值:11111111111111, --> obj
變數記憶體地址:0x7ff86c918a88, 變數值:0x7ff86a55a160, 指向物件值:11111111111111, --> blockObj - block
變數記憶體地址:0x7ff86c918a88, 變數值:0x7ff86a55a160, 指向物件值:11111111111111, --> blockObj

當外部 obj 指向 nil 的時候,obj 理應被釋放,但實際上 blockObj 依然強引用著 obj,obj 其實並沒有被真正釋放。因此使用 __block 並不能避免迴圈引用的問題。

但是我們可以通過手動釋放 blockObj 的方式來釋放 obj,這就需要我們在 block 內部將要退出的時候手動釋放掉 blockObj ,如下這種形式

MyObject *obj = [[MyObject alloc]init];
obj.text = @"11111111111111";
TLog(@"obj",obj);

__block MyObject *blockObj = obj;
obj = nil;
void(^testBlock)() = ^(){
    TLog(@"blockObj - block",blockObj);
    blockObj = nil;
};
obj = nil;
testBlock();
TLog(@"blockObj",blockObj);

這種形式既能保證在 block 內部能夠訪問到 obj,又可以避免迴圈引用的問題,但是這種方法也不是完美的,其存在下面幾個問題

  • 必須記住在 block 底部釋放掉 block 變數,這其實跟 MRC 的形式有些類似了,不太適合 ARC
  • 當在 block 外部修改了 blockObj 時,block 內部的值也會改變,反之在 block 內部修改 blockObj 在外部再使用時值也會改變。這就需要在寫程式碼時注意這個特性可能會帶來的一些隱患
  • __block 其實提升了變數的作用域,在 block 內外訪問的都是同一個 blockObj 可能會造成一些隱患

總結

__weak 本身是可以避免迴圈引用的問題的,但是其會導致外部物件釋放了之後,block 內部也訪問不到這個物件的問題,我們可以通過在 block 內部宣告一個 __strong 的變數來指向 weakObj,使外部物件既能在 block 內部保持住,又能避免迴圈引用的問題

__block 本身無法避免迴圈引用的問題,但是我們可以通過在 block 內部手動把 blockObj 賦值為 nil 的方式來避免迴圈引用的問題。另外一點就是 __block 修飾的變數在 block 內外都是唯一的,要注意這個特性可能帶來的隱患。