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

Posted 2020-10-05

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，这个类实现的就是这个功能，这个类我们在后面会经常用到。

本文出自 “戴柏阳的博客” 博客，请务必保留此出处http://daibaiyang119.blog.51cto.com/3145591/1962390