JavaScript如何優(yōu)化邏輯判斷代碼詳解

更新時間：2021年06月08日 16:32:31 作者：編程三昧

我們在編寫 JS 代碼時,經(jīng)常會遇到邏輯判斷復(fù)雜的情況,這篇文章主要給大家介紹了關(guān)于JavaScript如何優(yōu)化邏輯判斷代碼的相關(guān)資料,需要的朋友可以參考下

前言

我們?nèi)粘Ｊ褂玫降倪壿嬇袛嗾Z句有 if...else...、switch...case...、do...while...等。

在簡單場景下，我們可能對這些語法的性能沒有什么感覺，但當(dāng)遇到復(fù)雜的業(yè)務(wù)場景時，如果處理不善，就會出現(xiàn)大量的邏輯嵌套，可讀性差并且難以擴(kuò)展。

千里之行始于足下，編寫高可維護(hù)性和高質(zhì)量的代碼，我們就需要從細(xì)節(jié)處入手，我們今天主要討論 JavaScript 中如何優(yōu)化邏輯判斷代碼。

嵌套層級優(yōu)化

function supply(fruit, quantity) {
    const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
    // 條件 1: 水果存在
    if (fruit) {
        // 條件 2: 屬于紅色水果
        if (redFruits.includes(fruit)) {
            console.log('紅色水果');
            // 條件 3: 水果數(shù)量大于 10 個
            if (quantity > 10) {
                console.log('數(shù)量大于 10 個');
            }
        }
    } else {
        throw new Error('沒有水果啦!');
    }
}

通過上面這個例子，我們可以看到：判斷流程中規(guī)中矩，符合現(xiàn)實(shí)世界的映射。但是，因代碼層層嵌套，導(dǎo)致閱讀和維護(hù)都存在困難。

如果傳入了 fruit 參數(shù)，則每次執(zhí)行都至少需要經(jīng)過兩步 if 判斷，在性能上也存在問題。

我們來對上面的代碼進(jìn)行一下優(yōu)化處理：

function supply(fruit, quantity) {
    const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
    if (!fruit) throw new Error('沒有水果啦');     // 條件 1: 當(dāng) fruit 無效時，提前處理錯誤
    if (!redFruits.includes(fruit)) return; // 條件 2: 當(dāng)不是紅色水果時，提前 return

    console.log('紅色水果');

    // 條件 3: 水果數(shù)量大于 10 個
    if (quantity > 10) {
        console.log('數(shù)量大于 10 個');
    }
}

這里主要對嵌套層級做了優(yōu)化，提前終止掉了不符合的條件，將三層嵌套減少到了一層，簡化了代碼結(jié)果結(jié)構(gòu)，增強(qiáng)了可閱讀性。

多條件分支的優(yōu)化

相信我們很多人對下面這種代碼不陌生吧？（想想剛開始寫代碼那會啊）

function pick(color) {
    // 根據(jù)顏色選擇水果
    if (color === 'red') {
        return ['apple', 'strawberry'];
    } else if (color === 'yellow') {
        return ['banana', 'pineapple'];
    } else if (color === 'purple') {
        return ['grape', 'plum'];
    } else {
        return [];
    }
}

我們需要知道一點(diǎn)原則：if else 更適合于條件區(qū)間判斷，而 switch case 更適合于具體枚舉值的分支判斷。

我們使用 switch...case...進(jìn)行一下改寫：

function pick(color) {
    // 根據(jù)顏色選擇水果
    switch (color) {
        case 'red':
            return ['apple', 'strawberry'];
        case 'yellow':
            return ['banana', 'pineapple'];
        case 'purple':
            return ['grape', 'plum'];
        default:
            return [];
    }
}

switch...case... 優(yōu)化之后的代碼看上去格式整齊，思路很清晰，但還是很冗長。繼續(xù)優(yōu)化：

借助 Object 的 {key: value} 結(jié)構(gòu)，我們可以在 Object 中枚舉所有的情況，然后將 key 作為索引，直接通過 Object.key 或者 Object[key] 來獲取內(nèi)容:

const fruitColor = {
    red: ['apple', 'strawberry'],
    yellow: ['banana', 'pineapple'],
    purple: ['grape', 'plum'],
}
function pick(color) {
    return fruitColor[color] || [];
}

使用 Map 數(shù)據(jù)結(jié)構(gòu)，真正的(key, value) 鍵值對結(jié)構(gòu):

const fruitColor = new Map()
    .set('red', ['apple', 'strawberry'])
    .set('yellow', ['banana', 'pineapple'])
    .set('purple', ['grape', 'plum']);

function pick(color) {
    return fruitColor.get(color) || [];
}

優(yōu)化之后，代碼更簡潔、更容易擴(kuò)展。

為了更好的可讀性，還可以通過更加語義化的方式定義對象，然后使用 Array.filter 達(dá)到同樣的效果:

const fruits = [
    {name: 'apple', color: 'red'},
    {name: 'strawberry', color: 'red'},
    {name: 'banana', color: 'yellow'},
    {name: 'pineapple', color: 'yellow'},
    {name: 'grape', color: 'purple'},
    {name: 'plum', color: 'purple'}
];

function pick(color) {
    return fruits.filter(f => f.color == color);
}