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

aaahd 发表于 2018-11-3 11:48:58

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

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

查看完整版本: Redis实现微博后台业务逻辑系列（二）