leisure.springboot.id.SnowFlake Maven / Gradle / Ivy
The newest version!
package leisure.springboot.id;
/**
* leisure.id的结构如下(每部分用-分开) 0 - 0000000000 0000000000 0000000000 0000000000 0 -
* 00000 - 00000 - 000000000000
* 1位标识，由于long基本类型在Java中是带符号的，最高位是符号位，正数是0，负数是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左右
*/
//@Component
//@Scope(ConfigurableBeanFactory.SCOPE_SINGLETON)
public class SnowFlake
{
/** 开始时间截 (2015-01-01) */
private final long twepoch = 1420041600000L;
/** 机器id所占的位数 */
private final long workerIdBits = 5L;
/** 数据标识id所占的位数 */
private final long datacenterIdBits = 5L;
/** 支持的最大机器id，结果是31 (这个移位算法可以很快的计算出几位二进制数所能表示的最大十进制数) */
private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
/** 支持的最大数据标识id，结果是31 */
private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
/** 序列在id中占的位数 */
private final long sequenceBits = 12L;
/** 机器ID向左移12位 */
private final long workerIdShift = sequenceBits;
/** 数据标识id向左移17位(12+5) */
private final long datacenterIdShift = sequenceBits + workerIdBits;
/** 时间截向左移22位(5+5+12) */
private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
/** 生成序列的掩码，这里为4095 (0b111111111111=0xfff=4095) */
private final long sequenceMask = -1L ^ (-1L << sequenceBits);
/** 工作机器ID(0~31) */
private long workerId;
/** 数据中心ID(0~31) */
private long datacenterId;
/** 毫秒内序列(0~4095) */
private long sequence = 0L;
/** 上次生成ID的时间截 */
private long lastTimestamp = -1L;
/**
* 数据库最大编号(1-64)
*/
private long dbMaxId;
/**
* 实际最多数据库数量
*/
private long dbMaxIdCurrent;
/**
* 数据表最大编号(0-9)
*/
private long tbMaxId;
private final byte[] lock = new byte[1];
private SnowFlake()
{
// Environment environment = leisure.core.BeanFactory.getBeanByType(Environment.class);
// String strDatacenterId = environment.getProperty("datacenterid");
// String strWorkerId = environment.getProperty("leisure.workid");
// String strdbMaxId = environment.getProperty("DbMaxId");
// String strdbMaxId_Current = environment.getProperty("DbMaxId.Current");
// String strtbMaxId = environment.getProperty("TbMaxId");
// if (strDatacenterId != null && !strDatacenterId.isEmpty())
// {
// datacenterId = Long.valueOf(strDatacenterId);
// }
// if (strWorkerId != null && !strWorkerId.isEmpty())
// {
// workerId = Long.valueOf(strDatacenterId);
// }
// if (strdbMaxId != null && !strdbMaxId.isEmpty())
// {
// dbMaxId = Long.valueOf(strdbMaxId);
// }
// if (strtbMaxId != null && !strtbMaxId.isEmpty())
// {
// tbMaxId = Long.valueOf(strtbMaxId);
// }
// if (strdbMaxId_Current != null && !strdbMaxId_Current.isEmpty())
// {
// dbMaxIdCurrent = Long.valueOf(strdbMaxId_Current);
// }
//
// if (workerId > maxWorkerId || workerId < 0)
// {
// throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
// }
// if (datacenterId > maxDatacenterId || datacenterId < 0)
// {
// throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
// }
// if (dbMaxId > 99 || dbMaxId < 0)
// {
// throw new IllegalArgumentException(String.format("dbMaxId can't be greater than %d or less than 0", 99));
// }
// if (tbMaxId > 9 || tbMaxId < 0)
// {
// throw new IllegalArgumentException(String.format("tbMaxId can't be greater than %d or less than 0", 9));
// }
// if (dbMaxIdCurrent > dbMaxId || dbMaxIdCurrent < 0)
// {
// throw new IllegalArgumentException(String.format("dbMaxIdCurrent can't be greater than %d or less than 0", dbMaxId));
// }
// if (dbMaxId > dbMaxIdCurrent && (dbMaxId % dbMaxIdCurrent) != 0)
// {
// throw new IllegalArgumentException("dbMaxId % dbMaxIdCurrent must == 0");
// }
}
/**
* 获得全局分布式ID(包含分库分表信息) 如果没有分库分表,不要使用该方法
*/
public String getIdWithDb()
{
synchronized (lock)
{
long id = generateId();
if (dbMaxId == 0)
{
return String.valueOf(id);
}
String dbId = dbIdInfo(id);
String result = String.format("%s%s", dbId, id);
return result;
}
}
/**
* 获得全局分布式ID
*/
public String getId()
{
synchronized (lock)
{
long result = generateId();
return String.valueOf(result);
}
}
private long generateId()
{
long timestamp = timeGen();
// 如果当前时间小于上一次ID生成的时间戳，说明系统时钟回退过这个时候应当抛出异常
if (timestamp < lastTimestamp)
{
throw new RuntimeException(String.format("Clock moved backwards. Refusing to generate id for %d milliseconds", lastTimestamp - timestamp));
}
// 如果是同一时间生成的，则进行毫秒内序列
if (lastTimestamp == timestamp)
{
sequence = (sequence + 1) & sequenceMask;
// 毫秒内序列溢出
if (sequence == 0)
{
// 阻塞到下一个毫秒,获得新的时间戳
timestamp = tilNextMillis(lastTimestamp);
}
}
// 时间戳改变，毫秒内序列重置
else
{
sequence = 0L;
}
// 上次生成ID的时间截
lastTimestamp = timestamp;
// 移位并通过或运算拼到一起组成64位的ID
long result = ((timestamp - twepoch) << timestampLeftShift) //
| (datacenterId << datacenterIdShift) //
| (workerId << workerIdShift) //
| sequence;
return result;
}
/**
* 阻塞到下一个毫秒，直到获得新的时间戳
*
* @param lastTimestamp
* 上次生成ID的时间截
* @return 当前时间戳
*/
private long tilNextMillis(long lastTimestamp)
{
long timestamp = timeGen();
while (timestamp <= lastTimestamp)
{
timestamp = timeGen();
}
return timestamp;
}
/**
* 返回以毫秒为单位的当前时间
*
* @return 当前时间(毫秒)
*/
private long timeGen()
{
return System.currentTimeMillis();
}
/**
* 获取分库分表信息 数据库编号从1开始 数据表编号从0开始 数据库2位数 最多支持99套数据库 数据表1位数,最多支持分10个数据表
*
* @param id
* @return
*/
private String dbIdInfo(long id)
{
long dbInfo = ((id / (tbMaxId + 1)) % dbMaxId) + 1;
if (dbMaxId > dbMaxIdCurrent)
{
dbInfo = (dbInfo - 1) % dbMaxIdCurrent + 1;
}
long tbInfo = id % (tbMaxId + 1);
String strDbInfo = String.valueOf(dbInfo);
if (strDbInfo.length() < 2)
{
strDbInfo = String.format("0%s", strDbInfo);
}
String result = String.format("%s%s", strDbInfo, tbInfo);
return result;
}
}