优化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)
就我个人而言,我遵循的一个总的规则是当发现无效条件时尽早返回。
/_<font color=red> 当发现无效条件时尽早返回</font> _/
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 使得结构更加清晰。
条件反转会导致更多的思考过程(增加认知负担)。
因此,始终追求<font color=red>更少的嵌套,更早地返回</font>,但是不要过度。
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
}