Redis作為高性能的KV數(shù)據(jù)庫，并且操作還是原子性的，所以用來做支持高并發(fā)的發(fā)號器十分合適。
本文給大家介紹3種常見的全局ID生成方式。
1、全局遞增ID
目標：一直遞增的全局ID。
/**
* 一直遞增的全局id
*
* @param redisTemplate redis客戶端對象
* @param busId 業(yè)務id，可以按需配置
* @param step 步長，即每次遞增的間隔
*/
public static String getNo(RedisTemplate<String, Object> redisTemplate, String busId, int step) {
//保存redis中的key，注意不要重復
String redisKey = "uniqueNo_";
//利用increment即redis原生incrBy命令的原子性特性生成遞增的序列號
Long increment = redisTemplate.opsForValue().increment(redisKey, step);
if (increment == null) {
throw new RuntimeException("redis命令執(zhí)行失敗");
}
//業(yè)務id+遞增id，如果需要純數(shù)字，去掉業(yè)務id即可
return busId + increment;
}
2、以天為分割的全局ID
目標：生成格式為 yyyyMMdd + 遞增序列號的全局ID。
/**
* 以天為間隔的遞增序列號
* @param redisTemplate redis客戶端對象
* @param busId 業(yè)務id，可以按需配置
* @param step 步長，即每次遞增的間隔
*/
public static String getNo(RedisTemplate<String, Object> redisTemplate, String busId, int step) {
//當天日期，比如20221226
String date = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
//保存redis中的key，注意不要重復
String redisKey = "uniqueNo_" + date;
//利用increment即redis原生incrBy命令的原子性特性生成遞增的序列號
Long increment = redisTemplate.opsForValue().increment(redisKey, step);
if (increment == null) {
throw new RuntimeException("redis命令執(zhí)行失敗");
}
if (step == increment.intValue()) {
//首次執(zhí)行時，給redisKey設置ttl，第二天這個key就可以被redis自動刪除
redisTemplate.expire(redisKey, 25, TimeUnit.HOURS);
}
//組合 20221226 + 業(yè)務id + 0001(可以根據(jù)需要自由調整序列號的長度)
return date + busId + String.format("%04d", increment);
}
3、批量獲取ID
有時我們需要批量的獲取遞增ID，比如給一批訂單號設置ID。
/**
* 批量獲取id
*
* @param redisTemplate redis客戶端對象
* @param busId 業(yè)務id，可以按需配置
* @param size 獲取的id個數(shù)，與步長類似
*/
public static List<String> getNoByGroup(RedisTemplate<String, Object> redisTemplate, String busId, int size) {
//保存redis中的key，注意不要重復
String redisKey = "uniqueNo_group";
//設置步長為size，相當于一次性申請size個id
Long increment = redisTemplate.opsForValue().increment(redisKey, size);
if (increment == null) {
throw new RuntimeException("redis命令執(zhí)行失敗");
}
long begin = increment - Long.parseLong(size + "");
List<String> rs = new ArrayList<>();
for (long i = begin + 1; i <= increment; i++) {
rs.add(busId + i);
}
return rs;
}
總結
無論我們需要什么格式的ID，其實只要我們把握住其中的核心：incrBy命令，根據(jù)其原子性的特性，就可以生成我們需要的全局ID。
但是需要注意的是，雖然incrBy命令是原子性的，但是通過組合鍵進行組合時，其實是破壞了這種原子性。如果有特殊的ID格式要求，務必要進行充分的測試。
