redis 粉丝相互关注

原创

mob649e81697507 2024-10-17 13:31:56 ©著作权

文章标签 Redis 数据结构社交网络 文章分类 Redis 数据库

©著作权归作者所有：来自51CTO博客作者mob649e81697507的原创作品，请联系作者获取转载授权，否则将追究法律责任

Redis 粉丝与相互关注的实现

在现代应用程序中，社交网络功能是吸引用户的重要元素之一，用户间的“粉丝”和“相互关注”关系尤为重要。这篇文章将介绍如何利用 Redis 来实现这一功能，详细讨论相关数据结构的设计，并通过代码示例展现具体实现过程。

一、介绍 Redis

Redis 是一个开源的内存数据结构存储系统，可以作为数据库、缓存和消息中间件。它的数据结构如字符串、哈希、列表、集合和有序集合等，十分适合存储社交网络中的关系数据。

二、粉丝和相互关注的基本概念

粉丝: 用户 A 关注了用户 B，那么 B 就是 A 的粉丝。
相互关注: 用户 A 关注了用户 B，且 B 也关注了 A，那么我们称 A 和 B 是相互关注的。

三、数据结构设计

在 Redis 中，我们可以使用集合（Sets）来存储用户之间的关注关系。具体来说，假设用户 ID 为整型，我们可以设计以下数据结构：

user:1:follow – 表示用户 1 关注的用户集合。
user:2:follow – 表示用户 2 关注的用户集合。
user:1:fans – 表示用户 1 的粉丝集合。
user:2:fans – 表示用户 2 的粉丝集合。

示例

# 用户 1 关注 用户 2
SADD user:1:follow 2
# 用户 2 关注 用户 3
SADD user:2:follow 3
# 用户 2 关注 用户 1
SADD user:2:follow 1

四、实现关注功能

我们可以通过以下 Python 代码来实现用户关注其他用户的功能。

import redis

# 连接到 Redis
r = redis.StrictRedis(host='localhost', port=6379, db=0)

def follow(user_id, follow_id):
    r.sadd(f'user:{user_id}:follow', follow_id)
    r.sadd(f'user:{follow_id}:fans', user_id)

# 示例: 用户 1 关注 用户 2
follow(1, 2)
follow(2, 1)  # 用户 2 也关注了用户 1

五、查询粉丝和相互关注

要查询某个用户的粉丝和相互关注的用户，我们可以使用以下代码：

def get_fans(user_id):
    return r.smembers(f'user:{user_id}:fans')

def get_mutual_follows(user_id1, user_id2):
    user1_follows = r.smembers(f'user:{user_id1}:follow')
    user2_follows = r.smembers(f'user:{user_id2}:follow')
    
    return user1_follows.intersection(user2_follows)

# 示例: 获取用户 1 的粉丝
print(get_fans(1))  # 输出: {2}

# 示例: 获取用户 1 和用户 2 的相互关注
print(get_mutual_follows(1, 2))  # 输出: {2}

六、数据可视化

为了更好地理解用户之间的关系，我们可以绘制饼状图和 ER 图。

1. 饼状图

以下是表示用户之间关系的饼状图示例。使用 Mermaid 的语法来表示。

pie
    title 用户关系
    "用户 1 关注用户 2": 1
    "用户 2 关注用户 1": 1
    "用户 2 关注用户 3": 1

2. ER 图

接下来是用户关注关系的实体关系图（ER 图）。

erDiagram
    USER {
        INT id
        STRING name
    }
    FOLLOW {
        INT follower_id
        INT followed_id
    }
    USER ||--o{ FOLLOW : follows