Redis实现微博后台业务逻辑系列（四）
import redisclass RelatoinShip(object):
"""使用无序集合键保存用户关系"""
def __init__(self, client):
self.client = client
def follow(self, fans, target):
"""关注某人"""
# 将用户添加到目标用户粉丝集合中
target_fans_set = "weibo::user::" + str(target) + "::fans"
self.client.sadd(target_fans_set, fans)
# 将目标用户id添加到用户自身的关注集合中
self_following_set = "weibo::user::" + str(fans) + "::following"
self.client.sadd(self_following_set, target)
def is_following(self, fans, target):
"""检查fans用户是否关注了target用户"""
self_following_set = "weibo::user::" + str(fans) + "::following"
return self.client.sismember(self_following_set, target)
def is_following_each_other(self, user_a, user_b):
"""检查两个用户是否已互相关注"""
return self.is_following(user_a, user_b) and self.is_following(user_b, user_a)
def get_all_following(self, user):
"""返回用户关注的所有人"""
self_following_set = "weibo::user::" + str(user) + "::following"
return self.client.smembers(self_following_set)
def get_all_fans(self, user):
"""返回用户的所有粉丝"""
target_fans_set = "weibo::user::" + str(user) + "::fans"
if not self.client.exists(target_fans_set):
return False
else:
return self.client.smembers(target_fans_set)
def common_following(self, user_a, user_b):
"""返回两个用户共同关注的人"""
user_a_following = "weibo::user::" + str(user_a) + "::following"
user_b_following = "weibo::user::" + str(user_b) + "::following"
return self.client.sinter(user_a_following, user_b_following)
if __name__ == "__main__":
redis_client = redis.StrictRedis()
relation = RelatoinShip(redis_client)
relation.follow(10086, 12345)
relation.follow(10010, 12345)
print(relation.is_following(10086, 12345))
print(relation.get_all_following(10086))
print(relation.get_all_fans(12345))
print(relation.common_following(10086, 10010))
页:
[1]