SnowFlake 分布式ID生成Java实现

SnowFlake不依赖第三方介质，不像基于ZK，Redis等，每次用完一个区间还得通过网络去获取下一个区间，效率较低，基于SnowFlake的分布式ID生成是目前我见过的最快的

 

SnowFlake生成的是一个64位的数字，其中42位时间戳，接下来10位是自定义的数，其作用就是区分集群中的所有机器，最后12位是毫秒内序列，集群内每个机器能够在1毫秒内生成2^12 - 1个ID

 

 伦理片 http://www.dotdy.com/  

Java代码  

1. /** 
2. * 基于SnowFlake的序列号生成实现, 64位ID (42(毫秒)+5(机器ID)+5(业务编码)+12(重复累加)) 
3. */  
4. static class Generator {  
5.   
6.         private final static long TWEPOCH = 1288834974657L;  
7.   
8.         // 机器标识位数  
9.         private final static long WORKER_ID_BITS = 5L;  
10.   
11.         // 数据中心标识位数  
12.         private final static long DATA_CENTER_ID_BITS = 5L;  
13.   
14.         // 机器ID最大值 31  
15.         private final static long MAX_WORKER_ID = -1L ^ (-1L << WORKER_ID_BITS);  
16.   
17.         // 数据中心ID最大值 31  
18.         private final static long MAX_DATA_CENTER_ID = -1L ^ (-1L << DATA_CENTER_ID_BITS);  
19.   
20.         // 毫秒内自增位  
21.         private final static long SEQUENCE_BITS = 12L;  
22.   
23.         // 机器ID偏左移12位  
24.         private final static long WORKER_ID_SHIFT = SEQUENCE_BITS;  
25.   
26.         private final static long DATA_CENTER_ID_SHIFT = SEQUENCE_BITS + WORKER_ID_BITS;  
27.   
28.         // 时间毫秒左移22位  
29.         private final static long TIMESTAMP_LEFT_SHIFT = SEQUENCE_BITS + WORKER_ID_BITS + DATA_CENTER_ID_BITS;  
30.   
31.         private final static long SEQUENCE_MASK = -1L ^ (-1L << SEQUENCE_BITS);  
32.   
33.         private long lastTimestamp = -1L;  
34.   
35.         private long sequence = 0L;  
36.         private final long workerId;  
37.         private final long dataCenterId;  
38.           
39.         //private final AtomicBoolean lock = new AtomicBoolean(false);  
40.           
41.         Generator(long workerId, long dataCenterId) {  
42.             if (workerId > MAX_WORKER_ID || workerId < 0) {  
43.                 throw new IllegalArgumentException(String.format("%s must range from %d to %d", K_WORK_ID, 0,  
44.                         MAX_WORKER_ID));  
45.             }  
46.   
47.             if (dataCenterId > MAX_DATA_CENTER_ID || dataCenterId < 0) {  
48.                 throw new IllegalArgumentException(String.format("%s must range from %d to %d", K_DC_ID, 0,  
49.                         MAX_DATA_CENTER_ID));  
50.             }  
51.   
52.             this.workerId = workerId;  
53.             this.dataCenterId = dataCenterId;  
54.         }  
55.   
56.         synchronized long nextValue() throws SequenceException {  
57.             long timestamp = time();  
58.             if (timestamp < lastTimestamp) {  
59.                 throw new SequenceException("Clock moved backwards, refuse to generate id for "  
60.                         + (lastTimestamp - timestamp) + " milliseconds");  
61.             }  
62.   
63.             if (lastTimestamp == timestamp) {  
64.                 // 当前毫秒内，则+1  
65.                 sequence = (sequence + 1) & SEQUENCE_MASK;  
66.                 if (sequence == 0) {  
67.                     // 当前毫秒内计数满了，则等待下一秒  
68.                     timestamp = tilNextMillis(lastTimestamp);  
69.                 }  
70.             } else {  
71.                 sequence = 0;  
72.             }  
73.             lastTimestamp = timestamp;  
74.               
75.             // ID偏移组合生成最终的ID，并返回ID  
76.             long nextId = ((timestamp - TWEPOCH) << TIMESTAMP_LEFT_SHIFT)  
77.                     | (dataCenterId << DATA_CENTER_ID_SHIFT) | (workerId << WORKER_ID_SHIFT) | sequence;  
78.   
79.             return nextId;  
80.         }  
81.   
82.         private long tilNextMillis(final long lastTimestamp) {  
83.             long timestamp = this.time();  
84.             while (timestamp <= lastTimestamp) {  
85.                 timestamp = this.time();  
86.             }  
87.             return timestamp;  
88.         }  
89.   
90.         private long time() {  
91.             return System.currentTimeMillis();  
92.         }  
93.   
94.     }