在使用 JavaScript 時(shí),我們會(huì)處理很多條件語(yǔ)句,這里有 5 個(gè)技巧可以幫助您編寫更好、更簡(jiǎn)潔的條件語(yǔ)句。
1、對(duì)多個(gè)條件使用 Array.includes
讓我們看看下面的例子:
// condition
function test(fruit) {
if (fruit == 'apple' || fruit == 'strawberry') {
console.log('red');
}
}
乍一看,上面的例子看起來(lái)不錯(cuò)。然而,如果還有更多紅顏色的水果需要判斷呢,比如櫻桃和小紅莓,我們要用更多的 || 來(lái)擴(kuò)展這個(gè)表述嗎?
我們可以用 Array.includes 重寫上面的條件
function test(fruit) {
// extract conditions to array
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
if (redFruits.includes(fruit)) {
console.log('red');
}
}
我們將紅色水果(條件)提取到一個(gè)數(shù)組中。這樣做之后,代碼看起來(lái)更整潔。
2、更少的嵌套,盡早返回
讓我們擴(kuò)展前面的示例,以包含另外兩個(gè)條件:
如果沒有提供水果(名稱),拋出錯(cuò)誤。
如果(紅色水果)數(shù)量超過(guò) 10 個(gè),接受并打印。
看看上面的代碼,我們有:
1 組過(guò)濾無(wú)效條件的 if/else 語(yǔ)句
3層的 if 嵌套語(yǔ)句(條件 1、2 和 3)
我個(gè)人遵循的一般規(guī)則是,當(dāng)發(fā)現(xiàn)無(wú)效條件時(shí),提前返回。
/_ return early when invalid conditions found _/
function test(fruit, quantity) {
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
// condition 1: throw error early
if (!fruit) throw new Error('No fruit!');
// condition 2: must be red
if (redFruits.includes(fruit)) {
console.log('red');
// condition 3: must be big quantity
if (quantity > 10) {
console.log('big quantity');
}
}
}
這樣,我們就少了一層嵌套。這種編碼風(fēng)格很好,尤其是當(dāng)你有很長(zhǎng)的 if 語(yǔ)句時(shí)(想象一下,你需要滾動(dòng)到最底部才能知道還有一個(gè) else 語(yǔ)句,這并不酷)。
通過(guò)反轉(zhuǎn)條件和提早返回,我們可以進(jìn)一步減少嵌套。看看下面的條件 2,我們是怎么做的:
/_ return early when invalid conditions found _/
function test(fruit, quantity) {
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
if (!fruit) throw new Error('No fruit!'); // condition 1: throw error early
if (!redFruits.includes(fruit)) return; // condition 2: stop when fruit is not red
console.log('red');
// condition 3: must be big quantity
if (quantity > 10) {
console.log('big quantity');
}
}
通過(guò)反轉(zhuǎn)條件 2 的條件,我們的代碼現(xiàn)在沒有嵌套語(yǔ)句。當(dāng)我們有很長(zhǎng)的邏輯要處理時(shí),這種技術(shù)是有用的,當(dāng)一個(gè)條件沒有滿足時(shí),我們想要停止進(jìn)一步的處理。
然而,這并不是嚴(yán)格的規(guī)則。問問自己,這個(gè)版本(沒有嵌套)是否比前一個(gè)版本(嵌套的條件 2)更好、更易讀?
對(duì)于我來(lái)說(shuō),我將把它保留為以前的版本(條件 2 和嵌套)。這是因?yàn)椋?/p>
代碼簡(jiǎn)短而直接,如果嵌套,代碼就更清晰了
反轉(zhuǎn)條件可能會(huì)導(dǎo)致更多的思考過(guò)程(增加認(rèn)知負(fù)擔(dān))
因此,總是以更少的嵌套及盡早返回為目標(biāo),但不要過(guò)度。如果你感興趣的話,StackOverflow 有一篇相關(guān)的文章討論了這個(gè)話題:
Avoid Else, Return Early by Tim Oxley
StackOverflow discussion 關(guān)于 if/else 編碼風(fēng)格的討論
3、使用默認(rèn)的函數(shù)參數(shù)和解構(gòu)
我想下面的代碼對(duì)您來(lái)說(shuō)可能很熟悉,我們?cè)谑褂?JavaScript 時(shí)總是需要檢查 null 或 undefined 值并分配默認(rèn)值:
function test(fruit, quantity) {
if (!fruit) return;
const q = quantity || 1; // if quantity not provided, default to one
console.log(`We have ${q} ${fruit}!`);
}
//test results
test('banana'); // We have 1 banana!
test('apple', 2); // We have 2 apple!
事實(shí)上,我們可以通過(guò)指定默認(rèn)的函數(shù)參數(shù)來(lái)消除變量 q。
function test(fruit, quantity = 1) { // if quantity not provided, default to one
if (!fruit) return;
console.log(`We have ${quantity} ${fruit}!`);
}
//test results
test('banana'); // We have 1 banana!
test('apple', 2); // We have 2 apple!
更簡(jiǎn)單和直觀,不是嗎?請(qǐng)注意,每個(gè)參數(shù)都可以有自己的默認(rèn)函數(shù)參數(shù)。例如,我們也可以為 fruit 賦值:function test(fruit = 'unknown', quantity = 1)。
如果我們的 fruit 是一個(gè)對(duì)象呢?我們可以指定默認(rèn)參數(shù)嗎?
function test(fruit) {
// printing fruit name if value provided
if (fruit && fruit.name) {
console.log (fruit.name);
} else {
console.log('unknown');
}
}
//test results
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple
請(qǐng)看上面的示例,如果 fruit.name 是可用的,我們將打印該水果名稱,否則我們將打印 unknown。我們可以避免使用與默認(rèn)函數(shù)參數(shù)和解構(gòu)對(duì)條件 fruit && fruit.name 進(jìn)行檢查。
// destructing - get name property only
// assign default empty object {}
function test({name} = {}) {
console.log (name || 'unknown');
}
//test results
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple
因?yàn)槲覀冎恍枰械膶傩?name,所以我們可以使用 {name} 來(lái)解構(gòu),然后我們可以在代碼中使用 name 作為變量,而不是 fruit.name。
我們還將空對(duì)象 {} 指定為默認(rèn)值。如果我們不這樣做,當(dāng)執(zhí)行 test(undefined),不能解構(gòu) undefined 或 null 的屬性名時(shí),您將會(huì)得到錯(cuò)誤。因?yàn)樵?undefined中沒有 name 屬性。
如果您不介意使用第三方庫(kù),有一些方法可以減少 null 檢查:
使用 Lodash 的 get 函數(shù)
使用 Facebook 的開源庫(kù) idx(以及 Babeljs)
這是使用 Lodash 的例子:
// Include lodash library, you will get _
function test(fruit) {
console.log(__.get(fruit, 'name', 'unknown'); // get property name, if not available, assign default value 'unknown'
}
//test results
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple
您可以在 這里 運(yùn)行演示代碼。此外,如果你喜歡函數(shù)式編程(FP),你可以選擇使用 Lodash fp, 即 Lodash 的函數(shù)式版本(方法改為 get 或 getOr)。
4、選擇 Map 或?qū)ο笞置媪浚皇?Switch 語(yǔ)句
讓我們看看下面的例子,我們想要基于顏色打印水果名稱:
function test(color) {
// use switch case to find fruits in color
switch (color) {
case 'red':
return ['apple', 'strawberry'];
case 'yellow':
return ['banana', 'pineapple'];
case 'purple':
return ['grape', 'plum'];
default:
return [];
}
}
//test results
test(null); // []
test('yellow'); // ['banana', 'pineapple']
上面的代碼似乎沒有什么問題,但我發(fā)現(xiàn)它相當(dāng)冗長(zhǎng)。同樣的結(jié)果可以通過(guò)對(duì)象字面量和更簡(jiǎn)潔的語(yǔ)法來(lái)實(shí)現(xiàn):
// use object literal to find fruits in color
const fruitColor = {
red: ['apple', 'strawberry'],
yellow: ['banana', 'pineapple'],
purple: ['grape', 'plum']
};
function test(color) {
return fruitColor[color] || [];
}
或者,可以使用 Map 來(lái)實(shí)現(xiàn)相同的結(jié)果:
// use Map to find fruits in color
const fruitColor = new Map()
.set('red', ['apple', 'strawberry'])
.set('yellow', ['banana', 'pineapple'])
.set('purple', ['grape', 'plum']);
function test(color) {
return fruitColor.get(color) || [];
}
Map 是 ES2015 以后可用的對(duì)象類型,允許您存儲(chǔ)鍵值對(duì)。
我們應(yīng)該禁止使用 switch 語(yǔ)句嗎?不要把自己局限于此。就我個(gè)人而言,我盡可能地使用對(duì)象字面量,但是我不會(huì)設(shè)置嚴(yán)格的規(guī)則來(lái)阻止它,使用對(duì)您的場(chǎng)景有意義的任何一個(gè)。
Todd Motto 有一篇文章深入討論 switch 語(yǔ)句與對(duì)象字面量,你可以在 這里 閱讀。
TL;DR; 重構(gòu)的語(yǔ)法
對(duì)于上面的示例,我們實(shí)際上可以重構(gòu)代碼,以使用 Array.filter 獲得相同的結(jié)果。
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) {
// use Array filter to find fruits in color
return fruits.filter(f => f.color == color);
}
總有不止一種方法可以達(dá)到同樣的效果。我們展示了 4 個(gè)相同效果的例子。編碼是有趣的!
5、所有或部分使用 Array.every & Array.some 的條件
最后一個(gè)技巧是關(guān)于使用新的(但不是很新)Javascript 數(shù)組函數(shù)來(lái)減少代碼行。看看下面的代碼,我們想檢查所有的水果是否都是紅色的:
const fruits = [
{ name: 'apple', color: 'red' },
{ name: 'banana', color: 'yellow' },
{ name: 'grape', color: 'purple' }
];
function test() {
let isAllRed = true;
// condition: all fruits must be red
for (let f of fruits) {
if (!isAllRed) break;
isAllRed = (f.color == 'red');
}
console.log(isAllRed); // false
}
代碼太長(zhǎng)了!我們可以用 Array.every 來(lái)減少行數(shù):
const fruits = [
{ name: 'apple', color: 'red' },
{ name: 'banana', color: 'yellow' },
{ name: 'grape', color: 'purple' }
];
function test() {
// condition: short way, all fruits must be red
const isAllRed = fruits.every(f => f.color == 'red');
console.log(isAllRed); // false
}
現(xiàn)在干凈多了,對(duì)吧?類似地,如果我們想用一行代碼來(lái)判斷任何一個(gè)水果是否為紅色,我們可以使用 Array.some。
const fruits = [
{ name: 'apple', color: 'red' },
{ name: 'banana', color: 'yellow' },
{ name: 'grape', color: 'purple' }
];
function test() {
// condition: if any fruit is red
const isAnyRed = fruits.some(f => f.color == 'red');
console.log(isAnyRed); // true
}