如何在 JavaScript 中更好的使用条件判断?

2,999 阅读3分钟

我报名参加金石计划1期挑战——瓜分10万奖池,这是我的第1篇文章,点击查看活动详情

本文花费很短的时间来介绍一下在 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 导致代码不按预期执行。