在使用 JavaScript 时,我们常常要写不少的条件语句。这里有五个小技巧,可以让你写出更干净、漂亮的条件语句。
1. 使用 Array.includes 来处理多重条件
举个栗子:
// 条件语句
function test(fruit) {
if (fruit == 'apple' || fruit == 'strawberry') {
console.log('red');
}
}
乍一看,这么写似乎没什么大问题。然而,如果我们想要匹配更多的红色水果呢,比方说『樱桃』和『蔓越莓』?我们是不是得用更多的 || 来扩展这条语句?
我们可以使用 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 {
thrownewError('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) thrownewError('No fruit!');
// 条件2:必须为红色
if (redFruits.includes(fruit)) {
console.log('red');
// 条件 3:必须是大量存在
if (quantity > 10) {
console.log('big quantity');
}
}
}
如此一来,我们就少写了一层嵌套。这是种很好的代码风格,尤其是在 if 语句很长的时候(试想一下,你得滚动到底部才能知道那儿还有个 else 语句,是不是有点不爽)。
如果反转一下条件,我们还可以进一步地减少嵌套层级。注意观察下面的条件 2 语句,看看是如何做到这点的:
/_ 当发现无效条件时尽早返回 _/
function test(fruit, quantity) {
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
if (!fruit) thrownewError('No fruit!'); // 条件 1:尽早抛出错误
if (!redFruits.includes(fruit)) return; // 条件 2:当 fruit 不是红色的时候,直接返回
console.log('red');
// 条件 3:必须是大量存在
if (quantity > 10) {
console.log('big quantity');
}
}
通过反转条件 2 的条件,现在我们的代码已经没有嵌套了。当我们代码的逻辑链很长,并且希望当某个条件不满足时不再执行之后流程时,这个技巧会很好用。
然而,并没有任何硬性规则要求你这么做。这取决于你自己,对你而言,这个版本的代码(没有嵌套)是否要比之前那个版本(条件 2 有嵌套)的更好、可读性更强?
是我的话,我会选择前一个版本(条件 2 有嵌套)。原因在于:
这样的代码比较简短和直白,一个嵌套的 if 使得结构更加清晰。
条件反转会导致更多的思考过程(增加认知负担)。
因此,始终追求更少的嵌套,更早地返回,但是不要过度。感兴趣的话,这里有篇关于这个问题的文章以及 StackOverflow 上的讨论:
Avoid Else, Return Early by Tim Oxley
StackOverflow discussion on if/else coding style
3. 相较于 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 = newMap()
.set('red', ['apple', 'strawberry'])
.set('yellow', ['banana', 'pineapple'])
.set('purple', ['grape', 'plum']);
function test(color) {
return fruitColor.get(color) || [];
}
Map 是 ES2015 引入的新的对象类型,允许你存放键值对。
那是不是说我们应该禁止使用 switch 语句? 别把自己限制住。我自己会在任何可能的时候使用对象字面量,但是这并不是说我就不用 switch,这得视场景而定。
Todd Motto 有一篇文章深入讨论了 switch 语句和对象字面量,你也许会想看看。
懒人版:重构语法
就以上的例子,事实上我们可以通过重构我们的代码,使用 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);
}
解决问题的方法永远不只一种。对于这个例子我们展示了四种实现方法。Coding is fun!
总结
让我们一起写出可读性更高的代码吧。希望这篇文章能给你们带来一些帮助。
就是这样啦。Happy coding!
关于本文
译者:@Hopsken
译文:
https://juejin.im/post/5bb9e3085188255c352d7326(掘金翻译)
作者:@Jecelyn Yeen
原文:
https://scotch.io/tutorials/5-tips-to-write-better-conditionals-in-javascript
校对者:@Park-ma、@ThomasWhyne