ID生成器:
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,这个类实现的就是这个功能,这个类我们在后面会经常用到。
转载于:https://blog.51cto.com/daibaiyang119/1962390