保存微博用户之间的关系:
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