使用 Redis 生成分布式唯一 ID 的实现与优化

import redis
 
class RedisIdGenerator:
    def __init__(self, redis_host='localhost', redis_port=6379, redis_db=0):
        self.redis_client = redis.StrictRedis(host=redis_host, port=redis_port, db=redis_db)
 
    def generate_id(self, key_prefix):
        # 使用 Redis 的原子操作 INCR 和 GET 来生成唯一 ID
        self.redis_client.incr(key_prefix, 1)  # 原子递增
        next_id = self.redis_client.get(key_prefix)  # 获取当前值
        return next_id
 
# 使用示例
redis_id_generator = RedisIdGenerator()
user_id = redis_id_generator.generate_id('user:id:counter')
print(f'Generated unique ID: {user_id}')

这段代码展示了如何使用Redis的INCR命令生成唯一ID。首先，我们连接到Redis服务器，然后定义了一个方法用来生成唯一ID。在实际使用中，我们可以通过传入不同的key\_prefix来生成不同类型的ID，例如用户ID、订单ID等。这里的生成方式简单且高效，适合在分布式系统中使用。