用 JavaScript 编写枚举的最有效方法

摘要：假设有这样一个场景，我们需要统计员工的技术栈，目前我们需要标记的技术有 CSS、JavaScript、HTML、WebGL。然后我可以这样写枚举：

JavaScript 语言本身不支持枚举。如果我们想模拟枚举，我们可以使用一个对象。

假设有这样一个场景，我们需要统计一下员工的技术栈，目前我们需要标记的技术有 CSS、JavaScript、HTML、WebGL。

然后我可以这样写枚举：

const SKILLS = {
 CSS: 1 ,
 JS: 2,
 HTML: 3,
 WEB_GL: 4
}

之前是这样写的，但是最近看vue源码的时候，发现了一个高效使用枚举的技巧，在这里分享给大家。

定义枚举

我们可以这样写上面的枚举：

const SKILLS = {
 CSS: 1 ,
 JS: 1 << 1,
 HTML: 1 << 2,
 WEB_GL: 1 << 3
}

<< 是什么？

左移运算符 (<<) 将第一个操作数左移指定位数。向左移动的多余位被丢弃。零位从右侧移入。

例如：

1二进制是0000 0001，左移一位是0000 0010，十进制是2。
如果我们将其移动两位，它将变为0000 0100，即十进制的 4。
如果我们将它移动三位，它将变为0000 1000，即十进制的 8。
如果我们将它移动 N 位，它将变成2^N十进制。

用法

按照上面的方法定义好枚举后，我们可以这样使用：

const SKILLS = {
  CSS: 1 ,
  JS: 1 << 1,
  HTML: 1 << 2,
  WEB_GL: 1 << 3
}

// Use this value to store a user's tech-stack
let skills = 0

// add a skill for the user
function addSkill(skill) {
  skills = skills | skill
}

addSkill(SKILLS.CSS)
addSkill(SKILLS.JS)

// If this value is not 0, it means that the user has mastered the tech
console.log('Does he know CSS', SKILLS.CSS & skills)
console.log('Does he know JavaScript', SKILLS.JS & skills)
console.log('Does he know Web GL', SKILLS.WEB_GL & skills)

cons
t a = 5;        // 00000000000000000000000000000101
const b = 3;        // 00000000000000000000000000000011
console.log(a | b); // 00000000000000000000000000000111
// expected output: 7

如何理解这段代码？

在 JavaScript 中，整数存储在 4 个字节中，即 32 位。第一个代表正负，后面的31代表数字。

当我们用二进制表示 1 , 1 << 2 时，它们看起来像这样：

我们定义的枚举变量只有一个二进制格式的1，并且占据不同的位置。

当我们向技能添加枚举选项时，我们使用skills | skill。假设现在我们需要添加的技能是SKILLS.CSS，那么在执行过程中，就是：

我们可以发现，在技能中，SKILLS.CSS对应的位置会变成1。

反之，那么我们可以通过查看skills&SKILLS.CSS的结果是否为0来判断技能中是否存在SKILLS.CSS。

顺便说一句，这里我们也可以发现这个技巧有个缺点，就是枚举项不能超过 31 个。

我们为什么要使用这个技巧？

答案很简单，这样的代码运行起来更高效。CPU中有直接对应位操作的指令，因此效率更高。

我们也可以做一个性能测试。

如果我们不使用按位运算，而是使用传统的方法（数组或映射）来实现，那么代码如下。

Array 方法：

const SKILLS = {
 CSS: 1 ,
 JS: 1 << 1,
 HTML: 1 << 2,
 WEB_GL: 1 << 3
}
// Use an array to store the user's tech-stack
let skills = []
function addSkill(skill) {
 if (!skills.includes(skill)) { // Avoid duplicate storage
   skills.push(skill)
 }
}
addSkill(SKILLS.CSS)
addSkill(SKILLS.JS)
skills.includes(SKILLS.CSS)
skills.includes(SKILLS.JS)
skills.includes(SKILLS.WEB_GL)

Map 方法：

const SKILLS = {
 CSS: 1 ,
 JS: 1 << 1,
 HTML: 1 << 2,
 WEB_GL: 1 << 3
}
// Use a map to store the user's tech-stack
let skills = {}
function addSkill(skill) {
 if (!skills[skill]) {
   skills[skill] = true
 }
}
addSkill(SKILLS.CSS)
addSkill(SKILLS.JS)
skills[SKILLS.CSS]
skills[SKILLS.JS]
skills[SKILLS.WEB_GL]

以下是jsbench.me 的性能测试：

使用按位枚举，性能明显更高。

学习Vue源码

我是从 Vue 源代码中学到的。

export const enum ShapeFlags {
 ELEMENT = 1,
 FUNCTIONAL_COMPONENT = 1 << 1,
 STATEFUL_COMPONENT = 1 << 2,
 TEXT_CHILDREN = 1 << 3,
 ARRAY_CHILDREN = 1 << 4,
 SLOTS_CHILDREN = 1 << 5,
 TELEPORT = 1 << 6,
 SUSPENSE = 1 << 7,
 COMPONENT_SHOULD_KEEP_ALIVE = 1 << 8,
 COMPONENT_KEPT_ALIVE = 1 << 9,
 COMPONENT = ShapeFlags.STATEFUL_COMPONENT | ShapeFlags.FUNCTIONAL_COMPONENT
}

地址：https://github1s.com/vuejs/core/blob/HEAD/packages/shared/src/shapeFlags.ts

第一次看这段代码，有点迷茫，但总算弄明白了。希望这个技巧能帮到你。谢谢阅读。

翻译来源：https://betterprogramming.pub/the-most-efficient-way-to-write-enumerations-in-javascript-a1b9f41ea651

本文内容仅供个人学习、研究或参考使用，不构成任何形式的决策建议、专业指导或法律依据。未经授权，禁止任何单位或个人以商业售卖、虚假宣传、侵权传播等非学习研究目的使用本文内容。如需分享或转载，请保留原文来源信息，不得篡改、删减内容或侵犯相关权益。感谢您的理解与支持！

链接: https://shenqiku.cn/article/FLY_11787