JavaScript 条件判断使用技巧详解

引言

本文花费很短的时间来介绍一下在 JavaScript 中如何编写更简单的条件判断，帮助你编写更简洁的代码和可读性更高的代码。

假如我们有一个颜色值转换十六进制编码的函数。

function convertToHex(color) {
  if (typeof color === 'string') {
    if (color === 'slate') {
      return '#64748b'
    } else if (color === 'gray') {
      return '#6b7280'
    } else if (color === 'red') {
      return '#ef4444'
    } else if (color === 'orange') {
      return '#f97316'
    } else if (color === 'yellow') {
      return '#eab308'
    } else if (color === 'green') {
      return '#22c55e'
    } else {
      return '#ffffff'
    }
  } else {
    return '#ffffff'
  }
}

这个函数的作用很简单，就是传入颜色字符串，然后返回对应的十六进制，如果传入的不是一个字符串，或者什么都没传递，那么返回白色的十六进制。

接下来我们就开始对这段代码进行优化。

避免直接使用字符串作为条件

直接使用字符串作为条件有个问题，就是当我们拼写错误时，就会很尴尬。

convertToHex("salte")

为了避免这种错误，我们可以使用常量。

const Colors = {
	SLATE: 'slate',
  GRAY: 'gray',
  // ...
}
function convertToHex(color) {
  if (typeof color === 'string') {
    if (color === Colors.SLATE) {
      return '#64748b'
    } else if (color === Color.GRAY) {
      return '#6b7280'
    }
    // ...
  } else {
    return '#ffffff'
  }
}
convertToHex(Colors.SLATE)

如果你是用 typescript 的话，那么可以直接使用枚举。

使用 Object

其实从上面的代码中不难发现，我们可以直接把十六进制的值存储到对象的 value 中。

const Colors = {
	SLATE: '#64748b',
  GRAY: '#6b7280',
  // ...
}
function convertToHex(color) {
  if (color in Colors) {
    return Colors[color]
  } else {
    return '#ffffff'
  }
}
convertToHex(Colors.SLATE)

这样代码将会更加简洁易读。

不符合预期，提前 return

还有一个最佳实践是，我们可以把不符合预期的情况写到函数的最前面提前 return，避免忘记 return。

const Colors = {
	SLATE: '#64748b',
  GRAY: '#6b7280',
  // ...
}
function convertToHex(color) {
  if (!color in Colors) {
    return '#ffffff'
  }
  return Colors[color]
}
convertToHex(Colors.SLATE)

这样连 else 都不需要了。巧用这种技巧，我们可以消灭代码中大量的 else。

使用 Map 配合 Object

使用 map 更加专业，因为 map 可以存储任意类型的 key，而且它继承自 Map.prototype，具有更多方便的方法和属性。

而 Object 访问属性更加方便，我们可以继续使用 Object 实现枚举的作用。

const ColorsEnum = {
	SLATE: 'slate',
  GRAY: 'gray',
  // ...
}
const Colors = new Map()
Colors.set(ColorsEnum.SLATE, '#64748b')
Colors.set(ColorsEnum.GRAY, '#6b7280')
// ...
Colors.set('DEFAULT', '#ffffff')
function convertToHex(color) {
  if (!Colors.has(color)) {
    return Colors.get('DEFAULT')
  }
  return Colors.get(color)
}
convertToHex(Colors.SLATE)

Map 也可以存储函数

假设我们存储的颜色非常多，高达上千种，而且还要支持后端配置，通过某种运算过程才可以得到结果。

那么我们可以使用 Map 来存储函数。

return Colors.get(color)()

尽量避免三目表达式和 switch

三目表达式虽然简介，但是可读性大大降低，如果是多级条件的话，会非常难以阅读。

switch 和 if 相比没有明显优势，反而有时容易 return 导致代码不按预期执行。