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

新建用户功能:

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
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
 
    def get_by_id(selfid):
        """根据用户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""123@qq.com")
    print(uid)
    user_info = user.try_login("123@qq.com""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.尝试登陆,如果登陆成功则返回用户信息

本文转自戴柏阳的博客博客51CTO博客,原文链接http://blog.51cto.com/daibaiyang119/1962658如需转载请自行联系原作者


daibaiyang119

上一篇:c++友元函数和友元类详解


下一篇:C++温故笔记(三)