寫時復制

寫時復制

在swift中忧便,像Array队他、Dictionary捏境、Set等集合類型都是通過寫時復制(copy-on-write)技術實現的。

例如一個整形數組

let x = [1,2,3,4]
var y = x

創(chuàng)建新變量 y窖壕,把 x 賦值給 y 會發(fā)生復制忧勿,現在x和y都是獨立的結構體。然后這些Array 結構體含有指向某個內存的引用瞻讽。這個內存就是數組元素存儲在堆上的位置鸳吸。我們可以通過 swift Array 的

@inlinable public func withUnsafeBufferPointer<R>(_ body: (UnsafeBufferPointer<Element>) throws -> R) rethrows -> R

來獲取 Array 元素存儲的地址指針,我們可以看出數組x和y存儲地址和元素確實都是相同的

let xPoints1 = x.withUnsafeBufferPointer{$0}
let yPoints1 = y.withUnsafeBufferPointer{$0}

print(xPoints1, x)  // UnsafeBufferPointer(start: 0x000060000143b1e0, count: 4) [1, 2, 3, 4]
print(yPoints1, y)  // UnsafeBufferPointer(start: 0x000060000143b1e0, count: 4) [1, 2, 3, 4]

然后我們對數組 y 進行操作:

let xPoints2 = x.withUnsafeBufferPointer{$0}
let yPoints2 = y.withUnsafeBufferPointer{$0}

y.append(5)
let xPoints2 = x.withUnsafeBufferPointer{$0}
let yPoints2 = y.withUnsafeBufferPointer{$0}
print(xPoints2, x) // UnsafeBufferPointer(start: 0x000060000143b1e0, count: 4) [1, 2, 3, 4]
print(yPoints2, y) // UnsafeBufferPointer(start: 0x0000600002516a80, count: 5) [1, 2, 3, 4, 5]

通過以上代碼可以看出速勇,當y引用的存儲要改變時晌砾,存儲進行了復制。這就是所謂的"寫時復制"烦磁。

但是作為一個結構體的作者养匈,你不能免費獲得這種特性哼勇,你需要自己進行實現。當結構類型內部含有一個或者多個可變引用呕乎,同時想要保持值語義猴蹂,并且不必要的復制,為你的類型實現寫時復制是有意義的楣嘁。

接下來磅轻,我們用 NSMutableArray 作為內部引用類型來實現 swift 中的值類型的 Array 結構體。

寫實復制(昂貴模式)

我們首先將 _array 標記為結構體的私有屬性逐虚。我們不再直接變更 _array聋溜,而
是通過一個計算屬性 _arrayForWriting 來訪問它。這個計算屬性總是會復制 _array 并將其返回:

struct MyArray {
    fileprivate var _array: NSMutableArray
    fileprivate var _arrayForWriting: NSMutableArray {
        mutating get {
            _array = _array.mutableCopy() as! NSMutableArray
            return _array
        }
    }
    
    init(_ array: NSMutableArray) {
        self._array = array.mutableCopy() as! NSMutableArray
    }
    
    mutating func add(_ other: MyArray) {
        _arrayForWriting.addObjects(from: other._array as Array)
    }
    
    mutating func append(_ any: Any) {
        _arrayForWriting.add(any)
    }
    
    var description: String {
        return (_array as Array).description
    }
    
}

我們通過get方法操作時叭爱,每次均復制 _array撮躁,這個結構體具有值語義了。但這樣的做法雖然有效买雾,但是當我們多次改變同一個變量時把曼,效率很低,因為每次操作漓穿,都會通過 get 方法去復制 _array得到 _arrayForWriting嗤军。比如如下操作:

var array = MyArray(NSArray(array: [1, 2, 3, 4]))

for index in 5...9 {
    array.append(index)
}

每次 append 都會需要調用 get 方法去復制 _array 。如果我們能讓 _array 在沒有被共享之前晃危,對它進行原地變更就高效了叙赚。

寫時復制(高效方式)

上述問題說道 _array 沒有被共享之前,那就是我們要在它是唯一引用的時候進行原地修改僚饭,如果有別的變量也在強引用它震叮,那么我們就需要復制 _array 后去再修改它了。在 Swift 中鳍鸵,提供了 isKnownUniquelyReferenced 函數來檢查引用的唯一性苇瓣,沒有其他強應用將返回 true。由于對于 Objective-C 的類偿乖,它會直 接返回 false击罪,所以我們可以創(chuàng)建一個 Swift 類,來封裝 Objective-C 對象到 Swift類中汹想。

final class Box<T> {
    var uniqueValue: T
    init(_ value: T) {
        self.uniqueValue = value
    }
}

現在我們改造之前寫的 MyArray

 struct MyArray {

    private var _data: Box<NSMutableArray>
    private var _dataForWriting: NSMutableArray {
        mutating get {

            if !isKnownUniquelyReferenced(&_data) {
                _data = Box(_data.uniqueValue.mutableCopy() as! NSMutableArray)
                print("Making a copy")
            }
            return _data.uniqueValue
        }
    }

    init(_ array: NSArray) {
        self._data = Box(array.mutableCopy() as! NSMutableArray)
    }

    mutating func append(_ other: MyArray) {
        _dataForWriting.addObjects(from: other._data.uniqueValue as Array)
    }

    mutating func append(_ any: Any) {
        _dataForWriting.add(any)
    }

    var description: String {
        return (_data.uniqueValue as Array).description
    }
}

現在我們可以測試一下:

static func testMyArray() {
    var x = MyArray(NSArray(array: [1, 2, 3, 4]))
    // 原地操作
    x.append(5)
    var y = x
    print(x.description, y.description)  // [1, 2, 3, 4, 5] [1, 2, 3, 4, 5]
        
    y.append(6)  // Making a copy
    print(x.description, y.description)  // [1, 2, 3, 4, 5] [1, 2, 3, 4, 5, 6]
}

原地操作不會發(fā)生復制外邓,而我們賦值給 y 后撤蚊,對應進行操作古掏,此時發(fā)生了復制。至此寫時復制已經完成

對于有的書上說:Array 和 Dictionary下標取值值元素有不同侦啸,通過下標操作其實一樣是會進行復制:

static func testTrap() {   
    let arrayForDict = MyArray(NSArray(array: [1, 2, 3, 4]))
        
    var dict: [String: MyArray] = ["s": arrayForDict]
    dict["s"]!.append(6)  // Making a copy
         
    let arrayForArray = MyArray(NSArray(array: [5, 6, 7, 8]))
    var array = [arrayForArray]
    array[0].append(9)  // Making a copy
}
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末槽唾,一起剝皮案震驚了整個濱河市丧枪,隨后出現的幾起案子,更是在濱河造成了極大的恐慌庞萍,老刑警劉巖拧烦,帶你破解...
    沈念sama閱讀 218,525評論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現場離奇詭異钝计,居然都是意外死亡恋博,警方通過查閱死者的電腦和手機,發(fā)現死者居然都...
    沈念sama閱讀 93,203評論 3 395
  • 文/潘曉璐 我一進店門私恬,熙熙樓的掌柜王于貴愁眉苦臉地迎上來债沮,“玉大人,你說我怎么就攤上這事本鸣∫唏茫” “怎么了?”我有些...
    開封第一講書人閱讀 164,862評論 0 354
  • 文/不壞的土叔 我叫張陵荣德,是天一觀的道長闷煤。 經常有香客問我,道長涮瞻,這世上最難降的妖魔是什么鲤拿? 我笑而不...
    開封第一講書人閱讀 58,728評論 1 294
  • 正文 為了忘掉前任,我火速辦了婚禮署咽,結果婚禮上皆愉,老公的妹妹穿的比我還像新娘。我一直安慰自己艇抠,他們只是感情好幕庐,可當我...
    茶點故事閱讀 67,743評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著家淤,像睡著了一般异剥。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上絮重,一...
    開封第一講書人閱讀 51,590評論 1 305
  • 那天冤寿,我揣著相機與錄音,去河邊找鬼青伤。 笑死督怜,一個胖子當著我的面吹牛,可吹牛的內容都是我干的狠角。 我是一名探鬼主播号杠,決...
    沈念sama閱讀 40,330評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了姨蟋?” 一聲冷哼從身側響起屉凯,我...
    開封第一講書人閱讀 39,244評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎眼溶,沒想到半個月后悠砚,有當地人在樹林里發(fā)現了一具尸體,經...
    沈念sama閱讀 45,693評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡堂飞,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,885評論 3 336
  • 正文 我和宋清朗相戀三年灌旧,在試婚紗的時候發(fā)現自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片绰筛。...
    茶點故事閱讀 40,001評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡节榜,死狀恐怖,靈堂內的尸體忽然破棺而出别智,到底是詐尸還是另有隱情宗苍,我是刑警寧澤,帶...
    沈念sama閱讀 35,723評論 5 346
  • 正文 年R本政府宣布薄榛,位于F島的核電站讳窟,受9級特大地震影響,放射性物質發(fā)生泄漏敞恋。R本人自食惡果不足惜丽啡,卻給世界環(huán)境...
    茶點故事閱讀 41,343評論 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望硬猫。 院中可真熱鬧补箍,春花似錦、人聲如沸啸蜜。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,919評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽宜咒。三九已至,卻和暖如春恬吕,著一層夾襖步出監(jiān)牢的瞬間蜂林,已是汗流浹背遥诉。 一陣腳步聲響...
    開封第一講書人閱讀 33,042評論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留噪叙,地道東北人矮锈。 一個月前我還...
    沈念sama閱讀 48,191評論 3 370
  • 正文 我出身青樓,卻偏偏與公主長得像睁蕾,于是被迫代替她去往敵國和親苞笨。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,955評論 2 355

推薦閱讀更多精彩內容