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，这个类实现的就是这个功能，这个类我们在后面会经常用到。
本文转自戴柏阳的博客博客51CTO博客，原文链接http://blog.51cto.com/daibaiyang119/1962390如需转载请自行联系原作者
daibaiyang119