工具类封装实战(五)线程安全的雪花算法

举报
小鲍侃java 发表于 2021/09/09 22:43:45 2021/09/09
【摘要】 雪花算法实现 package com.yjd.comm.util;/** * Created by pc on 2017/8/16 0016. */ /** * Twitter_Snowflake<br> * SnowFlake的结构如下(每部分用-分开):<br> * 0 - 0000000000 00000...

雪花算法实现


  
  1. package com.yjd.comm.util;/**
  2. * Created by pc on 2017/8/16 0016.
  3. */
  4. /**
  5. * Twitter_Snowflake<br>
  6. * SnowFlake的结构如下(每部分用-分开):<br>
  7. * 0 - 0000000000 0000000000 0000000000 0000000000 0 - 00000 - 00000 - 000000000000 <br>
  8. * 1位标识,由于long基本类型在Java中是带符号的,最高位是符号位,正数是0,负数是1,所以id一般是正数,最高位是0<br>
  9. * 41位时间截(毫秒级),注意,41位时间截不是存储当前时间的时间截,而是存储时间截的差值(当前时间截 - 开始时间截)
  10. * 得到的值),这里的的开始时间截,一般是我们的id生成器开始使用的时间,由我们程序来指定的(如下下面程序IdWorker类的startTime属性)。41位的时间截,可以使用69年,年T = (1L << 41) / (1000L * 60 * 60 * 24 * 365) = 69<br>
  11. * 10位的数据机器位,可以部署在1024个节点,包括5位datacenterId和5位workerId<br>
  12. * 12位序列,毫秒内的计数,12位的计数顺序号支持每个节点每毫秒(同一机器,同一时间截)产生4096个ID序号<br>
  13. * 加起来刚好64位,为一个Long型。<br>
  14. * SnowFlake的优点是,整体上按照时间自增排序,并且整个分布式系统内不会产生ID碰撞(由数据中心ID和机器ID作区分),并且效率较高,经测试,SnowFlake每秒能够产生26万ID左右。
  15. */
  16. public class SnowflakeIdWorker {
  17. // ==============================Fields===========================================
  18. /**
  19. * 开始时间截 (2015-01-01)
  20. */
  21. private final long twepoch = 1420041600000L;
  22. /**
  23. * 机器id所占的位数
  24. */
  25. private final long workerIdBits = 5L;
  26. /**
  27. * 数据标识id所占的位数
  28. */
  29. private final long datacenterIdBits = 5L;
  30. /**
  31. * 支持的最大机器id,结果是31 (这个移位算法可以很快的计算出几位二进制数所能表示的最大十进制数)
  32. */
  33. private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
  34. /**
  35. * 支持的最大数据标识id,结果是31
  36. */
  37. private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
  38. /**
  39. * 序列在id中占的位数
  40. */
  41. private final long sequenceBits = 12L;
  42. /**
  43. * 机器ID向左移12位
  44. */
  45. private final long workerIdShift = sequenceBits;
  46. /**
  47. * 数据标识id向左移17位(12+5)
  48. */
  49. private final long datacenterIdShift = sequenceBits + workerIdBits;
  50. /**
  51. * 时间截向左移22位(5+5+12)
  52. */
  53. private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
  54. /**
  55. * 生成序列的掩码,这里为4095 (0b111111111111=0xfff=4095)
  56. */
  57. private final long sequenceMask = -1L ^ (-1L << sequenceBits);
  58. /**
  59. * 工作机器ID(0~31)
  60. */
  61. private long workerId;
  62. /**
  63. * 数据中心ID(0~31)
  64. */
  65. private long datacenterId;
  66. /**
  67. * 毫秒内序列(0~4095)
  68. */
  69. private long sequence = 0L;
  70. /**
  71. * 上次生成ID的时间截
  72. */
  73. private long lastTimestamp = -1L;
  74. //==============================Constructors=====================================
  75. /**
  76. * 构造函数
  77. *
  78. * @param workerId 工作ID (0~31)
  79. * @param datacenterId 数据中心ID (0~31)
  80. */
  81. public SnowflakeIdWorker(long workerId, long datacenterId) {
  82. if (workerId > maxWorkerId || workerId < 0) {
  83. throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
  84. }
  85. if (datacenterId > maxDatacenterId || datacenterId < 0) {
  86. throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
  87. }
  88. this.workerId = workerId;
  89. this.datacenterId = datacenterId;
  90. }
  91. // ==============================Methods==========================================
  92. /**
  93. * 获得下一个ID (该方法是线程安全的)
  94. *
  95. * @return SnowflakeId
  96. */
  97. public synchronized long nextId() {
  98. long timestamp = timeGen();
  99. //如果当前时间小于上一次ID生成的时间戳,说明系统时钟回退过这个时候应当抛出异常
  100. if (timestamp < lastTimestamp) {
  101. throw new RuntimeException(
  102. String.format("Clock moved backwards. Refusing to generate id for %d milliseconds", lastTimestamp - timestamp));
  103. }
  104. //如果是同一时间生成的,则进行毫秒内序列
  105. if (lastTimestamp == timestamp) {
  106. sequence = (sequence + 1) & sequenceMask;
  107. //毫秒内序列溢出
  108. if (sequence == 0) {
  109. //阻塞到下一个毫秒,获得新的时间戳
  110. timestamp = tilNextMillis(lastTimestamp);
  111. }
  112. }
  113. //时间戳改变,毫秒内序列重置
  114. else {
  115. sequence = 0L;
  116. }
  117. //上次生成ID的时间截
  118. lastTimestamp = timestamp;
  119. //移位并通过或运算拼到一起组成64位的ID
  120. return ((timestamp - twepoch) << timestampLeftShift) //
  121. | (datacenterId << datacenterIdShift) //
  122. | (workerId << workerIdShift) //
  123. | sequence;
  124. }
  125. /**
  126. * 阻塞到下一个毫秒,直到获得新的时间戳
  127. *
  128. * @param lastTimestamp 上次生成ID的时间截
  129. * @return 当前时间戳
  130. */
  131. protected long tilNextMillis(long lastTimestamp) {
  132. long timestamp = timeGen();
  133. while (timestamp <= lastTimestamp) {
  134. timestamp = timeGen();
  135. }
  136. return timestamp;
  137. }
  138. /**
  139. * 返回以毫秒为单位的当前时间
  140. *
  141. * @return 当前时间(毫秒)
  142. */
  143. protected long timeGen() {
  144. return System.currentTimeMillis();
  145. }
  146. //==============================Test=============================================
  147. /**
  148. * 测试
  149. */
  150. public static void main(String[] args) {
  151. SnowflakeIdWorker idWorker = new SnowflakeIdWorker(1, 1);
  152. long startime = System.currentTimeMillis();
  153. for (int i = 0; i < 4000000; i++) {
  154. long id = idWorker.nextId();
  155. // System.out.println(Long.toBinaryString(id));
  156. // System.out.println(id);
  157. }
  158. System.out.println(System.currentTimeMillis() - startime);
  159. }
  160. }

解决线程安全问题


  
  1. @Component
  2. public class SnowflakeComponent {
  3. @Value("${server.datacenterId}")
  4. private long datacenterId;
  5. @Value("${server.workId}")
  6. private long workId;
  7. private static volatile SnowflakeIdWorker instance;
  8. public SnowflakeIdWorker getInstance() {
  9. if (instance == null) {
  10. synchronized (SnowflakeIdWorker.class) {
  11. if (instance == null) {
  12. instance = new SnowflakeIdWorker(workId, datacenterId);
  13. }
  14. }
  15. }
  16. return instance;
  17. }
  18. }

application.yml


  
  1. server:
  2. port: 8001
  3. workId: 0
  4. datacenterId: 0 #雪花算法的数据中心id,在java 启动命令中定义,四台为0,1,2,3

调用生成唯一主键


  
  1. @Autowired
  2. private SnowflakeComponent snowflakeComponent;
  3. snowflakeComponent.getInstance().nextId()

文章来源: baocl.blog.csdn.net,作者:小黄鸡1992,版权归原作者所有,如需转载,请联系作者。

原文链接:baocl.blog.csdn.net/article/details/110871632

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。