weak實現(xiàn)原理

概括

runtime維護(hù)了一個weak表替蔬,用于存儲指向某個對象的所有weak指針玫恳。weak表是一個hash表傍菇,key是指向?qū)ο蟮牡刂肪芗琕alue是Weak指針的地址的數(shù)組秉馏,如下圖所示亿眠,key是0xffff贰逾,value數(shù)組中的某一項是0x1007


weak

所用方法

  • objc_initWeak(id *location, id newObj)
  • objc_storeWeak(id *location, id newObj)
  • objc_destroyWeak(id *location)
    其中init和destroy都會調(diào)用storeWeak荐吵,只不過傳遞的參數(shù)不同错英∪肴觯總體說來,通過initWeak函數(shù)初始化“weak修飾的屬性(location)”椭岩,weak修飾的屬性所指對象(newObj)銷毀后通過destoryWeak函數(shù)釋放該變量(location)茅逮。

objc_initWeak函數(shù)

初始化時:runtime會調(diào)用objc_initWeak函數(shù),objc_initWeak函數(shù)會初始化一個新的weak指針指向?qū)ο蟮牡刂贰?/p>

NSObject *obj = [[NSObject alloc] init];
id __weak obj1 = obj;

我們初始化一個weak變量時判哥,runtime會調(diào)用NSObject.mm中的objc_initWeak函數(shù)献雅,這個函數(shù)在源碼中聲明如下:

id _Nullable objc_initWeak(id _Nullable * _Nonnull location, id _Nullable val)

具體實現(xiàn)代碼為:

id objc_initWeak(id *location, id newObj)
{
    //如果新創(chuàng)建的對象無效,則直接返回nil
    if (!newObj) {
        *location = nil;
        return nil;
    }
    //新建對象有效塌计,則調(diào)用storeWeak方法進(jìn)行后續(xù)操作--添加引用
    return storeWeak<DontHaveOld, DoHaveNew, DoCrashIfDeallocating>
        (location, (objc_object*)newObj);
}

綜上所述:在將“附有weak修飾符的變量(location)”初始化為0(nil)后挺身,會將“賦值對象”(obj)作為參數(shù),調(diào)用objc_storeWeak函數(shù)锌仅。
也就是說章钾,weak 修飾的指針默認(rèn)值是 nil (在Objective-C中向nil發(fā)送消息是安全的)

objc_destroyWeak函數(shù)

當(dāng)weak屬性所指向?qū)ο螅╪ewObj)被銷毀后墙贱,obj_destroyWeak函數(shù)將0(nil)作為參數(shù),調(diào)用objc_storeWeak函數(shù)贱傀。
objc_storeWeak(&location, 0);
objc_storeWeak函數(shù)的第二個參數(shù)(newObj)如果為0(nil)惨撇,那么把weak修飾的屬性(location)的地址從weak表中刪除。

objc_storeWeak函數(shù)

weak實現(xiàn)原理的核心就是objc_storeWeak函數(shù)府寒,函數(shù)實現(xiàn)如下:

id objc_storeWeak(id *location, id newObj)
{
    // 這里傳遞了三個 bool 數(shù)值
    // 使用 template 進(jìn)行常量參數(shù)傳遞是為了優(yōu)化性能
    return storeWeak<DoHaveOld, DoHaveNew, DoCrashIfDeallocating>
        (location, (objc_object *)newObj);
}

看起來真正起作用的函數(shù)是storeWeak魁衙,

// HaveOld:     true - 變量有值
//             false - 需要被及時清理,當(dāng)前值可能為 nil
// HaveNew:     true - 需要被分配的新值株搔,當(dāng)前值可能為 nil
//             false - 不需要分配新值
// CrashIfDeallocating: true - 說明 newObj 已經(jīng)釋放或者 newObj 不支持弱引用纺棺,該過程需要暫停
//             false - 用 nil 替代存儲
template bool HaveOld, bool HaveNew, bool CrashIfDeallocating>
static id storeWeak(id *location, objc_object *newObj) {
    // 該過程用來更新弱引用指針的指向
    // 初始化 previouslyInitializedClass 指針
    Class previouslyInitializedClass = nil;
    id oldObj;
    // 聲明兩個 SideTable
    // ① 新舊散列創(chuàng)建
    SideTable *oldTable;
    SideTable *newTable;
    // 獲得新值和舊值的鎖存位置(用地址作為唯一標(biāo)示)
    // 通過地址來建立索引標(biāo)志,防止桶重復(fù)
    // 下面指向的操作會改變舊值
retry:
    if (HaveOld) {
        // 更改指針邪狞,獲得以 oldObj 為索引所存儲的值地址
        oldObj = *location;
        oldTable = &SideTables()[oldObj];
    } else {
        oldTable = nil;
    }
    if (HaveNew) {
        // 更改新值指針祷蝌,獲得以 newObj 為索引所存儲的值地址
        newTable = &SideTables()[newObj];
    } else {
        newTable = nil;
    }
    // 加鎖操作,防止多線程中競爭沖突
    SideTable::lockTwoHaveOld, HaveNew>(oldTable, newTable);
    // 避免線程沖突重處理
    // location 應(yīng)該與 oldObj 保持一致帆卓,如果不同巨朦,說明當(dāng)前的 location 已經(jīng)處理過 oldObj 可是又被其他線程所修改
    if (HaveOld  &&  *location != oldObj) {
        SideTable::unlockTwoHaveOld, HaveNew>(oldTable, newTable);
        goto retry;
    }
    // 防止弱引用間死鎖
    // 并且通過 +initialize 初始化構(gòu)造器保證所有弱引用的 isa 非空指向
    if (HaveNew  &&  newObj) {
        // 獲得新對象的 isa 指針
        Class cls = newObj->getIsa();
        // 判斷 isa 非空且已經(jīng)初始化
        if (cls != previouslyInitializedClass  &&
            !((objc_class *)cls)->isInitialized()) {
            // 解鎖
            SideTable::unlockTwoHaveOld, HaveNew>(oldTable, newTable);
            // 對其 isa 指針進(jìn)行初始化
            _class_initialize(_class_getNonMetaClass(cls, (id)newObj));
            // 如果該類已經(jīng)完成執(zhí)行 +initialize 方法是最理想情況
            // 如果該類 +initialize 在線程中
            // 例如 +initialize 正在調(diào)用 storeWeak 方法
            // 需要手動對其增加保護(hù)策略,并設(shè)置 previouslyInitializedClass 指針進(jìn)行標(biāo)記
            previouslyInitializedClass = cls;
            // 重新嘗試
            goto retry;
        }
    }
    // ② 清除舊值
    if (HaveOld) {
        weak_unregister_no_lock(&oldTable->weak_table, oldObj, location);
    }
    // ③ 分配新值
    if (HaveNew) {
        newObj = (objc_object *)weak_register_no_lock(&newTable->weak_table,
                                                      (id)newObj, location,
                                                      CrashIfDeallocating);
        // 如果弱引用被釋放 weak_register_no_lock 方法返回 nil
        // 在引用計數(shù)表中設(shè)置若引用標(biāo)記位
        if (newObj  &&  !newObj->isTaggedPointer()) {
            // 弱引用位初始化操作
            // 引用計數(shù)那張散列表的weak引用對象的引用計數(shù)中標(biāo)識為weak引用
            newObj->setWeaklyReferenced_nolock();
        }
        // 之前不要設(shè)置 location 對象剑令,這里需要更改指針指向
        *location = (id)newObj;
    }
    else {
        // 沒有新值糊啡,則無需更改
    }
    SideTable::unlockTwoHaveOld, HaveNew>(oldTable, newTable);
    return (id)newObj;
}
  • 如果是在objc_initWeak函數(shù)中調(diào)用objc_storeWeak函數(shù),則其作用是更新指針指向吁津,創(chuàng)建對應(yīng)的弱引用表棚蓄。
  • 如果是在objc_destroyWeak函數(shù)中調(diào)用objc_storeWeak函數(shù),則其作用是在對象被釋放的時候碍脏,根據(jù)對象地址獲取所有weak指針地址的數(shù)組梭依,然后遍歷這個數(shù)組把其中的數(shù)據(jù)設(shè)為nil,最后把這個entry從weak表中刪除典尾,最后清理對象的記錄役拴。

Reference

iOS 底層解析weak的實現(xiàn)原理

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市钾埂,隨后出現(xiàn)的幾起案子河闰,更是在濱河造成了極大的恐慌,老刑警劉巖褥紫,帶你破解...
    沈念sama閱讀 216,402評論 6 499
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件姜性,死亡現(xiàn)場離奇詭異,居然都是意外死亡髓考,警方通過查閱死者的電腦和手機(jī)部念,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,377評論 3 392
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人印机,你說我怎么就攤上這事矢腻。” “怎么了射赛?”我有些...
    開封第一講書人閱讀 162,483評論 0 353
  • 文/不壞的土叔 我叫張陵多柑,是天一觀的道長。 經(jīng)常有香客問我楣责,道長竣灌,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,165評論 1 292
  • 正文 為了忘掉前任秆麸,我火速辦了婚禮初嘹,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘沮趣。我一直安慰自己屯烦,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 67,176評論 6 388
  • 文/花漫 我一把揭開白布房铭。 她就那樣靜靜地躺著驻龟,像睡著了一般。 火紅的嫁衣襯著肌膚如雪缸匪。 梳的紋絲不亂的頭發(fā)上翁狐,一...
    開封第一講書人閱讀 51,146評論 1 297
  • 那天,我揣著相機(jī)與錄音凌蔬,去河邊找鬼露懒。 笑死,一個胖子當(dāng)著我的面吹牛砂心,可吹牛的內(nèi)容都是我干的懈词。 我是一名探鬼主播,決...
    沈念sama閱讀 40,032評論 3 417
  • 文/蒼蘭香墨 我猛地睜開眼计贰,長吁一口氣:“原來是場噩夢啊……” “哼钦睡!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起躁倒,我...
    開封第一講書人閱讀 38,896評論 0 274
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎洒琢,沒想到半個月后秧秉,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,311評論 1 310
  • 正文 獨居荒郊野嶺守林人離奇死亡衰抑,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,536評論 2 332
  • 正文 我和宋清朗相戀三年象迎,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 39,696評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡砾淌,死狀恐怖啦撮,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情汪厨,我是刑警寧澤赃春,帶...
    沈念sama閱讀 35,413評論 5 343
  • 正文 年R本政府宣布,位于F島的核電站劫乱,受9級特大地震影響织中,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜衷戈,卻給世界環(huán)境...
    茶點故事閱讀 41,008評論 3 325
  • 文/蒙蒙 一狭吼、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧殖妇,春花似錦刁笙、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,659評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至蔚润,卻和暖如春磅氨,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背嫡纠。 一陣腳步聲響...
    開封第一講書人閱讀 32,815評論 1 269
  • 我被黑心中介騙來泰國打工烦租, 沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人除盏。 一個月前我還...
    沈念sama閱讀 47,698評論 2 368
  • 正文 我出身青樓叉橱,卻偏偏與公主長得像,于是被迫代替她去往敵國和親者蠕。 傳聞我的和親對象是個殘疾皇子窃祝,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,592評論 2 353

推薦閱讀更多精彩內(nèi)容