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.     }  

 

展开阅读全文

没有更多推荐了,返回首页