392-用户登录业务代码和测试

缺乏、安全感 2022-09-03 07:25 124阅读 0赞

## 用户登录业务代码 ##

**我们先完善一下public.hpp**

#ifndef PUBLIC_H
    #define PUBLIC_H
    
    /*
    server和client的公共文件
    */
    enum EnMsgType
    {
        
        LOGIN_MSG = 1,//登录消息
        LOGIN_MSG_ACK,//登录响应消息
        REG_MSG,//注册消息
        REG_MSG_ACK,//注册响应消息
    
    };
    
    #endif

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70]  
**我们完善一下include下的server下的model下的usermodel.hpp**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 1]  
**我们完善src下的server下的model下的usermodel.cpp**

#include "usermodel.hpp"
    #include "db.hpp"
    #include <iostream>
    using namespace std;
    
    //User表的增加方法
    bool UserModel::insert(User &user)
    {
        
        //1.组装sql语句
        char sql[1024] = {
        0};
        sprintf(sql, "insert into user(name, password, state) values('%s', '%s', '%s')",
                user.getName().c_str(), user.getPwd().c_str(), user.getState().c_str());
    
        MySQL mysql;//定义一个mysql对象 
        if (mysql.connect())//连接成功了 
        {
        
            if (mysql.update(sql))//更新这个sql语句传进去 
            {
        
                //获取插入成功的用户数据生成的主键id
                user.setId(mysql_insert_id(mysql.getConnection()));//拿到数据库生成的id作为用户的id号 
                return true;
            }
        }
    
        return false;
    }
    
    //根据用户号码查询用户信息
    User UserModel::query(int id)
    {
        
        //1.组装sql语句
        char sql[1024] = {
        0};
        sprintf(sql, "select * from user where id = %d", id);
    
        MySQL mysql;//创建mysql对象 
        if (mysql.connect())//连接成功 
        {
        
            MYSQL_RES *res = mysql.query(sql);//调用mysql数据库的查询 
            if (res != nullptr)
            {
        
                MYSQL_ROW row = mysql_fetch_row(res);//获取行，用主键查的，查一行 
                if (row != nullptr)
                {
        
                    User user;
                    user.setId(atoi(row[0]));//atoi把字符串转为int 
                    user.setName(row[1]);
                    user.setPwd(row[2]);
                    user.setState(row[3]);
                    mysql_free_result(res);//释放资源 
                    return user;
                }
            }
        }
    
        return User();
    }
    
    //更新用户的状态信息
    bool UserModel::updateState(User user)
    {
        
        //1.组装sql语句
        char sql[1024] = {
        0};
        sprintf(sql, "update user set state = '%s' where id = %d", user.getState().c_str(), user.getId());
    
        MySQL mysql;//创建mysql对象 
        if (mysql.connect())
        {
        
            if (mysql.update(sql))
            {
        
                return true;
            }
        }
        return false;
    }

**我们完善chatservice.cpp**

#include "chatservice.hpp"
    #include "public.hpp"
    #include <muduo/base/Logging.h>//muduo的日志 
    using namespace std;
    using namespace muduo;
    
    //获取单例对象的接口函数
    ChatService *ChatService::instance()
    {
        
        static ChatService service;
        return &service;
    }
    
    //构造方法，注册消息以及对应的Handler回调操作
    ChatService::ChatService()
    {
        
        //用户基本业务管理相关事件处理回调注册
        _msgHandlerMap.insert({
        LOGIN_MSG, std::bind(&ChatService::login, this, _1, _2, _3)});
        _msgHandlerMap.insert({
        REG_MSG, std::bind(&ChatService::reg, this, _1, _2, _3)});
    
    }
    
    
    //获取消息对应的处理器
    MsgHandler ChatService::getHandler(int msgid)
    {
        
        //记录错误日志，msgid没有对应的事件处理回调
        auto it = _msgHandlerMap.find(msgid);
        if (it == _msgHandlerMap.end())//找不到 
        {
        
            //返回一个默认的处理器，空操作，=按值获取 
            return [=](const TcpConnectionPtr &conn, json &js, Timestamp) {
        
                LOG_ERROR << "msgid:" << msgid << " can not find handler!";//muduo日志会自动输出endl 
            };
        }
        else//成功的话 
        {
        
            return _msgHandlerMap[msgid];//返回这个处理器 
        }
    }
    
    //处理登录业务  id  pwd   检测pwd
    void ChatService::login(const TcpConnectionPtr &conn, json &js, Timestamp time)
    {
        
        int id = js["id"].get<int>();//获取id号 
        string pwd = js["password"];//获取密码 
    
        User user = _userModel.query(id);//查找 
        if (user.getId() == id && user.getPwd() == pwd)//查出来了，登录成功 
        {
        
            if (user.getState() == "online")
    		//该用户已经登录，不允许重复登录
            {
        
                json response;
                response["msgid"] = LOGIN_MSG_ACK;
                response["errno"] = 2;
                response["errmsg"] = "this account is using, input another!";
                conn->send(response.dump());
            }
            else
            {
        
                //登录成功，更新用户状态信息 state offline=>online
                user.setState("online");
                _userModel.updateState(user);
    
                json response;
                response["msgid"] = LOGIN_MSG_ACK;
                response["errno"] = 0;
                response["id"] = user.getId();
                response["name"] = user.getName();
                conn->send(response.dump());            
            }
        }
        else
        {
        
            //该用户不存在，用户存在但是密码错误，登录失败
            json response;
            response["msgid"] = LOGIN_MSG_ACK;
            response["errno"] = 1;
            response["errmsg"] = "id or password is invalid!";
            conn->send(response.dump());
        }
    }
    
    //处理注册业务  name  password
    void ChatService::reg(const TcpConnectionPtr &conn, json &js, Timestamp time)
    {
        
        string name = js["name"];//获取名字 
        string pwd = js["password"];//获取密码 
    
        User user;//创建用户对象 
        user.setName(name);
        user.setPwd(pwd);
        bool state = _userModel.insert(user);//新用户的插入 
        if (state)//插入成功 
        {
        
            //注册成功
            json response;
            response["msgid"] = REG_MSG_ACK;
            response["errno"] = 0;
            response["id"] = user.getId();
            conn->send(response.dump());//回调 ，返回json字符串 
        }
        else//插入失败 
        {
        
            //注册失败
            json response;
            response["msgid"] = REG_MSG_ACK;
            response["errno"] = 1;
            conn->send(response.dump());//回调 ，返回json字符串 
        }
    }

**保存。  
编译成功。**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 2]

## 用户登录业务测试 ##

**我们打开两个终端，启动**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 3]  
**登录成功！**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 4]  
**我输错密码，登录一下，登录失败！**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 5]  
**我们打开mysql看看**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 6]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 7]  
**状态变成online了！**  
**我们现在先把状态改为offline**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 8]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70]: /images/20220829/d31a6057f67d4b1199ff538beac4e419.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 1]: /images/20220829/4d9e4a95c9ef43fe82b27eefc596966e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 2]: /images/20220829/b981940a131d497ab862e58e94c84913.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 3]: /images/20220829/4a46c695b37d4f4db69e4694de57020b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 4]: /images/20220829/8a2f0be62f6042cf91de8e26b5b4804e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 5]: /images/20220829/706f3a3d6d71462281184f8f56e3767e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 6]: /images/20220829/09856730163b43b597f5a48f84360792.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 7]: /images/20220829/4c4cb485e93445a3b5298a5193ec77a8.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 8]: /images/20220829/7d9db0b8f4234f929aa41f7ced318ec2.png