snowflake的JAVA版本

1. /**
2. * @author zhujuan
3. * From: https://github.com/twitter/snowflake
4. * An object that generates IDs.
5. * This is broken into a separate class in case
6. * we ever want to support multiple worker threads
7. * per process
8. */
9. public class IdWorker {
11. protected static final Logger LOG = LoggerFactory.getLogger(IdWorker.class);
13. private long workerId;
14. private long datacenterId;
15. private long sequence = 0L;
17. private long twepoch = 1288834974657L;
19. private long workerIdBits = 5L;
20. private long datacenterIdBits = 5L;
21. private long maxWorkerId = -1L ^ (-1L << workerIdBits);
22. private long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
23. private long sequenceBits = 12L;
25. private long workerIdShift = sequenceBits;
26. private long datacenterIdShift = sequenceBits + workerIdBits;
27. private long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
28. private long sequenceMask = -1L ^ (-1L << sequenceBits);
30. private long lastTimestamp = -1L;
32. public IdWorker(long workerId, long datacenterId) {
33. // sanity check for workerId
34. if (workerId > maxWorkerId || workerId < 0) {
35. throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
36. }
37. if (datacenterId > maxDatacenterId || datacenterId < 0) {
38. throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
39. }
40. this.workerId = workerId;
41. this.datacenterId = datacenterId;
42. LOG.info(String.format("worker starting. timestamp left shift %d, datacenter id bits %d, worker id bits %d, sequence bits %d, workerid %d", timestampLeftShift, datacenterIdBits, workerIdBits, sequenceBits, workerId));
43. }
45. public synchronized long nextId() {
46. long timestamp = timeGen();
48. if (timestamp < lastTimestamp) {
49. LOG.error(String.format("clock is moving backwards. Rejecting requests until %d.", lastTimestamp));
50. throw new RuntimeException(String.format("Clock moved backwards. Refusing to generate id for %d milliseconds", lastTimestamp - timestamp));
51. }
53. if (lastTimestamp == timestamp) {
54. sequence = (sequence + 1) & sequenceMask;
55. if (sequence == 0) {
56. timestamp = tilNextMillis(lastTimestamp);
57. }
58. } else {
59. sequence = 0L;
60. }
62. lastTimestamp = timestamp;
64. return ((timestamp - twepoch) << timestampLeftShift) | (datacenterId << datacenterIdShift) | (workerId << workerIdShift) | sequence;
65. }
67. protected long tilNextMillis(long lastTimestamp) {
68. long timestamp = timeGen();
69. while (timestamp <= lastTimestamp) {
70. timestamp = timeGen();
71. }
72. return timestamp;
73. }
75. protected long timeGen() {
76. return System.currentTimeMillis();
77. }

1. }

测试类

1. public class IdWorkerTest {
3. static class IdWorkThread implements Runnable {
4. private Set<Long> set;
5. private IdWorker idWorker;
7. public IdWorkThread(Set<Long> set, IdWorker idWorker) {
8. this.set = set;
9. this.idWorker = idWorker;
10. }
12. @Override
13. public void run() {
14. while (true) {
15. long id = idWorker.nextId();
16. if (!set.add(id)) {
17. System.out.println("duplicate:" + id);
18. }
19. }
20. }
21. }
23. public static void main(String[] args) {
24. Set<Long> set = new HashSet<Long>();
25. final IdWorker idWorker1 = new IdWorker(0, 0);
26. final IdWorker idWorker2 = new IdWorker(1, 0);
27. Thread t1 = new Thread(new IdWorkThread(set, idWorker1));
28. Thread t2 = new Thread(new IdWorkThread(set, idWorker2));
29. t1.setDaemon(true);
30. t2.setDaemon(true);
31. t1.start();
32. t2.start();
33. try {
34. Thread.sleep(30000);
35. } catch (InterruptedException e) {
36. e.printStackTrace();
37. }
38. }
39. }