在使用 JavaScript 時,我們常常要寫不少的條件語句。這裡有五個小技巧,可以讓你寫出更乾淨、漂亮的條件語句。
1. 使用 Array.includes 來處理多重條件
舉個例子 ?:
// 條件語句functiontest(fruit) {
if (fruit == 'apple' || fruit == 'strawberry') {
console.log('red');
}
}
複製程式碼
複製程式碼
乍一看,這麼寫似乎沒什麼大問題。然而,如果我們想要匹配更多的紅色水果呢,比方說『櫻桃』和『蔓越莓』?我們是不是得用更多的 ||
來擴充套件這條語句?
我們可以使用 Array.includes
(Array.includes) 重寫以上條件句。
functiontest(fruit) {
// 把條件提取到陣列中const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
if (redFruits.includes(fruit)) {
console.log('red');
}
}
複製程式碼
複製程式碼
我們把紅色的水果
(條件)都提取到一個陣列中,這使得我們的程式碼看起來更加整潔。
2. 少寫巢狀,儘早返回
讓我們為之前的例子新增兩個條件:
-
如果沒有提供水果,丟擲錯誤。
-
如果該水果的數量大於 10,將其列印出來。
functiontest(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)
就我個人而言,我遵循的一個總的規則是當發現無效條件時儘早返回。
/_ 當發現無效條件時儘早返回 _/
functiontest(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 語句,看看是如何做到這點的:
/_ 當發現無效條件時儘早返回 _/
functiontest(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. 使用函式預設引數和解構
我猜你也許很熟悉以下的程式碼,在 JavaScript 中我們經常需要檢查 null
/ undefined
並賦予預設值:
functiontest(fruit, quantity) {
if (!fruit) return;
const q = quantity || 1; // 如果沒有提供 quantity,預設為 1console.log(`We have ${q}${fruit}!`);
}
//測試結果
test('banana'); // We have 1 banana!
test('apple', 2); // We have 2 apple!複製程式碼
複製程式碼
事實上,我們可以通過函式的預設引數來去掉變數 q
。
functiontest(fruit, quantity = 1) { // 如果沒有提供 quantity,預設為 1if (!fruit) return;
console.log(`We have ${quantity}${fruit}!`);
}
//測試結果
test('banana'); // We have 1 banana!
test('apple', 2); // We have 2 apple!複製程式碼
複製程式碼
是不是更加簡單、直白了?請注意,所有的函式引數都可以有其預設值。舉例來說,我們同樣可以為 fruit
賦予一個預設值:function test(fruit = 'unknown', quantity = 1)
。
那麼如果 fruit
是一個物件(Object)呢?我們還可以使用預設引數嗎?
functiontest(fruit) {
// 如果有值,則列印出來if (fruit && fruit.name) {
console.log (fruit.name);
} else {
console.log('unknown');
}
}
//測試結果
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple複製程式碼
複製程式碼
觀察上面的例子,當水果名稱屬性存在時,我們希望將其列印出來,否則列印『unknown』。我們可以通過預設引數和解構賦值的方法來避免寫出 fruit && fruit.name
這種條件。
// 解構 —— 只得到 name 屬性// 預設引數為空物件 {}functiontest({name} = {}) {
console.log (name || 'unknown');
}
//測試結果
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple複製程式碼
複製程式碼
既然我們只需要 fruit 的 name
屬性,我們可以使用 {name}
來將其解構出來,之後我們就可以在程式碼中使用 name
變數來取代 fruit.name
。
我們還使用 {}
作為其預設值。如果我們不這麼做的話,在執行 test(undefined)
時,你會得到一個錯誤 Cannot destructure property name of 'undefined' or 'null'.
,因為 undefined
上並沒有 name
屬性。(譯者注:這裡不太準確,其實因為解構只適用於物件(Object),而不是因為undefined
上並沒有 name
屬性(空物件上也沒有)。參考解構賦值 - MDN)
如果你不介意使用第三方庫的話,有一些方法可以幫助減少空值(null)檢查:
- 使用 Lodash get 函式
- 使用 Facebook 開源的 idx 庫(需搭配 Babeljs)
這裡有一個使用 Lodash 的例子:
// 使用 lodash 庫提供的 _ 方法functiontest(fruit) {
console.log(_.get(fruit, 'name', 'unknown'); // 獲取屬性 name 的值,如果沒有,設為預設值 unknown
}
//測試結果
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple複製程式碼
複製程式碼
你可以在這裡執行演示程式碼。另外,如果你偏愛函數語言程式設計(FP),你可以選擇使用 Lodash fp——函式式版本的 Lodash(方法名變為 get
或 getOr
)。
4. 相較於 switch,Map / Object 也許是更好的選擇
讓我們看下面的例子,我們想要根據顏色列印出各種水果:
functiontest(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']
};
functiontest(color) {
return fruitColor[color] || [];
}
複製程式碼
複製程式碼
或者,你也可以使用 Map 來實現同樣的效果:
// 使用 Map 來找到對應顏色的水果const fruitColor = newMap()
.set('red', ['apple', 'strawberry'])
.set('yellow', ['banana', 'pineapple'])
.set('purple', ['grape', 'plum']);
functiontest(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' }
];
functiontest(color) {
// 使用 Array filter 來找到對應顏色的水果return fruits.filter(f => f.color == color);
}
複製程式碼
複製程式碼
解決問題的方法永遠不只一種。對於這個例子我們展示了四種實現方法。Coding is fun!
5. 使用 Array.every 和 Array.some 來處理全部/部分滿足條件
最後一個小技巧更多地是關於使用新的(也不是很新了)JavaScript 陣列函式來減少程式碼行數。觀察以下的程式碼,我們想要檢查是否所有的水果都是紅色的:
const fruits = [
{ name: 'apple', color: 'red' },
{ name: 'banana', color: 'yellow' },
{ name: 'grape', color: 'purple' }
];
functiontest() {
let isAllRed = true;
// 條件:所有的水果都必須是紅色for (let f of fruits) {
if (!isAllRed) break;
isAllRed = (f.color == 'red');
}
console.log(isAllRed); // false
}
複製程式碼
複製程式碼
這段程式碼也太長了!我們可以通過 Array.every
來縮減程式碼:
const fruits = [
{ name: 'apple', color: 'red' },
{ name: 'banana', color: 'yellow' },
{ name: 'grape', color: 'purple' }
];
functiontest() {
// 條件:(簡短形式)所有的水果都必須是紅色const isAllRed = fruits.every(f => f.color == 'red');
console.log(isAllRed); // false
}
複製程式碼
複製程式碼
清晰多了對吧?類似的,如果我們想要檢查是否有至少一個水果是紅色的,我們可以使用 Array.some
僅用一行程式碼就實現出來。
const fruits = [
{ name: 'apple', color: 'red' },
{ name: 'banana', color: 'yellow' },
{ name: 'grape', color: 'purple' }
];
functiontest() {
// 條件:至少一個水果是紅色的const isAnyRed = fruits.some(f => f.color == 'red');
console.log(isAnyRed); // true
}
複製程式碼
複製程式碼
總結
讓我們一起寫出可讀性更高的程式碼吧。希望這篇文章能給你們帶來一些幫助。