forked from cloudyan/snowflake
/
snowflake.js
160 lines (144 loc) · 6.04 KB
/
snowflake.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
// import bigInt from 'big-integer';
// http://peterolson.github.io/BigInteger.js/
// https://blog.csdn.net/xiaopeng9275/article/details/72123709
// https://zhuanlan.zhihu.com/p/36330307
// https://segmentfault.com/a/1190000011282426
/**
* Twitter_Snowflake
*
* SnowFlake的结构如下(共64bits,每部分用-分开):
* 0 - 0000000000 0000000000 0000000000 0000000000 0 - 00000 - 00000 - 000000000000
* | ----------------------|---------------------- --|-- --|-- -----|------
* 1bit不用 41bit 时间戳 数据标识id 机器id 序列号id
*
* - 1位标识,二进制中最高位为1的都是负数,但是我们生成的id一般都使用整数,所以这个最高位固定是0
* - 41位时间截(毫秒级),注意,41位时间截不是存储当前时间的时间截,而是存储时间截的差值(当前时间截 - 开始时间截得到的值),这里的的开始时间截,一般是我们的id生成器开始使用的时间,由我们程序来指定的(如下下面程序IdWorker类的startTime属性)。41位的时间截,可以使用69年,年T = (1L << 41) / (1000L * 60 * 60 * 24 * 365) = 69
* - 10位的数据机器位,可以部署在1024个节点,包括5位datacenterId和5位workerId
* - 12位序列,毫秒内的计数,12位的计数顺序号支持每个节点每毫秒(同一机器,同一时间截)产生4096个ID序号
* - 加起来刚好64位,为一个Long型。
* SnowFlake的优点是
* - 整体上按照时间自增排序
* - 并且整个分布式系统内不会产生ID碰撞(由数据中心ID和机器ID作区分)
* - 并且效率较高,经测试,SnowFlake每秒能够产生26万ID左右。
*/
export default class SnowFlake {
/**
* 构造函数
* 运行在内阁
* @param {bigint} workerId 工作ID (0~31)
* @param {bigint} dataCenterId 数据中心ID (0~31)
* @param {bigint} sequence 毫秒内序列 (0~4095)
*/
constructor(workerId, dataCenterId) {
// 开始时间截 (2018-02-01),这个可以设置开始使用该系统的时间,可往后使用69年
this.twepoch = 1548988646430n;
// 位数划分 [数据标识id(5bit 31)、机器id(5bit 31)](合计共支持1024个节点)、序列id(12bit 4095)
this.workerIdBits = 5n;
this.dataCenterIdBits = 5n;
this.sequenceBits = 12n;
// this.timestampBits = 41n;
// 支持的最大十进制id
// 这个移位算法可以很快的计算出几位二进制数所能表示的最大十进制数
// -1 左移5位后与 -1 异或
this.maxWorkerId = -1n ^ (-1n << this.workerIdBits);
this.maxDataCenterId = -1n ^ (-1n << this.dataCenterIdBits);
// 生成序列的掩码,这里为4095 (0b111111111111=0xfff=4095)
this.sequenceMask = -1n ^ (-1n << this.sequenceBits);
// 机器ID向左移12位 数据标识id向左移17位(12+5) 时间截向左移22位(5+5+12)
this.workerIdShift = this.sequenceBits;
this.dataCenterIdShift = this.sequenceBits + this.workerIdBits;
this.timestampLeftShift = this.dataCenterIdShift + this.dataCenterIdBits;
// 工作机器ID(0~31) 数据中心ID(0~31) 毫秒内序列(0~4095)
// this.workerId;
// this.dataCenterId;
this.sequence = 0n;
// 上次生成ID的时间截(这个是在内存中?系统时钟回退+重启后呢)
this.lastTimestamp = -1n;
const { maxWorkerId, maxDataCenterId } = this;
if (workerId > maxWorkerId || workerId < 0n) {
throw new Error(
`workerId can't be greater than ${maxWorkerId} or less than 0`
);
}
if (dataCenterId > maxDataCenterId || dataCenterId < 0n) {
throw new Error(
`dataCenterId can't be greater than ${maxDataCenterId} or less than 0`
);
}
this.workerId = workerId;
this.dataCenterId = dataCenterId;
return this;
}
/**
* 获得下一个ID (该方法是线程安全的)
*
* @returns {bigint} SnowflakeId 返回 id
*/
nextId() {
let timestamp = this.timeGen();
// 如果当前时间小于上一次ID生成的时间戳,说明系统时钟回退过这个时候应当抛出异常
const diff = timestamp - this.lastTimestamp;
if (diff < 0n) {
throw new Error(
`Clock moved backwards. Refusing to generate id for ${-diff} milliseconds`
);
}
// 如果是同一时间生成的,则进行毫秒内序列
if (diff === 0n) {
this.sequence = (this.sequence + 1n) & this.sequenceMask;
// 毫秒内序列溢出
if (this.sequence === 0n) {
// 阻塞到下一个毫秒,获得新的时间戳
timestamp = this.tilNextMillis(this.lastTimestamp);
}
} else {
// 时间戳改变,毫秒内序列重置
this.sequence = 0n;
}
// 保存上次生成ID的时间截
this.lastTimestamp = timestamp;
// 移位并通过或运算拼到一起组成64位的ID
// 将各 bits 位数据移位后或运算合成一个大的64位二进制数据
return (
((timestamp - this.twepoch) << this.timestampLeftShift) | // 时间数据左移22
(this.dataCenterId << this.dataCenterIdShift) | // 数据标识id左移 17
(this.workerId << this.workerIdShift) | // 机器id左移 12
this.sequence
);
}
/**
* 阻塞到下一个毫秒,直到获得新的时间戳
* @param {bigint} lastTimestamp 上次生成ID的时间截
* @return {bigint} 当前时间戳
*/
tilNextMillis(lastTimestamp) {
let timestamp = this.timeGen();
while (timestamp <= lastTimestamp) {
timestamp = this.timeGen();
}
return timestamp;
}
/**
* 返回以毫秒为单位的当前时间
* @return {bigint} 当前时间(毫秒)
*/
timeGen() {
return BigInt(+new Date());
}
}
// ========== TEST ==========
// 测试
// function main() {
// console.time('id');
// const idWorker = new SnowFlake(1n, 1n);
// const tempIds = [];
// for (let i = 0; i < 50000; i++) {
// const id = idWorker.nextId();
// // console.log(id);
// tempIds.push(id);
// }
// console.log(tempIds.length);
// // const end = +new Date();
// console.timeEnd('id');
// }
// main();