不再介绍数据库做实时排行榜的弊端,直接介绍Redis的有序集合的强大作用。
有序集合的数据和集合一样,不能重复,但每个元素又可以关联一个分数,这个分数可以重复。
需要注意的是,redis版本和命令变化较大,注意运行环境。
运行环境:
redis 库版本:3.3.11
redis版本:3.2.1
1、从MongoDB中导入数据到Redis
生成数据:
import pymongo
import random
handler = pymongo.MongoClient('mongodb://127.0.0.1:27017').chapter_9.rank_data
for i in range(10000, 10100):
data = {'user_id': i,
'score': round(random.random() * 1000, 1)}
handler.insert_one(data)
导入到有序集合中:
import pymongo
import redis
handler = pymongo.MongoClient('mongodb://127.0.0.1:27017').chapter_9.rank_data
client = redis.Redis()
rows = handler.find({}, {'_id': 0})
for row in rows:
client.zadd("rank",{row['user_id']:row['score']})
zadd
的第二个参数应该是字典。
查看导入的数据:
2、显示特定排名
import redis
client = redis.Redis()
count = client.zcard('rank')
print(f'当前共有{count}人')
position = client.zrevrank('rank', 10086)+1
print(f'用户‘10086’的排名为{position}')
3、显示全部排名
import redis
client = redis.Redis()
rank = client.zrevrangebyscore("rank", 100000, 0, withscores=True)
for index,one in enumerate(rank):
print(f'排名:{index+1}, 用户ID:{one[0].decode()}, 积分:{one[1]}')