JS代码优化——逻辑判断

安迪小宝

于 2023-05-15 11:45:20 发布

阅读量462

点赞数

分类专栏：前端文章标签： javascript 前端 node.js

本文链接：https://blog.csdn.net/wujianyouhun/article/details/130681425

版权

前端专栏收录该内容

14 篇文章 0 订阅

订阅专栏

文章目录

- JavaScript 语法篇

JavaScript 语法篇

嵌套层级优化

function supply(fruit, quantity) {
  const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
  // 条件 1: 水果存在
  if(fruit) {
    // 条件 2: 属于红色水果
    if(redFruits.includes(fruit)) {
      console.log('红色水果');
      // 条件 3: 水果数量大于 10 个
      if (quantity > 10) {
        console.log('数量大于 10 个');
      }
    }
  } else {
    throw new Error('没有水果啦!');
  }
}

分析上面的条件判断，存在三层 if 条件嵌套。

如果提前 return 掉无效条件，将 if else的多重嵌套层次减少到一层，更容易理解和维护。

function supply(fruit, quantity) {
  const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
  if(!fruit) throw new Error('没有水果啦'); // 条件 1: 当 fruit 无效时，提前处理错误
  if(!redFruits.includes(fruit)) return; // 条件 2: 当不是红色水果时，提前 return

  console.log('红色水果');

  // 条件 3: 水果数量大于 10 个
  if (quantity > 10) {
    console.log('数量大于 10 个');
  }
}

多条件分支的优化处理

当需要枚举值处理不同的业务分支逻辑时，第一反应是写下 if else ？我们来看一下：

function pick(color) {
  // 根据颜色选择水果
  if(color === 'red') {
    return ['apple', 'strawberry']; 
  } else if (color === 'yellow') {
    return ['banana', 'pineapple'];
  } else if (color === 'purple') {
    return ['grape', 'plum'];
  } else {
    return [];
  }
}

在上面的实现中：

if else 分支太多
if else 更适合于条件区间判断，而 switch case 更适合于具体枚举值的分支判断

使用 switch case 优化上面的代码后：

function pick(color) {
  // 根据颜色选择水果
  switch (color) {
    case 'red':
      return ['apple', 'strawberry'];
    case 'yellow':
      return ['banana', 'pineapple'];
    case 'purple':
      return ['grape', 'plum'];
    default:
      return [];
  }
}

switch case 优化之后的代码看上去格式整齐，思路很清晰，但还是很冗长。继续优化：

借助 Object 的 { key: value } 结构，我们可以在 Object 中枚举所有的情况，然后将 key 作为索引，直接通过 Object.key 或者 Object[key] 来获取内容

const fruitColor = {                                                                        
  red: ['apple', 'strawberry'],
  yellow: ['banana', 'pineapple'],
  purple: ['grape', 'plum'],
}
function pick(color) {
  return fruitColor[color] || [];
}

使用 Map 数据结构，真正的 (key, value) 键值对结构；

const fruitColor = new Map()
.set('red', ['apple', 'strawberry'])
.set('yellow', ['banana', 'pineapple'])
.set('purple', ['grape', 'plum']);

function pick(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 pick(color) {
  return fruits.filter(f => f.color == color);
}

使用数组新特性简化逻辑判断

巧妙的利用 ES6 中提供的数组新特性，也可以让我们更轻松的处理逻辑判断。

多条件判断

编码时遇到多个判断条件时，本能的写下下面的代码（其实也是最能表达业务逻辑的面向过程编码）。

function judge(fruit) {
  if (fruit === 'apple' || fruit === 'strawberry' || fruit === 'cherry' || fruit === 'cranberries' ) {
    console.log('red');
  }
}

但是当 type 未来到 10 种甚至更多时，我们只能继续添加 || 来维护代码么 ?

试试 Array.includes ~

// 将判断条件抽取成一个数组
const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
function judge(type) {
  if (redFruits.includes(fruit)) {
    console.log('red');
  }
}

判断数组中是否所有项都满足某条件

const fruits = [
  { name: 'apple', color: 'red' },
  { name: 'banana', color: 'yellow' },
  { name: 'grape', color: 'purple' }
];

function match() {
  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 match() {
  // 条件：所有水果都必须是红色
  const isAllRed = fruits.every(f => f.color == 'red');

  console.log(isAllRed); // false
}

判断数组中是否有某一项满足条件

Array.some ，它主要处理的场景是判断数组中是否有一项满足条件。

如果想知道是否有红色水果，可以直接使用 Array.some 方法：

const fruits = [
  { name: 'apple', color: 'red' },
  { name: 'banana', color: 'yellow' },
  { name: 'grape', color: 'purple' }
];

// 条件：是否有红色水果 
const isAnyRed = fruits.some(f => f.color == 'red');

还有许多其他数组新特性，比如 Array.find、Array.slice、Array.findIndex、Array.reduce、Array.splice 等，在实际场景中可以根据需要选择使用。

函数默认值

使用默认参数

const buyFruit = (fruit,amount) => {
  if(!fruit){
    return
  }
  amount = amount || 1;
  console.log(amount)
}

我们经常需要处理函数内部的一些参数默认值，上面的代码大家都不陌生，使用函数的默认参数，可以很好的帮助处理这种场景。

const buyFruit = (fruit,amount = 1) => {
  if(!fruit){
    return
  }
  console.log(amount,'amount')
}

我们可以通过 Babel 的转译来看一下默认参数是如何实现的。

从上面的转译结果可以发现，只有参数为 undefined 时才会使用默认参数。

测试的执行结果如下：

buyFruit('apple','');  // amount
buyFruit('apple',null);  //null amount
buyFruit('apple');  //1 amount

所以使用默认参数的情况下，我们需要注意的是默认参数 amount=1 并不等同于 amount || 1。

使用解构与默认参数

当函数参数是对象时，我们可以使用解构结合默认参数来简化逻辑。

Before:

const buyFruit = (fruit,amount) => {
    fruit = fruit || {};
    if(!fruit.name || !fruit.price){
        return;
    }
    ...
  amount = amount || 1;
  console.log(amount)
}

After:

const buyFruit = ({ name, price }={},amount) => {
  if(!name || !prices){
      return;
  }
  console.log(amount)
}

复杂数据解构

当处理比较简的对象时，解构与默认参数的配合是非常好的，但在一些复杂的场景中，我们面临的可能是更复杂的结构。

const oneComplexObj = {
    firstLevel:{
        secondLevel:[{
            name: "",
            price: ""
        }]
    }
}

这个时候如果再通过解构去获取对象里的值。

const {
  firstLevel:{
    secondLevel: [{name, price]=[]
  }={}
} = oneComplexObj;

可读性就会比较差，而且需要考虑多层解构的默认值以及数据异常情况。

这种情况下，如果项目中使用 lodash 库，可以使用其中的 lodash/get 方法。

import lodashGet from 'lodash/get';

const { name, price} = lodashGet(oneComplexObj,'firstLevel.secondLevel[0]',{});

策略模式优化分支逻辑处理

策略模式：定义一系列的算法，把它们一个个封装起来，并且使它们可相互替换。

使用场景：策略模式属于对象行为模式，当遇到具有相同行为接口、行为内部不同逻辑实现的实例对象时，可以采用策略模式；或者是一组对象可以根据需要动态的选择几种行为中的某一种时，也可以采用策略模式；这里以第二种情况作为示例：

Before:

const TYPE = {
  JUICE:'juice',
  SALAD:'salad',
  JAM:'jam'
}
function enjoy({type = TYPE.JUICE,fruits}){
  if(!fruits || !fruits.length) {
    console.log('请先采购水果！');
    return;
  }
  if(type === TYPE.JUICE) {
    console.log('榨果汁中...');
    return '果汁';
  }
  if(type === TYPE.SALAD) {
    console.log('做沙拉中...');
    return '拉沙';
  }
  if(type === TYPE.JAM) {
    console.log('做果酱中...');
    return '果酱';
  }
  return;
}

enjoy({type:'juice',fruits});

使用思路：定义策略对象封装不同行为、提供策略选择接口，在不同的规则时调用相应的行为。

After：

const TYPE = {
  JUICE:'juice',
  SALAD:'salad',
  JAM:'jam'
}

const strategies = {
  [TYPE.JUICE]: function(fruits){
    console.log('榨果汁中...');
    return '果汁';
  },
  [TYPE.SALAD]:function(fruits){
    console.log('做沙拉中...');
    return '沙拉';
  },
  [TYPE.JAM]:function(fruits){
    console.log('做果酱中...');
    return '果酱';
  },
}

function enjoy({type = TYPE.JUICE,fruits}) {
  if(!type) {
    console.log('请直接享用！');
    return;
  }
  if(!fruits || !fruits.length) {
    console.log('请先采购水果！');
    return;
  }
  return strategies[type](fruits);
}

enjoy({type: 'juice',fruits});

sole.log(‘做沙拉中…’);
return ‘沙拉’;
},
[TYPE.JAM]:function(fruits){
console.log(‘做果酱中…’);
return ‘果酱’;
},
}

function enjoy({type = TYPE.JUICE,fruits}) {
if(!type) {
console.log(‘请直接享用！’);
return;
}
if(!fruits || !fruits.length) {
console.log(‘请先采购水果！’);
return;
}
return strategiestype;
}

enjoy({type: ‘juice’,fruits});