《Redis实战》—读书笔记—第2章
用redis来控制session会话的数量
时间：2017年10月02日03:07:37
一、生成token并与用户建立关系
①其中有两个数据结构：一个是hset，命名为login:，一个是recent,命名为recent:。
login:用于保存token和用户的映射，recent:用于管理token的时间戳。
数据结构如下:
+--login:--------------------hash---+
|token1 | username1 |
|token2 | username2 |
|token3 | username1 |
| | |
+-----------+-----------------------+
+--zset-key----------zset---+
|token1 | timestamp1 | // 最老的token会被取出来删除
|token2 | timestamp2 |
|token3 | timestamp3 |
+-----------+---------------+
其中大部分复杂操作在更新的时候执行token的时候执行了。
如下为更新token操作:
def update_token(conn, token, user, item=None):
timestamp = time.time()
conn.hset('login:', token, user)
conn.zadd('recent:', token, timestamp)
②我们希望将token数量控制在1000万以下。
QUIT = False
LIMIT = 10_000_000 // 1000万
def clean_sessions(conn):
while not QUIT:
size = conn.zcard('recent:') // 找出目前已经存在的数量
if size <= LIMIT: // 小于1000万则睡一秒继续
time.sleep(1)
continue
end_index = min(size - LIMIT, 100)
tokens = conn.zrange('rencent:', 0, end_index -1) 找到最老时间的token们
conn.hdel('login:', *tokens) // 删除
conn.zrem('recent:', *tokens)
此代码可轮询，也可以按照周期性运行。
