微博后台业务逻辑
- 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://www.bubuko.com/infodetail-2291020.html