使用Redis构建简易社交网站(2)-处理用户关系

admin • 2023-12-04 20:33 • 5G

目的

本文目的：实现用户关注和取消关注功能。（完整代码附在文章末尾）

实战例题

编写 follow(uid, other_uid) 函数，实现关注用户的功能，具体参数与要求如下：

方法参数uid为当前用户编号，other_uid为被关注的用户编号；
避免重复关注的实现：如果被关注的用户编号已经在当前用户的关注列表following:{uid}中，则不重复关注，直接返回None；
建立关注关系的实现：使用事务一次性提交：
- 将被关注的用户编号加入到当前用户的关注列表following:{uid}中，分值为当前时间戳。
- 将当前用户编号加入到被关注用户的粉丝列表中followers:{other_uid}，分值为当前时间戳。
修改统计数据的实现：若关系建立成功，则使用事务一次性提交：
- 将当前用户详情user:{uid}中的关注数following加1
- 将被关注用户详情user:{other_uid}中的粉丝数followers加1
返回执行结果的实现：返回True

编写 unfollow(uid, other_uid) 函数，实现取消关注的功能，具体参数与要求如下：

方法参数uid为当前用户编号，other_uid为被取消关注的用户编号；
避免重复取消关注的实现：如果被关注的用户编号已经不在当前用户的关注列表following:{uid}中，则不重复取消关注，直接返回None；
删除关注关系的实现：使用事务一次性提交：
- 从当前用户的关注列表following:{uid}中移除被取消关注用户编号。
- 从被取消关注用户的粉丝列表中followers:{other_uid}移除当前用户编号。
修改统计数据的实现：若关系删除成功，则使用事务一次性提交：
- 将当前用户详情user:{uid}中的关注数following减1
- 将被取消关注用户详情user:{other_uid}中的粉丝数followers减1
返回执行结果的实现：返回True

注意： 关注列表和粉丝列表均为有序集合，存储成员时，分值均为当前时间戳；用户详情为上一关中创建的哈希结构。

测试输入：

预期输出：

测试 follow 方法...
用户 9 关注 用户 4
关注结果: True
用户 9 的关注列表内容为: ['4']
用户 4 的粉丝列表内容为: ['9']
用户 9 的用户详情为: {'login_name': 'test_user9', 'posts': '0', 'real_name': 'Test user9', 'followers': '0', 'following': '1', 'id': '9'}
用户 4 的用户详情为: {'login_name': 'test_user4', 'posts': '0', 'real_name': 'Test user4', 'followers': '1', 'following': '0', 'id': '4'}

用户 9 再次关注 用户 4
关注结果: None

测试 unfollow 方法...
用户 9 取消关注 用户 4
取消关注结果: True
用户 9 的关注列表内容为: []
用户 4 的粉丝列表内容为: []
用户 9 的用户详情为: {'login_name': 'test_user9', 'posts': '0', 'real_name': 'Test user9', 'followers': '0', 'following': '0', 'id': '9'}
用户 4 的用户详情为: {'login_name': 'test_user4', 'posts': '0', 'real_name': 'Test user4', 'followers': '0', 'following': '0', 'id': '4'}

用户 9 再次取消关注 用户 4
取消关注结果: None

code.py

#code.py
#-*- coding:utf-8 -*-

import re
import time
import redis

conn = redis.Redis()

# 关注用户
def follow(uid, other_uid):
    # 请在下面完成要求的功能
    #********* Begin *********#
    fkey1 = "following:%s"%(uid)
    fkey2 = "followers:%s"%(other_uid)

    if conn.zscore(fkey1, other_uid):
        return None

    now = time.time()
    pipe = conn.pipeline(True)
    pipe.zadd(fkey1, other_uid, now)
    pipe.zadd(fkey2, uid, now)
    following, followers = pipe.execute()

    pipe.hincrby("user:%s"%(uid), 'following', int(following))
    pipe.hincrby("user:%s"%(other_uid), 'followers', int(followers))
    pipe.execute()

    return True
    #********* End *********#

# 取消关注
def unfollow(uid, other_uid):
    # 请在下面完成要求的功能
    #********* Begin *********#
    fkey1 = "following:%s"%(uid)
    fkey2 = "followers:%s"%(other_uid)

    if not conn.zscore(fkey1, other_uid):
        return None

    pipe = conn.pipeline(True)
    pipe.zrem(fkey1, other_uid)
    pipe.zrem(fkey2, uid)
    following, followers = pipe.execute()

    pipe.hincrby("user:%s"%(uid), 'following', -int(following))
    pipe.hincrby("user:%s"%(other_uid), 'followers', -int(followers))
    pipe.execute()

    return True
    #********* End *********#

# 创建新用户
def create_user(login_name, real_name):
    login_name = login_name.lower()
    if conn.hget("users", login_name):
        return None

    uid = conn.incr("user:id")
    pipe = conn.pipeline(True)
    pipe.hset("users", login_name, uid)
    pipe.hmset("user:%i"%(uid), {
        'login_name': login_name,
        'id': uid,
        'real_name': real_name,
        'followers': 0,
        'following': 0,
        'posts': 0,
        'last_signup': time.time(),
    })
    pipe.execute()

    return uid

# 为用户创建新动态
def create_post(uid, content):
    pipe = conn.pipeline(True)
    pipe.hget("user:%i"%(uid), 'login_name')
    pipe.incr("post:id")
    login_name, pid = pipe.execute()

    if not login_name:
        return None

    pipe.hmset("post:%i"%(pid), {
        'id': pid,
        'uid': uid,
        'content': content,
        'posted': time.time(),
        'user_name': login_name,
    })
    pipe.hincrby("user:%i"%(uid), 'posts')
    pipe.execute()

    return pid

read.py

#read.py
#-*- coding:utf-8 -*-

import os
import sys
import time
import redis
import pprint
from code import *

conn = redis.Redis()
retry_time = 0
while True:
    try:
        conn.ping()
        break
    except redis.exceptions.ConnectionError:
        os.system("redis-server > /dev/null 2>&1 &")
        retry_time += 1
        if retry_time > 3:
            break

pipe = conn.pipeline(True)
pipe.delete("users", "user:id")
keys = conn.keys("user:*") + conn.keys("followers:*") + conn.keys("following:*")
for key in keys:
    pipe.delete(key)
pipe.execute()

# 创建测试数据
join_str = " "
for i in xrange(10):
    login_name = "test_user%i"%(i+1)
    real_name = join_str.join(login_name.split("_")).capitalize()
    create_user(login_name, real_name)

uid = int(sys.stdin.readline().strip())
other_uid = int(sys.stdin.readline().strip())

print "测试 follow 方法..."
print "用户 %i 关注 用户 %i"%(uid, other_uid)
f_result = follow(uid, other_uid)
print "关注结果: " + str(f_result)
print "用户 %i 的关注列表内容为: %s"%(uid, str(conn.zrange("following:%i"%(uid), 0, -1)))
print "用户 %i 的粉丝列表内容为: %s"%(other_uid, str(conn.zrange("followers:%i"%(other_uid), 0, -1)))
uid_info = conn.hgetall("user:%i"%(uid))
uid_info.pop("last_signup", "404")
other_uid_info = conn.hgetall("user:%i"%(other_uid))
other_uid_info.pop("last_signup", "404")
print "用户 %i 的用户详情为: %s"%(uid, str(uid_info))
print "用户 %i 的用户详情为: %s"%(other_uid, str(other_uid_info))
print

print "用户 %i 再次关注 用户 %i"%(uid, other_uid)
oth_f_result = follow(uid, other_uid)
print "关注结果: " + str(oth_f_result)
print

print "测试 unfollow 方法..."
print "用户 %i 取消关注 用户 %i"%(uid, other_uid)
unf_result = unfollow(uid, other_uid)
print "取消关注结果: " + str(unf_result)
print "用户 %i 的关注列表内容为: %s"%(uid, str(conn.zrange("following:%i"%(uid), 0, -1)))
print "用户 %i 的粉丝列表内容为: %s"%(other_uid, str(conn.zrange("followers:%i"%(other_uid), 0, -1)))
uid_info = conn.hgetall("user:%i"%(uid))
uid_info.pop("last_signup", "404")
other_uid_info = conn.hgetall("user:%i"%(other_uid))
other_uid_info.pop("last_signup", "404")
print "用户 %i 的用户详情为: %s"%(uid, str(uid_info))
print "用户 %i 的用户详情为: %s"%(other_uid, str(other_uid_info))
print

print "用户 %i 再次取消关注 用户 %i"%(uid, other_uid)
oth_unf_result = unfollow(uid, other_uid)
print "取消关注结果: " + str(oth_unf_result)

pipe = conn.pipeline(True)
pipe.delete("users", "user:id")
keys = conn.keys("user:*") + conn.keys("followers:*") + conn.keys("following:*")
for key in keys:
    pipe.delete(key)
pipe.execute()

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。

THE END

bootstrap mybatis

二维码

苹果TF签名全称TestFlight签名，需要怎么做才可以上架呢？

< <上一篇

关于leapmotion手势识别在unity中的应用（保姆级教程2023年）

下一篇>>

搜索内容

使用Redis构建简易社交网站(2)-处理用户关系

目的

相关知识

redis基本命令

python基本命令

实战例题

最新文章

分类

标签云