在分布式系統設計中,全局唯一ID是一個基礎而關鍵的組件。隨著業務規模擴大和系統架構向微服務演進,傳統的單機自增ID已無法滿足需求。高并發、高可用的分布式ID生成方案成為構建可靠分布式系統的必要條件。
Redis具備高性能、原子操作及簡單易用的特性,因此我們可以基于Redis實現全局唯一ID的生成。
分布式ID的核心需求
一個優秀的分布式ID生成方案應滿足以下要求
- 全局唯一性:在整個分布式系統中保證ID不重復
- 高性能:能夠快速生成ID,支持高并發場景
- 高可用:避免單點故障,確保服務持續可用
- 趨勢遞增:生成的ID大致呈遞增趨勢,便于數據庫索引和分片
- 安全性(可選) :不包含敏感信息,不易被推測和偽造
1. 基于INCR命令的簡單自增ID
原理
這是最直接的Redis分布式ID實現方式,利用Redis的INCR
命令原子性遞增一個計數器,確保在分布式環境下ID的唯一性。
代碼實現
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;@Component
public class RedisSimpleIdGenerator {private final RedisTemplate<String, String> redisTemplate;private final String ID_KEY;public RedisSimpleIdGenerator(RedisTemplate<String, String> redisTemplate) {this.redisTemplate = redisTemplate;this.ID_KEY = "distributed:id:generator";}/*** 生成下一個ID* @return 唯一ID*/public long nextId() {Long id = redisTemplate.opsForValue().increment(ID_KEY);if (id == null) {throw new RuntimeException("Failed to generate id");}return id;}/*** 為指定業務生成ID* @param bizTag 業務標簽* @return 唯一ID*/public long nextId(String bizTag) {String key = ID_KEY + ":" + bizTag;Long id = redisTemplate.opsForValue().increment(key);if (id == null) {throw new RuntimeException("Failed to generate id for " + bizTag);}return id;}/*** 獲取當前ID值但不遞增* @param bizTag 業務標簽* @return 當前ID值*/public long currentId(String bizTag) {String key = ID_KEY + ":" + bizTag;String value = redisTemplate.opsForValue().get(key);return value != null ? Long.parseLong(value) : 0;}
}
優缺點
優點
- 實現極其簡單,僅需一次Redis操作
- ID嚴格遞增,適合作為數據庫主鍵
- 支持多業務ID隔離
缺點
- Redis單點故障會導致ID生成服務不可用
- 主從切換可能導致ID重復
- 無法包含業務含義
適用場景
- 中小規模系統的自增主鍵生成
- 對ID連續性有要求的業務場景
- 單數據中心部署的應用
2. 基于Lua腳本的批量ID生成
原理
通過Lua腳本一次性獲取一批ID,減少網絡往返次數,客戶端可在內存中順序分配ID,顯著提高性能。
代碼實現
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;import java.util.Collections;
import java.util.List;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;@Component
public class RedisBatchIdGenerator {private final RedisTemplate<String, String> redisTemplate;private final String ID_KEY = "distributed:batch:id";private final DefaultRedisScript<Long> batchIncrScript;// 批量獲取的大小private final int BATCH_SIZE = 1000;// 本地計數器和鎖private AtomicLong currentId = new AtomicLong(0);private AtomicLong endId = new AtomicLong(0);private final Lock lock = new ReentrantLock();public RedisBatchIdGenerator(RedisTemplate<String, String> redisTemplate) {this.redisTemplate = redisTemplate;// 創建Lua腳本String scriptText = "local key = KEYS[1] " +"local step = tonumber(ARGV[1]) " +"local currentValue = redis.call('incrby', key, step) " +"return currentValue";this.batchIncrScript = new DefaultRedisScript<>();this.batchIncrScript.setScriptText(scriptText);this.batchIncrScript.setResultType(Long.class);}/*** 獲取下一個ID*/public long nextId() {// 如果當前ID超過了分配范圍,則重新獲取一批if (currentId.get() >= endId.get()) {lock.lock();try {// 雙重檢查,防止多線程重復獲取if (currentId.get() >= endId.get()) {// 執行Lua腳本獲取一批IDLong newEndId = redisTemplate.execute(batchIncrScript, Collections.singletonList(ID_KEY),String.valueOf(BATCH_SIZE));if (newEndId == null) {throw new RuntimeException("Failed to generate batch ids");}// 設置新的ID范圍endId.set(newEndId);currentId.set(newEndId - BATCH_SIZE);}} finally {lock.unlock();}}// 分配下一個IDreturn currentId.incrementAndGet();}/*** 為指定業務生成ID*/public long nextId(String bizTag) {// 實際項目中應該為每個業務標簽維護獨立的計數器和范圍// 這里簡化處理,僅使用不同的Redis keyString key = ID_KEY + ":" + bizTag;Long newEndId = redisTemplate.execute(batchIncrScript, Collections.singletonList(key),String.valueOf(1));return newEndId != null ? newEndId : -1;}
}
優缺點
優點
- 顯著減少Redis網絡請求次數
- 客戶端緩存ID段,大幅提高性能
- 降低Redis服務器壓力
- 支持突發流量處理
缺點
- 實現復雜度增加
- 服務重啟可能導致ID段浪費
適用場景
- 高并發系統,需要極高ID生成性能的場景
- 對ID連續性要求不嚴格的業務
- 能容忍小部分ID浪費的場景
3. 基于Redis的分段式ID分配(號段模式)
原理
號段模式是一種優化的批量ID生成方案,通過預分配號段(ID范圍)減少服務間競爭,同時引入雙Buffer機制提高可用性。
代碼實現
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;import java.util.Collections;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;@Component
public class RedisSegmentIdGenerator {private final RedisTemplate<String, String> redisTemplate;private final String SEGMENT_KEY = "distributed:segment:id";private final DefaultRedisScript<Long> segmentScript;// 號段大小private final int SEGMENT_STEP = 1000;// 加載因子,當前號段使用到這個百分比時就異步加載下一個號段private final double LOAD_FACTOR = 0.7;// 存儲業務號段信息的Mapprivate final Map<String, SegmentBuffer> businessSegmentMap = new ConcurrentHashMap<>();public RedisSegmentIdGenerator(RedisTemplate<String, String> redisTemplate) {this.redisTemplate = redisTemplate;// 創建Lua腳本String scriptText = "local key = KEYS[1] " +"local step = tonumber(ARGV[1]) " +"local value = redis.call('incrby', key, step) " +"return value";this.segmentScript = new DefaultRedisScript<>();this.segmentScript.setScriptText(scriptText);this.segmentScript.setResultType(Long.class);}/*** 獲取下一個ID* @param bizTag 業務標簽* @return 唯一ID*/public long nextId(String bizTag) {// 獲取或創建號段緩沖區SegmentBuffer buffer = businessSegmentMap.computeIfAbsent(bizTag, k -> new SegmentBuffer(bizTag));return buffer.nextId();}/*** 內部號段緩沖區類,實現雙Buffer機制*/private class SegmentBuffer {private String bizTag;private Segment[] segments = new Segment[2]; // 雙Bufferprivate volatile int currentPos = 0; // 當前使用的segment位置private Lock lock = new ReentrantLock();private volatile boolean isLoadingNext = false; // 是否正在異步加載下一個號段public SegmentBuffer(String bizTag) {this.bizTag = bizTag;segments[0] = new Segment(0, 0);segments[1] = new Segment(0, 0);}/*** 獲取下一個ID*/public long nextId() {// 獲取當前號段Segment segment = segments[currentPos];// 如果當前號段為空或已用完,切換到另一個號段if (!segment.isInitialized() || segment.getValue() > segment.getMax()) {lock.lock();try {// 雙重檢查當前號段狀態segment = segments[currentPos];if (!segment.isInitialized() || segment.getValue() > segment.getMax()) {// 切換到另一個號段currentPos = (currentPos + 1) % 2;segment = segments[currentPos];// 如果另一個號段也未初始化或已用完,則同步加載if (!segment.isInitialized() || segment.getValue() > segment.getMax()) {loadSegmentFromRedis(segment);}}} finally {lock.unlock();}}// 檢查是否需要異步加載下一個號段long value = segment.incrementAndGet();if (value > segment.getMin() + (segment.getMax() - segment.getMin()) * LOAD_FACTOR&& !isLoadingNext) {isLoadingNext = true;// 異步加載下一個號段new Thread(() -> {Segment nextSegment = segments[(currentPos + 1) % 2];loadSegmentFromRedis(nextSegment);isLoadingNext = false;}).start();}return value;}/*** 從Redis加載號段*/private void loadSegmentFromRedis(Segment segment) {String key = SEGMENT_KEY + ":" + bizTag;// 執行Lua腳本獲取號段最大值Long max = redisTemplate.execute(segmentScript, Collections.singletonList(key),String.valueOf(SEGMENT_STEP));if (max == null) {throw new RuntimeException("Failed to load segment from Redis");}// 設置號段范圍long min = max - SEGMENT_STEP + 1;segment.setMax(max);segment.setMin(min);segment.setValue(min - 1); // 設置為min-1,第一次incrementAndGet返回minsegment.setInitialized(true);}}/*** 內部號段類,存儲號段的范圍信息*/private class Segment {private long min; // 最小值private long max; // 最大值private AtomicLong value; // 當前值private volatile boolean initialized; // 是否已初始化public Segment(long min, long max) {this.min = min;this.max = max;this.value = new AtomicLong(min);this.initialized = false;}public long getValue() {return value.get();}public void setValue(long value) {this.value.set(value);}public long incrementAndGet() {return value.incrementAndGet();}public long getMin() {return min;}public void setMin(long min) {this.min = min;}public long getMax() {return max;}public void setMax(long max) {this.max = max;}public boolean isInitialized() {return initialized;}public void setInitialized(boolean initialized) {this.initialized = initialized;}}
}
優缺點
優點
- 雙Buffer設計,高可用性
- 異步加載下一個號段,性能更高
- 大幅降低Redis訪問頻率
- 即使Redis短暫不可用,仍可分配一段時間的ID
缺點
- 實現復雜,代碼量大
- 多實例部署時,各實例獲取的號段不連續
- 重啟服務時號段內的ID可能浪費
- 需要在內存中維護狀態
適用場景
- 對ID生成可用性要求高的業務
- 需要高性能且多服務器部署的分布式系統
4. 性能對比與選型建議
策略 | 性能 | 可用性 | ID長度 | 實現復雜度 | 單調遞增 |
---|---|---|---|---|---|
INCR命令 | ★★★☆☆ | ★★☆☆☆ | 遞增整數 | 低 | 嚴格遞增 |
Lua批量生成 | ★★★★★ | ★★★☆☆ | 遞增整數 | 中 | 批次內遞增 |
分段式ID | ★★★★★ | ★★★★☆ | 遞增整數 | 高 | 段內遞增 |
5. 實踐優化技巧
1. Redis高可用配置
// 配置Redis哨兵模式,提高可用性
@Bean
public RedisConnectionFactory redisConnectionFactory() {RedisSentinelConfiguration sentinelConfig = new RedisSentinelConfiguration().master("mymaster").sentinel("127.0.0.1", 26379).sentinel("127.0.0.1", 26380).sentinel("127.0.0.1", 26381);return new LettuceConnectionFactory(sentinelConfig);
}
2. ID預熱策略
// 系統啟動時預熱ID生成器
@PostConstruct
public void preWarmIdGenerator() {// 預先獲取一批ID,確保系統啟動后立即可用for (int i = 0; i < 10; i++) {try {segmentIdGenerator.nextId("order");segmentIdGenerator.nextId("user");segmentIdGenerator.nextId("payment");} catch (Exception e) {log.error("Failed to pre-warm ID generator", e);}}
}
3. 降級策略
// Redis不可用時的降級策略
public long nextIdWithFallback(String bizTag) {try {return segmentIdGenerator.nextId(bizTag);} catch (Exception e) {log.warn("Failed to get ID from Redis, using local fallback", e);// 使用本地UUID或其他替代方案return Math.abs(UUID.randomUUID().getMostSignificantBits());}
}
6. 結論
選擇合適的分布式ID生成策略時,需要綜合考慮系統規模、性能需求、可靠性要求和實現復雜度。無論選擇哪種方案,都應注重高可用性設計,增加監控和預警機制,確保ID生成服務的穩定運行。
在實踐中,可以基于業務需求對這些方案進行組合和優化,例如為不同業務選擇不同策略,或者在ID中嵌入業務標識等,打造更適合自身系統的分布式ID生成解決方案。