JavaScript深拷貝的一些坑

YDJFE發表於2018-06-15

前言

之前去一家公司面試的時候，面試官問了我一個問題，說:"如何才能深拷貝一個物件"。當時我心裡有些竊喜，這麼簡單的問題還用想嗎？於是脫口而出:"平時常用的有兩種辦法，第一種用JSON.parse(JSON.stringify(obj))，第二種可以使用for...in加遞迴完成"。面試官聽了以後點了點頭覺得挺滿意的。當時我也並沒有太過在乎這個問題，直到前段時間又想起這個問題，發現上面說的兩種方法都是有Bug的。

提出問題

那麼上面所說的Bug是什麼呢？

特殊物件拷貝

首先讓我們試想有這麼一個物件，在不考慮普通型別的情況下，它有如下成員:


const obj = {
    arr: [111, 222],
    obj: {key: '物件'},
    a: () => {console.log('函式')},
    date: new Date(),
    reg: /正則/ig
}

複製程式碼

然後我們用上面兩種方式分別拷貝一次

JSON法


JSON.parse(JSON.stringify(obj))

複製程式碼

輸出結果:

可以從中看出，obj中的普通物件和陣列都能拷貝，然而date物件成了字串，函式直接就不見了，正則成了一個空物件。

再來看看for...in加遞迴的方法

遞迴


function isObj(obj) {
    return (typeof obj === 'object' || typeof obj === 'function') && obj !== null
}
function deepCopy(obj) {
    let tempObj = Array.isArray(obj) ? [] : {}
    for(let key in obj) {
        tempObj[key] = isObj(obj[key]) ? deepCopy(obj[key]) : obj[key]
    }
    return tempObj
}

複製程式碼

結果:

結論

通過上面的測試可知，這兩個方法都無法拷貝函式，date，reg型別的物件;

什麼是環?

環就是物件迴圈引用，導致自己成為一個閉環，例如下面這個物件:


var a = {}

a.a = a

複製程式碼

使用上面兩個方法拷貝一下會直接報錯

解決方案

可以使用一個WeakMap結構儲存已經被拷貝的物件，每一次進行拷貝的時候就先向WeakMap查詢該物件是否已經被拷貝，如果已經被拷貝則取出該物件並返回，將deepCopy函式改造成如下


function deepCopy(obj, hash = new WeakMap()) {
    if(hash.has(obj)) return hash.get(obj)
    let cloneObj = Array.isArray(obj) ? [] : {}
    hash.set(obj, cloneObj)
    for (let key in obj) {
        cloneObj[key] = isObj(obj[key]) ? deepCopy(obj[key], hash) : obj[key];
    }
    return cloneObj
}

複製程式碼

拷貝環結果:

特殊物件的拷貝

這個問題的解決比較麻煩，因為需要特別對待的物件種類實在太多，於是我參考了MDN上的結構化拷貝，然後結合解決環的方案：


// 只解決date，reg型別，其他的可以自己新增

function deepCopy(obj, hash = new WeakMap()) {
    let cloneObj
    let Constructor = obj.constructor
    switch(Constructor){
        case RegExp:
            cloneObj = new Constructor(obj)
            break
        case Date:
            cloneObj = new Constructor(obj.getTime())
            break
        default:
            if(hash.has(obj)) return hash.get(obj)
            cloneObj = new Constructor()
            hash.set(obj, cloneObj)
    }
    for (let key in obj) {
        cloneObj[key] = isObj(obj[key]) ? deepCopy(obj[key], hash) : obj[key];
    }
    return cloneObj
}

複製程式碼