写好JavaScript条件语句的5个技巧

786 阅读6分钟
原文链接: my.oschina.net

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的关于这主题的更多讨论:

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
}

 

总结

让我们一起生成更多可读代码。 我希望你能在本文中学到一些新东西。
就这样。 写代码快乐!
(完)