Redis - 全局ID生成器 RedisIdWorker

文章目录

Redis - 全局ID生成器 RedisIdWorker
- 一、引言
- 二、实现原理
- 三、代码实现
- - 代码说明
- 四、使用示例
- - 示例说明
- 五、总结

Redis - 全局ID生成器 RedisIdWorker

在这里插入图片描述

一、引言

在分布式系统中，生成全局唯一ID是一个常见的需求。传统的自增ID生成方式在分布式环境下容易出现冲突，而UUID虽然可以保证唯一性，但长度较长且不够紧凑。RedisIdWorker是一种基于Redis实现的全局ID生成器，它结合了时间戳和自增序列号，能够在分布式环境中高效地生成唯一且有序的ID。

二、实现原理

RedisIdWorker的核心思想是利用Redis的自增特性和时间戳来生成唯一且有序的ID。其生成的ID由两部分组成：

时间戳部分：使用当前时间戳减去一个起始时间戳（例如某个特定日期的时间戳），得到一个相对时间戳。时间戳部分通常占用31位，以秒为单位，这样可以保证在69年内生成的ID是唯一的。
自增序列号部分：使用Redis的INCR命令生成一个自增序列号，确保在相同的时间戳下，ID是唯一的。序列号部分通常占用32位，这意味着每秒可以生成2^32个不同的ID。

在具体实现中，时间戳部分和自增序列号部分通过位运算组合在一起。时间戳部分左移32位，然后与序列号部分进行按位或操作，最终生成一个64位的全局唯一ID。这种设计不仅保证了ID的唯一性，还确保了ID的递增性，有利于数据库索引的创建。

这种实现方式充分利用了Redis的原子操作特性，确保在高并发环境下生成的ID仍然是唯一的。同时，由于时间戳和序列号的结合，生成的ID具有一定的规律性，但又不会直接暴露业务逻辑。

三、代码实现

以下是RedisIdWorker的Java代码实现：

java复制

@Component
public class RedisIdWorker {// 开始时间戳（例如2022年1月1日）private static final long BEGIN_TIMESTAMP = 1640995200L;// 序列号的位数private static final int COUNT_BITS = 32;private StringRedisTemplate stringRedisTemplate;public RedisIdWorker(StringRedisTemplate stringRedisTemplate) {this.stringRedisTemplate = stringRedisTemplate;}// 获取下一个自动生成的IDpublic long nextId(String keyPrefix) {// 1. 生成时间戳LocalDateTime now = LocalDateTime.now();long nowSecond = now.toEpochSecond(ZoneOffset.UTC);long timestamp = nowSecond - BEGIN_TIMESTAMP;// 2. 获取当前日期，用于生成keyString date = now.format(DateTimeFormatter.ofPattern("yyyy:MM:dd"));// 3. 获取自增序列号long count = stringRedisTemplate.opsForValue().increment("incr:" + keyPrefix + ":" + date);// 4. 拼接并返回IDreturn timestamp << COUNT_BITS | count;}
}

代码说明

时间戳部分：通过LocalDateTime获取当前时间戳，并减去起始时间戳BEGIN_TIMESTAMP。
自增序列号部分：使用StringRedisTemplate的increment方法，为每个日期生成一个自增序列号。
ID拼接：将时间戳左移32位，然后与序列号进行按位或操作，生成最终的ID。

四、使用示例

以下是一个简单的使用示例：

java复制

@SpringBootTest
public class RedisIdWorkerTest {@Resourceprivate RedisIdWorker redisIdWorker;@Testpublic void testIdWorker() {// 生成订单IDlong orderId = redisIdWorker.nextId("order");System.out.println("Generated Order ID: " + orderId);// 生成用户IDlong userId = redisIdWorker.nextId("user");System.out.println("Generated User ID: " + userId);}
}