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