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