mm_machine 2.9.3 → 2.9.5

package/id_gen.js

@@ -0,0 +1,138 @@
+/**
+ * IdGen 雪花算法 ID 生成器
+ * 基于雪花算法（Snowflake）的分布式唯一 ID 生成器，生成 64 位整数 ID
+ */
+class IdGen {
+  /**
+   * 构造函数
+   * @param {object} config 配置参数
+   * @param {number} config.data_center_no 数据中心编号 (0-31)
+   * @param {number} config.machine_no 机器编号 (0-31)
+   * @param {number} [config.epoch=1700000000000] 自定义起始时间戳（毫秒），默认 2023-11-15 00:00:00 UTC
+   * @param {number} [config.sequence=0] 初始序列号
+   */
+  constructor(config = {}) {
+    this.data_center_no = config.data_center_no || config.dc_id || 0;
+    this.machine_no = config.machine_no || config.worker_id || 0;
+    this.sequence = config.sequence || 0;
+    this.epoch = config.epoch || 1700000000000;
+    this.last_timestamp = -1;
+
+    this._initBits();
+  }
+}
+
+/**
+ * 获取当前时间戳
+ * @returns {number} 当前时间戳
+ */
+IdGen.prototype._timeGen = function () {
+  return Date.now();
+};
+
+/**
+ * 等待下一毫秒
+ * @param {number} last_timestamp 上次时间戳
+ * @returns {number} 新的时间戳
+ */
+IdGen.prototype._tilNextMillis = function (last_timestamp) {
+  var timestamp = this._timeGen();
+  while (timestamp <= last_timestamp) {
+    timestamp = this._timeGen();
+  }
+  return timestamp;
+};
+
+/**
+ * 初始化位运算参数
+ * @private
+ */
+IdGen.prototype._initBits = function () {
+  this.dc_id_bits = 5;
+  this.worker_id_bits = 5;
+  this.sequence_bits = 12;
+
+  this.max_dc_id = -1 ^ (-1 << this.dc_id_bits);
+  this.max_worker_id = -1 ^ (-1 << this.worker_id_bits);
+  this.sequence_mask = -1 ^ (-1 << this.sequence_bits);
+
+  this.worker_id_shift = this.sequence_bits;
+  this.dc_id_shift = this.sequence_bits + this.worker_id_bits;
+  this.timestamp_left_shift = this.sequence_bits + this.worker_id_bits + this.dc_id_bits;
+
+  if (this.data_center_no > this.max_dc_id || this.data_center_no < 0) {
+    throw new TypeError('数据中心编号必须在0-' + this.max_dc_id + '之间');
+  }
+  if (this.machine_no > this.max_worker_id || this.machine_no < 0) {
+    throw new TypeError('机器编号必须在0-' + this.max_worker_id + '之间');
+  }
+};
+
+/**
+ * 生成下一个ID
+ * @returns {string} 唯一ID
+ */
+IdGen.prototype.nextId = function () {
+  var timestamp = this._timeGen();
+
+  if (timestamp < this.last_timestamp) {
+    throw new Error('时钟回拨，拒绝生成ID');
+  }
+
+  if (this.last_timestamp === timestamp) {
+    this.sequence = (this.sequence + 1) & this.sequence_mask;
+    if (this.sequence === 0) {
+      timestamp = this._tilNextMillis(this.last_timestamp);
+    }
+  } else {
+    this.sequence = 0;
+  }
+
+  this.last_timestamp = timestamp;
+
+  var id =
+    (BigInt(timestamp - this.epoch) << BigInt(this.timestamp_left_shift)) |
+    (BigInt(this.data_center_no) << BigInt(this.dc_id_shift)) |
+    (BigInt(this.machine_no) << BigInt(this.worker_id_shift)) |
+    BigInt(this.sequence);
+
+  return id.toString();
+};
+
+/**
+ * 生成带前缀的ID
+ * @param {string} prefix 前缀
+ * @returns {string} 带前缀的唯一ID
+ */
+IdGen.prototype.generateId = function (prefix) {
+  if (typeof prefix !== 'string') {
+    throw new TypeError('前缀必须是字符串');
+  }
+  return prefix + '_' + this.nextId();
+};
+
+/**
+ * 解析ID信息
+ * @param {string} id ID字符串
+ * @returns {object} ID信息对象
+ */
+IdGen.prototype.parseId = function (id) {
+  if (typeof id !== 'string') {
+    throw new TypeError('ID必须是字符串');
+  }
+  var num = BigInt(id.replace(/^[a-zA-Z_]+_/, ''));
+  var timestamp = Number((num >> 22n) + BigInt(this.epoch));
+  var dc_id = Number((num >> 17n) & 31n);
+  var worker_id = Number((num >> 12n) & 31n);
+  var sequence = Number(num & 4095n);
+
+  return {
+    timestamp: timestamp,
+    datetime: new Date(timestamp).toISOString(),
+    data_center_no: dc_id,
+    machine_no: worker_id,
+    sequence: sequence
+  };
+};
+
+module.exports = { IdGen };