1. 程式人生 > >優化js條件語句

優化js條件語句

1、使用 Array.includes 來處理多重條件
舉個栗子 :

// 條件語句
function test(fruit) {
  if (fruit == 'apple' || fruit == 'strawberry') {
    console.log('red');
  }
}

乍一看,這麼寫似乎沒什麼大問題。然而,如果我們想要匹配更多的紅色水果呢,比方說『櫻桃』和
『蔓越莓』?我們是不是得用更多的 || 來擴充套件這條語句?
我們可以使用 Array.includes(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 {
    throw new Error('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) throw new Error('No fruit!');
  
  // 條件2:必須為紅色
  if (redFruits.includes(fruit)) {
    console.log('red');
    
    // 條件 3:必須是大量存在
    if (quantity > 10) {
      console.log('big quantity');
    }
  }
}

這樣的程式碼比較簡短和直白,一個巢狀的 if 使得結構更加清晰。
條件反轉會導致更多的思考過程(增加認知負擔)。

因此,始終追求更少的巢狀,更早地返回,但是不要過度。


3、使用函式預設引數和解構
我們很熟悉以下的程式碼,在 JavaScript 中我們經常需要檢查 null / undefined 並賦予預設值:

function test(fruit, quantity) {
  if (!fruit) return;
  const q = quantity || 1; // 如果沒有提供 quantity,預設為 1

  console.log(`We have ${q} ${fruit}!`);
}

//測試結果
test('banana'); // We have 1 banana!
test('apple', 2); // We have 2 apple!

事實上,我們可以通過函式的預設引數來去掉變數 q。

function test(fruit, quantity = 1) { // 如果沒有提供 quantity,預設為 1
  if (!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)呢?我們還可以使用預設引數嗎?

function test(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 屬性
// 預設引數為空物件 {}
function test({name} = {}) {
  console.log (name || 'unknown');
}

//測試結果
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple

既然我們只需要 fruit 的 name 屬性,我們可以使用 {name} 來將其解構出來,之後我們
就可以在程式碼中使用 name 變數來取代 fruit.name


4、相較於 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 = new Map()
    .set('red', ['apple', 'strawberry'])
    .set('yellow', ['banana', 'pineapple'])
    .set('purple', ['grape', 'plum']);

function test(color) {
  return fruitColor.get(color) || [];
}

就以上的例子,事實上我們可以通過重構我們的程式碼,使用 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);
}

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;

  // 條件:所有的水果都必須是紅色
  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() {
  // 條件:(簡短形式)所有的水果都必須是紅色
  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() {
  // 條件:至少一個水果是紅色的
  const isAnyRed = fruits.some(f => f.color == 'red');
  console.log(isAnyRed); // true
}