def get_redis_one(to_user, from_user): to_user_msg = REDIS_DB.get(to_user) if to_user_msg: to_user_msg = json.loads(to_user_msg) count = to_user_msg.get(from_user, 0) to_user_msg[from_user] = 0 REDIS_DB.set(to_user, json.dumps(to_user_msg)) return count else: return 0
def get_msg_one(sender, to_user): user_msg_redis = REDIS_DB.get(to_user) if user_msg_redis: user_msg_dict = json.loads(user_msg_redis) if user_msg_dict.get(sender): # return user_msg_dict.get(sender) user_msg_dict[sender] = 0 else: user_msg_dict = {sender:0} user_msg_redis = json.dumps(user_msg_dict) REDIS_DB.set(to_user,user_msg_redis)
def set_redis(to_user, from_user): # toy:{app:2} to_user_msg = REDIS_DB.get(to_user) if to_user_msg: to_user_msg = json.loads(to_user_msg) if to_user_msg.get(from_user): to_user_msg[from_user] += 1 else: to_user_msg[from_user] = 1 else: to_user_msg = {from_user: 1} REDIS_DB.set(to_user, json.dumps(to_user_msg))
def get_msg_list(user): user_msg_redis = REDIS_DB.get(user) if user_msg_redis: user_msg_dict = json.loads(user_msg_redis) user_msg_dict["count"] = sum(user_msg_dict.values()) else: user_msg_dict = {"count":0} return user_msg_dict
def get_redis_all(to_user): to_user_msg = REDIS_DB.get(to_user) if to_user_msg: to_user_msg = json.loads(to_user_msg) # {"toy": 1, "toy2": 1, "toy3": 2} to_user_msg["count"] = sum(to_user_msg.values()) return to_user_msg else: return {"count": 0}
def save_msg(sender, to_user): # 1.查询一下子龙的Redis是否有数据 user_msg_redis = REDIS_DB.get(to_user) if user_msg_redis: # 2.将子龙的数据反序列化成字典 { sender : n } user_msg_dict = json.loads(user_msg_redis) # 3.判断有没有 sender 的用户发来的消息数量 if user_msg_dict.get(sender): user_msg_dict[sender] += 1 else: user_msg_dict[sender] = 1 # 4.如果子龙是刚建立好的用户,他是没有消息的,字典是空 else: user_msg_dict = {sender: 1} # 5.序列化用户消息字典user_msg_dict user_msg_redis = json.dumps(user_msg_dict) # 6.存回Redis REDIS_DB.set(to_user, user_msg_redis)
def get_user_msg_one(sender, to_user): user_msg_redis = REDIS_DB.get(to_user) if user_msg_redis: user_msg_dict = json.loads(user_msg_redis) if user_msg_dict.get(sender): return user_msg_dict.get(sender)