个人随笔
目录
雪花算法(SnowFlake)的原理和Java版本的实现
2019-10-31 23:57:48

分布式id生成算法的有很多种,Twitter的SnowFlake就是其中经典的一种。

算法原理

SnowFlake算法生成id的结果是一个64bit大小的整数,它的结构如下图:

  • 1bit,不用,因为二进制中最高位是符号位,1表示负数,0表示正数。生成的id一般都是用整数,所以最高位固定为0。

  • 41bit-时间戳,用来记录时间戳,毫秒级。
    41位可以表示个数字,
    如果只用来表示正整数(计算机中正数包含0),可以表示的数值范围是:0 至 ,减1是因为可表示的数值范围是从0开始算的,而不是1。
    也就是说41位可以表示个毫秒的值,转化成单位年则是年

  • 10bit-工作机器id,用来记录工作机器id。
    可以部署在个节点,包括5位datacenterId和5位workerId
    5位(bit)可以表示的最大正整数是,即可以用0、1、2、3、….31这32个数字,来表示不同的datecenterId或workerId

  • 12bit-序列号,序列号,用来记录同毫秒内产生的不同id。
    12位(bit)可以表示的最大正整数是,即可以用0、1、2、3、….4094这4095个数字,来表示同一机器同一时间截(毫秒)内产生的4095个ID序号。

由于在Java中64bit的整数是long类型,所以在Java中SnowFlake算法生成的id就是long来存储的。

SnowFlake可以保证:

  1. 所有生成的id按时间趋势递增
  2. 整个分布式系统内不会产生重复id(因为有datacenterId和workerId来做区分)整个分布式系统内不会产生重复id(因为有datacenterId和workerId来做区分)

算法实现(Java)
Twitter官方给出的算法实现 是用Scala写的,这里不做分析,可自行查看。

Java版算法实现

  1. package com.suibibk.utils;
  2. public class SnowflakeIdUtil {
  3. // ==============================Fields===========================================
  4. /** 开始时间截 (2015-01-01) */
  5. private final long twepoch = 1420041600000L;
  6. /** 机器id所占的位数 */
  7. private final long workerIdBits = 5L;
  8. /** 数据标识id所占的位数 */
  9. private final long datacenterIdBits = 5L;
  10. /** 支持的最大机器id,结果是31 (这个移位算法可以很快的计算出几位二进制数所能表示的最大十进制数) */
  11. private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
  12. /** 支持的最大数据标识id,结果是31 */
  13. private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
  14. /** 序列在id中占的位数 */
  15. private final long sequenceBits = 12L;
  16. /** 机器ID向左移12位 */
  17. private final long workerIdShift = sequenceBits;
  18. /** 数据标识id向左移17位(12+5) */
  19. private final long datacenterIdShift = sequenceBits + workerIdBits;
  20. /** 时间截向左移22位(5+5+12) */
  21. private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
  22. /** 生成序列的掩码,这里为4095 (0b111111111111=0xfff=4095) */
  23. private final long sequenceMask = -1L ^ (-1L << sequenceBits);
  24. /** 工作机器ID(0~31) */
  25. private long workerId;
  26. /** 数据中心ID(0~31) */
  27. private long datacenterId;
  28. /** 毫秒内序列(0~4095) */
  29. private long sequence = 0L;
  30. /** 上次生成ID的时间截 */
  31. private long lastTimestamp = -1L;
  32. private static SnowflakeIdUtil idWorker;
  33. /**
  34. * 以SnowFlake算法,获取唯一有序id
  35. * @return
  36. */
  37. public static long getSnowflakeId() {
  38. if(idWorker == null) {
  39. synchronized (SnowflakeIdUtil.class) {
  40. if(idWorker == null) {
  41. idWorker = new SnowflakeIdUtil(0, 0);
  42. }
  43. }
  44. }
  45. return idWorker.nextId();
  46. }
  47. // ==============================Methods==========================================
  48. private SnowflakeIdUtil() {
  49. }
  50. //==============================Constructors=====================================
  51. /**
  52. * 构造函数
  53. * @param workerId 工作ID (0~31)
  54. * @param datacenterId 数据中心ID (0~31)
  55. */
  56. private SnowflakeIdUtil(long workerId, long datacenterId) {
  57. if (workerId > maxWorkerId || workerId < 0) {
  58. throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
  59. }
  60. if (datacenterId > maxDatacenterId || datacenterId < 0) {
  61. throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
  62. }
  63. this.workerId = workerId;
  64. this.datacenterId = datacenterId;
  65. }
  66. /**
  67. * 阻塞到下一个毫秒,直到获得新的时间戳
  68. * @param lastTimestamp 上次生成ID的时间截
  69. * @return 当前时间戳
  70. */
  71. protected long tilNextMillis(long lastTimestamp) {
  72. long timestamp = timeGen();
  73. while (timestamp <= lastTimestamp) {
  74. timestamp = timeGen();
  75. }
  76. return timestamp;
  77. }
  78. /**
  79. * 返回以毫秒为单位的当前时间
  80. * @return 当前时间(毫秒)
  81. */
  82. protected long timeGen() {
  83. return System.currentTimeMillis();
  84. }
  85. /**
  86. * 获得下一个ID (该方法是线程安全的)
  87. * @return SnowflakeId
  88. */
  89. protected synchronized long nextId() {
  90. long timestamp = timeGen();
  91. //如果当前时间小于上一次ID生成的时间戳,说明系统时钟回退过这个时候应当抛出异常
  92. if (timestamp < lastTimestamp) {
  93. throw new RuntimeException(
  94. String.format("Clock moved backwards. Refusing to generate id for %d milliseconds", lastTimestamp - timestamp));
  95. }
  96. //如果是同一时间生成的,则进行毫秒内序列
  97. if (lastTimestamp == timestamp) {
  98. sequence = (sequence + 1) & sequenceMask;
  99. //毫秒内序列溢出
  100. if (sequence == 0) {
  101. //阻塞到下一个毫秒,获得新的时间戳
  102. timestamp = tilNextMillis(lastTimestamp);
  103. }
  104. }
  105. //时间戳改变,毫秒内序列重置
  106. else {
  107. sequence = 0L;
  108. }
  109. //上次生成ID的时间截
  110. lastTimestamp = timestamp;
  111. //移位并通过或运算拼到一起组成64位的ID
  112. return ((timestamp - twepoch) << timestampLeftShift) //
  113. | (datacenterId << datacenterIdShift) //
  114. | (workerId << workerIdShift) //
  115. | sequence;
  116. }
  117. public static void main(String[] args) {
  118. long snowflakeId = getSnowflakeId();
  119. System.out.println(snowflakeId);
  120. }
  121. }
 1788

啊!这个可能是世界上最丑的留言输入框功能~


当然,也是最丑的留言列表

有疑问发邮件到 : suibibk@qq.com 侵权立删
Copyright : 个人随笔   备案号 : 粤ICP备18099399号-2