【Redis实战】快速简单搭建聊天室04——实现页面功能

01    def get_chat_list(self):
02        """
03        你需要实现这个方法
04     
05        获取聊天消息列表。
06        使用Redis的列表实现。Key为self.chat_list属性中保存的字符串，可以直接使用。
07        获取列表右端20条信息，但不要删除。
08     
09        需要注意，从Redis中获取的数据一个列表，列表里面是bytes型的字符串，所以需要
10        先把这个列表展开，把里面的bytes型的字符串解密为普通字符串以后再用json解析为
11        字典。接下来讲解析出来的字典放入一个新的列表中。最后返回新的列表。
12        :return: 包含字典的列表
13        """
14        chat_list = self.client.lrange(self.chat_list, -20, -1)
15        chat_info_list = []
16        for chat in chat_list:
17            chat_info = json.loads(chat.decode())
18            chat_info_list.append(chat_info)
19        return chat_info_list

01    127.0.0.1:6379> lpush chat_list '{"msg": "我是人工添加的消息", "nick": "青南", "post_time": "2018-07-22 16:15:00"}'
02    (integer) 1
03    127.0.0.1:6379> lpush chat_list '{"msg": "我是青南的助手", "nick": "青南的助手", "post_time": "2018-07-22 16:15:00"}'
04    (integer) 2
05    127.0.0.1:6379>

01    def push_chat_info(self, chat_info):
02        """
03        把聊天信息存入列表的右侧。
04     
05        使用Redis的集合实现，对应的Key为self.chat_list中保存的字符串。
06        把chat_info字典先转化为JSON字符串，再存入Redis中列表中。
07     
08        为了防止列表消息太长，因此需要使用ltrim命令删除多余的信息，只保留
09        列表最右侧的20条
10        :param chat_info: 字典，格式为{'msg': '信息', 'nick': '青南', 'post_time': '2018-07-22 10:00:12'}
11        :return: None
12        """
13        self.client.rpush(self.chat_list, json.dumps(chat_info))
14        self.client.ltrim(self.chat_list, -20, -1)

01    def set_nick_msg_expire_time(self, nick, msg):
02        """
03        你需要实现这个方法
04     
05        设定Key的过期时间，这个功能的目的是限定同一个用户在2分钟内不能发送同样的内容。
06        为了防止信息太长，因此把信息编码为md5以后再与昵称拼接以缩短Key的长度。
07        使用Redis的字符串实现，字符串的Key为昵称+信息的md5编码，值为1.使用set命令的
08        ex参数设定Key的过期时间为120秒，时间到了以后Redis会自动删除这个Key
09        :param nick: 昵称
10        :param msg: 信息
11        :return: None
12        """
13        msg_md5 = hashlib.md5(msg.encode()).hexdigest()
14        duplicate_msg_check_flag = nick + msg_md5
15        self.client.set(duplicate_msg_check_flag, 1, ex=120)

01    def get_nick_msg_expire_time(self, nick, msg):
02        """
03        获取某一个昵称发送某一条消息的过期时间。这个功能的作用是
04        为了防止同一个用户短时间发送大量相同信息刷屏。
05     
06        为了防止信息太长，因此把信息编码为md5以后再与昵称拼接以缩短Key的长度。
07        使用Redis的ttl命令来实现，ttl命令如果返回None，说明不存在这个Key，
08        返回None。如果ttl返回-1，说明这个Key没有设定过期时间，这个Key可以一直存在
09        如果ttl返回一个大于0的正整数，说明在这个整数对应的秒过于以后，Redis会自动
10        删除这个Key
11     
12        :param nick: 昵称
13        :param msg: 信息
14        :return: None 或者 数字
15        """
16        msg_md5 = hashlib.md5(msg.encode()).hexdigest()
17        duplicate_msg_check_flag = nick + msg_md5
18        expire_time = self.client.ttl(duplicate_msg_check_flag)
19        return expire_time