Redis实现微博后台业务逻辑系列（二）

423212

ID生成器：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

import redis class IdGenerator(object):     """生成用户ID并返回"""     def __init__(self, key, client):         self.key = key         self.client = client     def init(self, n):         self.client.set(self.key, n)     def gen(self):         new_id = self.client.incr(self.key)         return int(new_id) if __name__ == "__main__":     redis_client = redis.StrictRedis()             generator = IdGenerator("user-id", redis_client)  # 创建一个ID生成器     generator.init(10000)  # 保留前一万个id     print(generator.gen())  # 10001     print(generator.gen())  # 10002

    这个类我们实现的是自动生成用户ID，我们知道每当我们注册一个账号时，系统自动就会给我们分配一个用户ID，这个类实现的就是这个功能，这个类我们在后面会经常用到。