参考:
Twitter 官方 Scala 原版实现
https://github.com/twitter-archive/snowflake
package com.app.main.snowflake;
/**
* Created with IDEA
* author:Dingsheng Huang
* Date:2019/8/4
* Time:下午4:28
*
* Twitter_Snowflake<br>
* SnowFlake的结构如下(每部分用-分开):<br>
* 0 - 0000000000 0000000000 0000000000 0000000000 0 - 00000 - 00000 - 000000000000 <br>
* 1位标识,由于long基本类型在Java中是带符号的,最高位是符号位,正数是0,负数是1,所以id一般是正数,最高位是0<br>
* 41位时间截(毫秒级),注意,41位时间截不是存储当前时间的时间截,而是存储时间截的差值(当前时间截 - 开始时间截)
* 得到的值),这里的的开始时间截,一般是我们的id生成器开始使用的时间,由我们程序来指定的(如下下面程序IdWorker类的startTime属性)。41位的时间截,可以使用69年,年T = (1L << 41) / (1000L * 60 * 60 * 24 * 365) = 69<br>
* 10位的数据机器位,可以部署在1024个节点,包括5位datacenterId和5位workerId<br>
* 12位序列,毫秒内的计数,12位的计数顺序号支持每个节点每毫秒(同一机器,同一时间截)产生4096个ID序号<br>
* 加起来刚好64位,为一个Long型。<br>
* SnowFlake的优点是,整体上按照时间自增排序,并且整个分布式系统内不会产生ID碰撞(由数据中心ID和机器ID作区分),并且效率较高,经测试,SnowFlake每秒能够产生26万ID左右。
*/
public class SnowflakeIdWorker {
// 起始时间戳 2019-08-04 15:57:17
private final long twepoch = 1564905437000L;
// 数据中心id 所占位数
private final long datacenterIdBits = 5L;
// 机器id 所占位数
private final long workerIdBits = 5L;
// 支持的最大机器 id (通过移位运算 快速得出 n 位 二进制数所能表示的最大十进制数, 这里等同于 "2 的5次方-1", 也就是31)
private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
// 支持的最大数据中心 id
private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
// 序列号所占位数
private final long sequenceBits = 12L;
// 机器 id 左移位数(序列号12)
private final long workIdShift = sequenceBits;
// 数据中心 id 左移位数 (序列号12 + 机器id 5)
private final long datacenterIdShift = sequenceBits + workerIdBits;
// 时间戳左移位数(序列号12 + 机器 id 5 + 数据中心 id 5)
private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
// 生成序列的掩码,这里为4095
private final long sequenceMask = -1L ^ (-1L << sequenceBits);
// 机器 id
private long workId;
// 数据中心 id
private long datacenterId;
// 序列号
private long sequence = 0L;
// 上一次生成 id 的时间戳
private long lastTimestamp = -1L;
// construct
public SnowflakeIdWorker(long workId, long datacenterId) {
// 边界检查
if (workId > maxWorkerId || workId < 0) {
throw new IllegalArgumentException("work id can not be greater than " + maxWorkerId + "or less than 0");
}
if (datacenterId > maxDatacenterId || datacenterId < 0) {
throw new IllegalArgumentException("data centerId can not be greater than " + maxDatacenterId + "or less than 0");
}
this.workId = workId;
this.datacenterId = datacenterId;
}
public synchronized long nextId() {
// 获取当前时间
long timestamp = timeGen();
// 如果当前时间小于上一次 ID 生成的时间戳, 说明系统发生了时钟回拨
if (timestamp < lastTimestamp) {
throw new IllegalArgumentException("clock moved backwords . currTimestamp: " + timestamp + "lastTimestamp: " + lastTimestamp);
}
// 如果是同一时间生成的,走毫秒内序列增长
if (lastTimestamp == timestamp) {
// 获取序列号,防止序列号溢出
sequence = (sequence + 1) & sequenceMask;
// 当前毫秒内序列号溢出
if (sequence == 0) {
// 阻塞到下一个毫秒,获得新的时间戳
timestamp = tilNextMillis(lastTimestamp);
}
} else {
// 当前毫秒内序列号增长结束, 序列号重置
sequence = 0L;
}
// 存储上一次生成 id的时间戳
lastTimestamp = timestamp;
// 移位和或运算结合组成64位 id 号
Long resultId = ((timestamp - twepoch) << timestampLeftShift) | (datacenterId << datacenterIdShift) | (workId << workIdShift) | sequence;
return resultId;
}
/**
* 阻塞到下一个毫秒,直到获取新的时间戳
* @param lastTimestamp
* @return
*/
private long tilNextMillis(long lastTimestamp) {
long timestamp = timeGen();
while (timestamp <= lastTimestamp) {
timestamp = timeGen();
}
return timestamp;
}
/**
* 获取当前时间,毫秒
* @return
*/
private long timeGen() {
return System.currentTimeMillis();
}
// test
public static void main(String[] args) {
SnowflakeIdWorker idWorker = new SnowflakeIdWorker(1, 0);
for (int i = 0; i < 100; i++) {
long id = idWorker.nextId();
System.out.println(Long.toBinaryString(id));
System.out.println(id);
}
}
}