博客

编写高质量可维护的代码之优化逻辑判断

if else、switch case 是日常开发中最常见的条件判断语句，这种看似简单的语句，当遇到复杂的业务场景时，如果处理不善，就会出现大量的逻辑嵌套，可读性差并且难以扩展。

下面我们会分别从 JavaScript 语法和 React JSX 语法两个方面来分享一些优化的技巧。

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: ['