寫好JavaScript條件語句的5個技巧
1、使用Array.includes處理多種條件
讓我們來看一下的例子:
// conditionfunction test(fruit) {
if (fruit == 'apple' || fruit == 'strawberry') {
console.log('red');
}}
一眼看去,以上的例子貌似沒有什麼問題。但是,如果我們加入更多的紅色水果,比如車釐子(cherry)和蔓越橘(cranberries)?那就要使用||寫更多的條件判斷了。
為了解決以上問題,我們可以使用Array.includes來重寫以上的條件判斷。看以下例子:
function test(fruit) { // extract conditions to array const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries']; if (redFruits.includes(fruit)) { console.log('red'); }}
將這些“紅色水果”(條件)提取到數組裡面,這樣寫就會看起來很簡潔了。
2、減少巢狀,儘早return
下面我們增加兩個條件判斷來擴充套件下上一個例子:
- 如果沒有水果則丟擲異常
- 如果數量超出10則輸出水果數量
function test(fruit, quantity) { const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries']; // condition 1: fruit must has value if (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'); } } } else { throw new Error('No fruit!'); } } // test results test(null); // error: No fruits test('apple'); // print: red test('apple', 20); // print: red, big quantity
看著以上的程式碼,發現以下的問題:
- 一個過濾無效條件的if/else語句
- 巢狀三級 if條件語句
我個人遵循的一般規則:在處理無效條件時儘早return
/_ 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'); } } }
通過這種寫法,可以減少一層巢狀。當if語句很長的時候,這種寫法就比較好。(試想一下,你需要滾動到很底部才能看到else語言,這樣不酷)。
我們也可以用相反的條件判斷和儘早返回來減少if語句巢狀,看以下第二個條件是如何處理的。
/_ 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');
}
}
通過在第二個條件裡取反的條件來判斷,現在的程式碼釋放了一個巢狀。當我們要寫很長的邏輯以及想停止更一步的過程時,這種方法很湊效。
但是,這樣寫並不是硬性的,要看具體的場景。有時候,你要問下自己,這樣的寫法(沒有巢狀)是否比之前那種(在條件2巢狀)更好或者更具可讀性。
對我來說,我只會遠離之前的那種寫法(在條件2巢狀),注意有以下兩個原因:
- 對比巢狀if語句,這種寫法更短,更簡潔
- 取反的條件判斷可能招致更多的思考過程(增加認知負荷)
因此,始終旨在減少巢狀和及時及早返回(return),但是不要過度去這樣做。如果感興趣可以檢視以下的一篇相關文章和在StackOverflow的關於這主題的更多討論:
- 避免else,儘早return
- StackOverflow關於if/else程式碼風格的討論
3、使用函式預設引數和解構
我猜以下的程式碼你看起來應該很熟悉了,我們仍然需要在執行JavaScript時檢查空值(undefined或null)。
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!
事實上,我們可以消除變數p來分配函式預設引數。看以下例子:
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!
是不是看起來很簡單,很直觀?請記住,每個引數都有自己的預設函式引數。例如,我們也可以為fruit分配預設值(即以上程式碼的第一個引數):
function test(fruit = 'unknown', quantity = 1)
另外一個問題來了:如果fruit引數是Object型別怎麼辦?我們可以指定預設值嗎?
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
看看上面的例子,我們想要打印出fruit的名稱如果值存在的話,否則打印出“unknow”。我們可以通過檢查預設函式引數和物件解構來避免fruit&&fruit.name這個條件。看看以下例子:
// 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
由於我們只需要fruit物件的name屬性,可以使用{name}來解構物件,然後我們可以在程式碼中使用name作為變數代替fruit.name。
我們還將分配空物件{}作為預設引數。如果不這樣做的話,在執行test(undefined)這行程式碼時就會報錯:無法解析’undefined’或’null’的屬性name,因為undefined沒有name這個屬性。
如果你不介意使用第三方庫,有幾個方法可以減少空值檢查。
- 使用Lodash get方法
- 使用Facebook 開源的idx庫(和Bablejs)
以下是使用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
你可以在這裡執行這個demo。另外,如果你熱衷於使用函數語言程式設計(FP),你可能會選擇用Lodash fp,Lodash的函式式版本(方法改為get或getOr)。Java架構交流學習圈:874811168 面向1-3年經驗 Java開發人員 幫助突破瓶頸 提升思維能力
4、支援Map或物件字面量而不是switch宣告
我們來看看以下例子:
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']
以上的程式碼看起來貌似沒什麼錯誤,但是我發現程式碼十分雜亂。其實可以用字面量物件更清晰的語法來實現相同的結果。例如:
// 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來實現同樣的結果,例如:
// 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是自ES6開始才可用的物件型別,可以用來儲存鍵值對。
5、對所有或部分規則使用Array.every和Array.some
最後一個技巧是關於利用新的(其實不是很新了)JavaScript陣列方法來減少程式碼函式。看看下面的程式碼,我們想所有的水果是否都是紅色的。
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
}
上面的程式碼真的好冗長!我們可以用Array.every來減少程式碼行數。來看看:
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
}
現在是不是簡潔很多了?同樣類似的方法,如果你想測試是否所有的水果都是紅色,可以用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
}
總結
讓我們一起生成更多可讀程式碼。 我希望你能在本文中學到一些新東西。
就這樣。 寫程式碼快樂!
(完)