寫好 JS 條件語(yǔ)句的 5 條守則
在用 JavaScript 工作時(shí),我們經(jīng)常和條件語(yǔ)句打交道,這里有5條讓你寫出更好/干凈的條件語(yǔ)句的建議。
目錄:
- 多重判斷時(shí)使用 Array.includes
- 更少的嵌套,盡早 return
- 使用默認(rèn)參數(shù)和解構(gòu)
- 傾向于遍歷對(duì)象而不是 Switch 語(yǔ)句
- 對(duì) 所有/部分 判斷使用 Array.every & Array.some
- 總結(jié)
1. 多重判斷時(shí)使用 Array.includes
讓我們看一下下面這個(gè)例子:
- // condition
- function test(fruit) {
- if (fruit == 'apple' || fruit == 'strawberry') {
- console.log('red');
- }
- }
第一眼,上面這個(gè)例子看起來(lái)沒問(wèn)題。如果我們有更多名字叫 cherry 和 cranberries 的紅色水果呢?我們準(zhǔn)備用更多的 || 來(lái)拓展條件語(yǔ)句嗎?
我們可以用 Array.includes (Array.includes)重寫條件語(yǔ)句。
- function test(fruit) {
- const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
- if (redFruits.includes(fruit)) {
- console.log('red');
- }
- }
我們把紅色的水果(red fruits)這一判斷條件提取到一個(gè)數(shù)組。這樣一來(lái),代碼看起來(lái)更整潔。
2. 更少的嵌套,盡早 Return
讓我們拓展上一個(gè)例子讓它包含兩個(gè)條件。
- 如果沒有傳入?yún)?shù) fruit,拋出錯(cuò)誤
- 接受 quantity 參數(shù),并且在 quantity 大于 10 時(shí)打印出來(lái)
- function test(fruit, quantity) {
- const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
- // 條件 1: fruit 必須有值
- if (fruit) {
- // 條件 2: 必須是red的
- if (redFruits.includes(fruit)) {
- console.log('red');
- // 條件 3: quantity大于10
- if (quantity > 10) {
- console.log('big quantity');
- }
- }
- } else {
- throw new Error('No fruit!');
- }
- }
- // 測(cè)試結(jié)果
- test(null); // error: No fruits
- test('apple'); // print: red
- test('apple', 20); // print: red, big quantity
在上面的代碼, 我們有:
- 1個(gè) if/else 語(yǔ)句篩選出無(wú)效的語(yǔ)句
- 3層if嵌套語(yǔ)句 (條件 1, 2 & 3)
我個(gè)人遵循的規(guī)則一般是在發(fā)現(xiàn)無(wú)效條件時(shí),盡早Return。
- /_ 當(dāng)發(fā)現(xiàn)無(wú)效語(yǔ)句時(shí),盡早Return _/
- function test(fruit, quantity) {
- const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
- // 條件 1: 盡早拋出錯(cuò)誤
- if (!fruit) throw new Error('No fruit!');
- // 條件 2: 必須是紅色的
- if (redFruits.includes(fruit)) {
- console.log('red');
- // 條件 3: 必須是大質(zhì)量的
- if (quantity > 10) {
- console.log('big quantity');
- }
- }
- }
這樣一來(lái),我們少了一層嵌套語(yǔ)句。這種編碼風(fēng)格非常好,尤其是當(dāng)你有很長(zhǎng)的if語(yǔ)句的時(shí)候(想象你需要滾動(dòng)到最底層才知道還有else語(yǔ)句,這并不酷)
我們可以通過(guò) 倒置判斷條件 & 盡早return 進(jìn)一步減少if嵌套。看下面我們是怎么處理判斷 條件2 的:
- /_ 當(dāng)發(fā)現(xiàn)無(wú)效語(yǔ)句時(shí),盡早Return _/
- function test(fruit, quantity) {
- const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
- // 條件 1: 盡早拋出錯(cuò)誤
- if (!fruit) throw new Error('No fruit!');
- // 條件 2: 當(dāng)水果不是紅色時(shí)停止繼續(xù)執(zhí)行
- if (!redFruits.includes(fruit)) return;
- console.log('red');
- // 條件 3: 必須是大質(zhì)量的
- if (quantity > 10) {
- console.log('big quantity');
- }
- }
通過(guò)倒置判斷條件2,我們的代碼避免了嵌套語(yǔ)句。這個(gè)技巧在我們需要進(jìn)行很長(zhǎng)的邏輯判斷時(shí)是非常有用的,特別是我們希望能夠在條件不滿足時(shí)能夠停止下來(lái)進(jìn)行處理。
而且這么做并不困難。問(wèn)問(wèn)自己,這個(gè)版本(沒有嵌套)是不是比之前的(兩層條件嵌套)更好,可讀性更高?
但對(duì)于我,我會(huì)保留先前的版本(包含兩層嵌套)。這是因?yàn)?
- 代碼比較短且直接,包含if嵌套的更清晰
- 倒置判斷條件可能加重思考的負(fù)擔(dān)(增加認(rèn)知載荷)
因此,應(yīng)當(dāng)盡力減少嵌套和盡早return,但不要過(guò)度。如果你感興趣的話,可以看一下關(guān)于這個(gè)話題的一篇文章和 StackOverflow 上的討論。
- Avoid Else, Return Early by Tim Oxley
- StackOverflow discussion on if/else coding style
3. 使用默認(rèn)參數(shù)和解構(gòu)
我猜下面的代碼你可能會(huì)熟悉,在JavaScript中我們總是需要檢查 null / undefined的值和指定默認(rèn)值:
- function test(fruit, quantity) {
- if (!fruit) return;
- // 如果 quantity 參數(shù)沒有傳入,設(shè)置默認(rèn)值為 1
- const q = quantity || 1;
- 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) {
- // 如果 quantity 參數(shù)沒有傳入,設(shè)置默認(rèn)值為 1
- if (!fruit) return;
- console.log(`We have ${quantity} ${fruit}!`);
- }
- //test results
- test('banana'); // We have 1 banana!
- test('apple', 2); // We have 2 apple!
這更加直觀,不是嗎?注意,每個(gè)聲明都有自己的默認(rèn)參數(shù).
例如,我們也能給fruit分配默認(rèn)值:function test(fruit = 'unknown', quantity = 1)。
如果fruit是一個(gè)object會(huì)怎么樣?我們能分配一個(gè)默認(rèn)參數(shù)嗎?
- function test(fruit) {
- // 當(dāng)值存在時(shí)打印 fruit 的值
- 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
看上面這個(gè)例子,我們想打印 fruit 對(duì)象中可能存在的 name 屬性。否則我們將打印unknown。我們可以通過(guò)默認(rèn)參數(shù)以及解構(gòu)從而避免判斷條件 fruit && fruit.name
- // 解構(gòu) - 僅僅獲取 name 屬性
- // 為其賦默認(rèn)值為空對(duì)象
- function test({name} = {}) {
- console.log (name || 'unknown');
- }
- // test results
- test(undefined); // unknown
- test({ }); // unknown
- test({ name: 'apple', color: 'red' }); // apple
由于我們只需要 name 屬性,我們可以用 {name} 解構(gòu)出參數(shù),然后我們就能使用變量 name 代替 fruit.name。
我們也需要聲明空對(duì)象 {} 作為默認(rèn)值。如果我們不這么做,當(dāng)執(zhí)行 test(undefined) 時(shí),你將得到一個(gè)無(wú)法對(duì) undefined 或 null 解構(gòu)的的錯(cuò)誤。因?yàn)樵?undefined 中沒有 name 屬性。
如果你不介意使用第三方庫(kù),這有一些方式減少null的檢查:
- 使用 Lodash get函數(shù)
- 使用Facebook開源的idx庫(kù)(with Babeljs)
這是一個(gè)使用Lodash的例子:
- function test(fruit) {
- // 獲取屬性名,如果屬性名不可用,賦默認(rèn)值為 unknown
- console.log(__.get(fruit, 'name', 'unknown');
- }
- // test results
- test(undefined); // unknown
- test({ }); // unknown
- test({ name: 'apple', color: 'red' }); // apple
你可以在jsbin運(yùn)行demo代碼。除此之外,如果你是函數(shù)式編程的粉絲,你可能選擇使用 Lodash fp,Lodash的函數(shù)式版本(方法變更為get或者getOr)。
4. 傾向于對(duì)象遍歷而不是Switch語(yǔ)句
讓我們看下面這個(gè)例子,我們想根據(jù) color 打印出水果:
- function test(color) {
- // 使用條件語(yǔ)句來(lái)尋找對(duì)應(yīng)顏色的水果
- 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']
上面的代碼看起來(lái)沒有錯(cuò)誤,但是我找到了一些累贅。用對(duì)象遍歷實(shí)現(xiàn)相同的結(jié)果,語(yǔ)法看起來(lái)更簡(jiǎn)潔:
- const fruitColor = {
- red: ['apple', 'strawberry'],
- yellow: ['banana', 'pineapple'],
- purple: ['grape', 'plum']
- };
- function test(color) {
- return fruitColor[color] || [];
- }
或者你也可以使用 Map實(shí)現(xiàn)相同的結(jié)果:
- 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 規(guī)范之后實(shí)現(xiàn)的對(duì)象類型,允許你存儲(chǔ) key 和 value 的值。
但我們是否應(yīng)當(dāng)禁止switch語(yǔ)句的使用呢?答案是不要限制你自己。從個(gè)人來(lái)說(shuō),我會(huì)盡可能的使用對(duì)象遍歷,但我并不嚴(yán)格遵守它,而是使用對(duì)當(dāng)前的場(chǎng)景更有意義的方式。
Todd Motto有一篇關(guān)于 switch 語(yǔ)句對(duì)比對(duì)象遍歷的更深入的文章,你可以在這個(gè)地方閱讀
TL;DR; 重構(gòu)語(yǔ)法
在上面的例子,我們能夠用Array.filter 重構(gòu)我們的代碼,實(shí)現(xiàn)相同的效果。
- 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) {
- return fruits.filter(f => f.color == color);
- }
有著不止一種方法能夠?qū)崿F(xiàn)相同的結(jié)果,我們以上展示了 4 種。
5. 對(duì)所有/部分 判斷使用Array.every & Array.some
這最后一個(gè)建議更多是關(guān)于利用 JavaScript Array 的內(nèi)置方法來(lái)減少代碼行數(shù)??聪旅娴拇a,我們想要檢查是否所有水果都是紅色:
- const fruits = [
- { name: 'apple', color: 'red' },
- { name: 'banana', color: 'yellow' },
- { name: 'grape', color: 'purple' }
- ];
- function test() {
- let isAllRed = true;
- // 條件:所有水果都是紅色
- for (let f of fruits) {
- if (!isAllRed) break;
- isAllRed = (f.color == 'red');
- }
- console.log(isAllRed); // false
- }
代碼那么長(zhǎng)!我們可以通過(guò) Array.every減少代碼行數(shù):
- const fruits = [
- { name: 'apple', color: 'red' },
- { name: 'banana', color: 'yellow' },
- { name: 'grape', color: 'purple' }
- ];
- function test() {
- const isAllRed = fruits.every(f => f.color == 'red');
- console.log(isAllRed); // false
- }
現(xiàn)在更簡(jiǎn)潔了,不是嗎?相同的方式,如果我們想測(cè)試是否存在紅色的水果,我們可以使用 Array.some 一行代碼實(shí)現(xiàn)。
- const fruits = [
- { name: 'apple', color: 'red' },
- { name: 'banana', color: 'yellow' },
- { name: 'grape', color: 'purple' }
- ];
- function test() {
- // 條件:任何一個(gè)水果是紅色
- const isAnyRed = fruits.some(f => f.color == 'red');
- console.log(isAnyRed); // true
- }
6. 總結(jié)
讓我們一起生產(chǎn)更多可讀性高的代碼。我希望你能從這篇文章學(xué)到東西。
這就是所有的內(nèi)容。編碼快樂(lè)!