写好 JS 条件语句的 5 条守则--摘自前端大学

在用 JavaScript 工做时，咱们常常和条件语句打交道，这里有5条让你写出更好/干净的条件语句的建议。

目录：

1.多重判断时使用 Array.includes

2.更少的嵌套，尽早 return

3.使用默认参数和解构

4.倾向于遍历对象而不是 Switch 语句

5.对 全部/部分 判断使用 Array.every & Array.some

6.总结

1.多重判断时使用 Array.includes

让咱们看一下下面这个例子:

// condition
function test(fruit) {
  if (fruit == 'apple' || fruit == 'strawberry') {
    console.log('red');
  }
}

 

第一眼，上面这个例子看起来没问题。若是咱们有更多名字叫 cherry 和 cranberries 的红色水果呢？咱们准备用更多的 || 来拓展条件语句吗？

咱们能够用 Array.includes (Array.includes)重写条件语句。

function test(fruit) {
  const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];

  if (redFruits.includes(fruit)) {
    console.log('red');
  }
}

 

咱们把红色的水果(red fruits)这一判断条件提取到一个数组。这样一来，代码看起来更整洁。

2.更少的嵌套，尽早 Return

让咱们拓展上一个例子让它包含两个条件。

• 若是没有传入参数 fruit，抛出错误

• 接受 quantity 参数，而且在 quantity 大于 10 时打印出来

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!');
  }
}

// 测试结果
test(null); // error: No fruits
test('apple'); // print: red
test('apple', 20); // print: red, big quantity

 

在上面的代码, 咱们有:

 

• 1个 if/else 语句筛选出无效的语句

• 3层if嵌套语句 (条件 1, 2 & 3)

 

我我的遵循的规则通常是在发现无效条件时，尽早Return。

/_ 当发现无效语句时，尽早Return _/

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语句的时候(想象你须要滚动到最底层才知道还有else语句，这并不酷)

 

咱们能够经过 倒置判断条件 & 尽早return 进一步减小if嵌套。看下面咱们是怎么处理判断 条件2 的:

/_ 当发现无效语句时，尽早Return _/

function test(fruit, quantity) {
  const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];

  // 条件 1: 尽早抛出错误
  if (!fruit) throw new Error('No fruit!');
  // 条件 2: 当水果不是红色时中止继续执行
  if (!redFruits.includes(fruit)) return; 

  console.log('red');

  // 条件 3: 必须是大质量的
  if (quantity > 10) {
    console.log('big quantity');
  }
}

 

经过倒置判断条件2，咱们的代码避免了嵌套语句。这个技巧在咱们须要进行很长的逻辑判断时是很是有用的，特别是咱们但愿可以在条件不知足时可以中止下来进行处理。

 

并且这么作并不困难。问问本身，这个版本(没有嵌套)是否是比以前的(两层条件嵌套)更好，可读性更高？

 

但对于我，我会保留先前的版本(包含两层嵌套)。这是由于:

 

• 代码比较短且直接，包含if嵌套的更清晰

• 倒置判断条件可能加剧思考的负担(增长认知载荷)

 

所以，应当尽力减小嵌套和尽早return，但不要过分。若是你感兴趣的话，能够看一下关于这个话题的一篇文章和 StackOverflow 上的讨论。

 

• Avoid Else, Return Early by Tim Oxley

• StackOverflow discussion on if/else coding style

3.使用默认参数和解构

我猜下面的代码你可能会熟悉，在JavaScript中咱们老是须要检查 null / undefined的值和指定默认值:

function test(fruit, quantity) {
  if (!fruit) return;
  // 若是 quantity 参数没有传入，设置默认值为 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!

 实际上，咱们能够经过声明 默认函数参数 来消除变量 q。

function test(fruit, quantity = 1) {
  // 若是 quantity 参数没有传入，设置默认值为 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!

 

这更加直观，不是吗？注意，每一个声明都有本身的默认参数.

 

例如，咱们也能给fruit分配默认值:function test(fruit = 'unknown', quantity = 1)。

 

若是fruit是一个object会怎么样？咱们能分配一个默认参数吗？

function test(fruit) { 
  // 当值存在时打印 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

 看上面这个例子，咱们想打印 fruit 对象中可能存在的 name 属性。不然咱们将打印unknown。咱们能够经过默认参数以及解构从而避免判断条件 fruit && fruit.name

// 解构 - 仅仅获取 name 属性
// 为其赋默认值为空对象
function test({name} = {}) {
  console.log (name || 'unknown');
}

// test results
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple

 

因为咱们只须要 name 属性，咱们能够用 {name} 解构出参数，而后咱们就能使用变量 name 代替 fruit.name。

 

咱们也须要声明空对象 {} 做为默认值。若是咱们不这么作，当执行 test(undefined) 时，你将获得一个没法对 undefined 或 null 解构的的错误。由于在 undefined 中没有 name 属性。

 

若是你不介意使用第三方库，这有一些方式减小null的检查:

 

• 使用 Lodash get函数

• 使用Facebook开源的idx库(with Babeljs)

 

这是一个使用Lodash的例子:

function test(fruit) {
  // 获取属性名，若是属性名不可用，赋默认值为 unknown
  console.log(__.get(fruit, 'name', 'unknown'); 
}

// test results
test(undefined); // unknown
test({ }); // unknown
test({ name: 'apple', color: 'red' }); // apple

 你能够在jsbin运行demo代码。除此以外，若是你是函数式编程的粉丝，你可能选择使用 Lodash fp，Lodash的函数式版本(方法变动为get或者getOr)。

4.倾向于对象遍历而不是Switch语句

让咱们看下面这个例子，咱们想根据 color 打印出水果:

function test(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']

 上面的代码看起来没有错误，可是我找到了一些累赘。用对象遍历实现相同的结果，语法看起来更简洁:

const fruitColor = {
  red: ['apple', 'strawberry'],
  yellow: ['banana', 'pineapple'],
  purple: ['grape', 'plum']
};

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

 或者你也可使用 Map实现相同的结果:

  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 规范以后实现的对象类型，容许你存储 key 和 value 的值。

 

但咱们是否应当禁止switch语句的使用呢？答案是不要限制你本身。从我的来讲，我会尽量的使用对象遍历，但我并不严格遵照它，而是使用对当前的场景更有意义的方式。

 

Todd Motto有一篇关于 switch 语句对比对象遍历的更深刻的文章，你能够在这个地方阅读

TL;DR; 重构语法

在上面的例子，咱们可以用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) {
  return fruits.filter(f => f.color == color);
}

 

有着不止一种方法可以实现相同的结果，咱们以上展现了 4 种。

5.对 全部/部分 判断使用Array.every & Array.some

这最后一个建议更可能是关于利用 JavaScript Array 的内置方法来减小代码行数。看下面的代码，咱们想要检查是否全部水果都是红色:

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
}

 

6.总结

让咱们一块儿生产更多可读性高的代码。我但愿你能从这篇文章学到东西。