本文共 2392 字,大约阅读时间需要 7 分钟。
保存微博用户之间的关系:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import redis class 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 )) |
关注集合,用于存储用户关注的人的ID,“weibo::user::<id>::following”
粉丝集合,用于存储用户的粉丝的ID,“weibo::user::<id>::fans”
我们在微博中看到喜欢的人可以点击关注,当然我们自己也可以被人关注,我们可以看到自己所有的粉丝,也可以看到自己关注的所有人,有显示是否已互相关注,微博也会显示自己和好友共同关注的人。
在这个类中我们使用无序集合来保存用户的关系,集合的特性是去重,绝对没有重复值。无序集合这种数据结构特别适合保存不能有重复值,并且不在乎存储顺序的数据。就像我们在朋友圈点赞时,先点赞的人不一定排在前面,后点赞的人不一定排在后面显示,他们的显示顺序是无序的,但又是唯一的。后面我们在实现点赞(投票)功能时,也使用的无序集合这种数据结构来存储。
现在可以思考一下如果要取消关注该怎么办呢!?
本文转自戴柏阳的博客博客51CTO博客,原文链接http://blog.51cto.com/daibaiyang119/1962665如需转载请自行联系原作者
daibaiyang119