在使用 JavaScript 時,我們常常要寫不少的條件語句。這裏有五個小技巧,可以讓你寫出更乾淨、漂亮的條件語句。
1. 使用 Array.includes 來處理多重條件
舉個栗子:
// 條件語句
function test(fruit) {
if (fruit == 'apple' || fruit == 'strawberry') {
console.log('red');
}
}
乍一看,這麼寫似乎沒什麼大問題。然而,如果我們想要匹配更多的紅色水果呢,比方說『櫻桃』和『蔓越莓』?我們是不是得用更多的 || 來擴展這條語句?
我們可以使用 Array.includes重寫以上條件句。
function test(fruit) {
// 把條件提取到數組中
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
if (redFruits.includes(fruit)) {
console.log('red');
}
}
我們把紅色的水果(條件)都提取到一個數組中,這使得我們的代碼看起來更加整潔。
2. 少寫嵌套,儘早返回
讓我們爲之前的例子添加兩個條件:
如果沒有提供水果,拋出錯誤。
如果該水果的數量大於 10,將其打印出來。
function test(fruit, quantity) {
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
// 條件 1:fruit 必須有值
if (fruit) {
// 條件 2:必須爲紅色
if (redFruits.includes(fruit)) {
console.log('red');
// 條件 3:必須是大量存在
if (quantity > 10) {
console.log('big quantity');
}
}
} else {
thrownewError('No fruit!');
}
}
// 測試結果
test(null); // 報錯:No fruits
test(‘apple’); // 打印:red
test(‘apple’, 20); // 打印:red,big quantity複製代碼
讓我們來仔細看看上面的代碼,我們有:
1 個 if/else 語句來篩選無效的條件
3 層 if 語句嵌套(條件 1,2 & 3)
就我個人而言,我遵循的一個總的規則是當發現無效條件時儘早返回。
/_ 當發現無效條件時儘早返回 _/
function test(fruit, quantity) {
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
// 條件 1:儘早拋出錯誤
if (!fruit) thrownewError('No fruit!');
// 條件2:必須爲紅色
if (redFruits.includes(fruit)) {
console.log('red');
// 條件 3:必須是大量存在
if (quantity > 10) {
console.log('big quantity');
}
}
}
如此一來,我們就少寫了一層嵌套。這是種很好的代碼風格,尤其是在 if 語句很長的時候(試想一下,你得滾動到底部才能知道那兒還有個 else 語句,是不是有點不爽)。
如果反轉一下條件,我們還可以進一步地減少嵌套層級。注意觀察下面的條件 2 語句,看看是如何做到這點的:
/_ 當發現無效條件時儘早返回 _/
function test(fruit, quantity) {
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
if (!fruit) thrownewError('No fruit!'); // 條件 1:儘早拋出錯誤
if (!redFruits.includes(fruit)) return; // 條件 2:當 fruit 不是紅色的時候,直接返回
console.log('red');
// 條件 3:必須是大量存在
if (quantity > 10) {
console.log('big quantity');
}
}
通過反轉條件 2 的條件,現在我們的代碼已經沒有嵌套了。當我們代碼的邏輯鏈很長,並且希望當某個條件不滿足時不再執行之後流程時,這個技巧會很好用。
然而,並沒有任何硬性規則要求你這麼做。這取決於你自己,對你而言,這個版本的代碼(沒有嵌套)是否要比之前那個版本(條件 2 有嵌套)的更好、可讀性更強?
是我的話,我會選擇前一個版本(條件 2 有嵌套)。原因在於:
這樣的代碼比較簡短和直白,一個嵌套的 if 使得結構更加清晰。
條件反轉會導致更多的思考過程(增加認知負擔)。
因此,始終追求更少的嵌套,更早地返回,但是不要過度。感興趣的話,這裏有篇關於這個問題的文章以及 StackOverflow 上的討論:
Avoid Else, Return Early by Tim Oxley
StackOverflow discussion on if/else coding style
3. 相較於 switch,Map / Object 也許是更好的選擇
讓我們看下面的例子,我們想要根據顏色打印出各種水果:
function test(color) {
// 使用 switch case 來找到對應顏色的水果
switch (color) {
case 'red':
return ['apple', 'strawberry'];
case 'yellow':
return ['banana', 'pineapple'];
case 'purple':
return ['grape', 'plum'];
default:
return [];
}
}
//測試結果
test(null); // []
test('yellow'); // ['banana', 'pineapple']
上面的代碼看上去並沒有錯,但是就我個人而言,它看上去很冗長。同樣的結果可以通過對象字面量來實現,語法也更加簡潔:
// 使用對象字面量來找到對應顏色的水果
const fruitColor = {
red: ['apple', 'strawberry'],
yellow: ['banana', 'pineapple'],
purple: ['grape', 'plum']
};
function test(color) {
return fruitColor[color] || [];
}
或者,你也可以使用 Map 來實現同樣的效果:
// 使用 Map 來找到對應顏色的水果
const fruitColor = newMap()
.set('red', ['apple', 'strawberry'])
.set('yellow', ['banana', 'pineapple'])
.set('purple', ['grape', 'plum']);
function test(color) {
return fruitColor.get(color) || [];
}
Map 是 ES2015 引入的新的對象類型,允許你存放鍵值對。
那是不是說我們應該禁止使用 switch 語句? 別把自己限制住。我自己會在任何可能的時候使用對象字面量,但是這並不是說我就不用 switch,這得視場景而定。
Todd Motto 有一篇文章深入討論了 switch 語句和對象字面量,你也許會想看看。
懶人版:重構語法
就以上的例子,事實上我們可以通過重構我們的代碼,使用 Array.filter 實現同樣的效果。
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 test(color) {
// 使用 Array filter 來找到對應顏色的水果
return fruits.filter(f => f.color == color);
}
解決問題的方法永遠不只一種。對於這個例子我們展示了四種實現方法。Coding is fun!
總結
讓我們一起寫出可讀性更高的代碼吧。希望這篇文章能給你們帶來一些幫助。
就是這樣啦。Happy coding!
關於本文
譯者:@Hopsken
譯文:
https://juejin.im/post/5bb9e3085188255c352d7326(掘金翻譯)
作者:@Jecelyn Yeen
原文:
https://scotch.io/tutorials/5-tips-to-write-better-conditionals-in-javascript
校對者:@Park-ma、@ThomasWhyne