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

作者：System 时间：2024年08月23日分类：所有,分布式字数：761

这篇文章距离上次修改已过225天，其中的内容可能已经有所变动。

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等。这里的生成方式简单且高效，适合在分布式系统中使用。

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

评论已关闭

推荐阅读