Redis实现微博后台业务逻辑系列

Posted 2020-10-05

新建用户功能：

import redis

class User(object):
    """使用Redis散列键保存用户信息，并尝试登陆"""
    def __init__(self, client):
        self.client = client
        self.key = "weibo::email_to_uid"
    
    def create(self, name, passwd, email):
        """创建用户"""
        # 生成新的ID
        new_id = IdGenerator("weibo::uid", self.client).gen()
        # 生成新用户的键
        user_key = "weibo::user::" + str(new_id)
        # 创建用户，使用散列键保存用户信息，使用hmset时需要将key、value写成字典格式传递给redis
        self.client.hmset(user_key, {"id": new_id, "name": name, "passwd": passwd, "email": email})
        # 使用散列键保存email和ID的对应关系
        self.client.hset(self.key, email, new_id)
        return new_id.decode()

    def get_by_id(self, id):
        """根据用户ID返回用户信息"""
        user_key = "weibo::user::" + str(id)
        return self.client.hgetall(user_key)
        
    def try_login(self, email, passwd):
        """尝试登陆，返回用户信息"""
        uid = self.client.hget(self.key, email)
        if uid is None:
            return None
        
        # 获取用户信息，并将字典内二进制数据转换成字符串数据并重新存储
        user_info = self.get_by_id(uid.decode())
        user = dict()
        for key,value in user_info.items():
            user[key.decode()] = value.decode()
        
        if user["passwd"] == passwd:
            return user

            
if __name__ == "__main__":
    redis_client = redis.StrictRedis()
    user = User(redis_client)
    uid = user.create("daiby", "abcd1234", "[email protected]")
    print(uid)
    user_info = user.try_login("[email protected]", "abcd1234")
    print(user_info)

  在这个类里，我们使用Redis散列键的数据结构来存储用户账户信息。为什么要用散列键来存储呢？Redis散列键的结构是：hmset key field value field value ...，这样我们使用"weibo::user::<id>"这个散列键key来保存用户的ID、邮箱、姓名、密码等信息，再使用"weibo::email_to_uid"这个散列键来保存email和id之间的关联关系，为什么要保存email和id之间的关联关系呢？因为我们使用email名登陆时，通过"weibo::email_to_uid"这个散列键，取得对应的ID了。类中我们实现了三个方法：

    1.创建新用户

    2.根据用户ID返回用户信息

    3.尝试登陆，如果登陆成功则返回用户信息

本文出自 “戴柏阳的博客” 博客，请务必保留此出处http://daibaiyang119.blog.51cto.com/3145591/1962658