398-添加好友业务代码和测试

Bertha 。 2022-09-03 12:29 200阅读 0赞

**我们在mysql数据库有Friend表**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70]  
表中两个字段，一个是用户id，一个是用户的friend的id，这种关系不用写多次，所以用联合主键。我跟你是好友写一般就可以了。  
**我把你添加好友，服务器会给我返回好友列表信息，然后我根据列表上的用户的id号进行聊天。**

## 添加好友业务代码 ##

**我们完善public.hpp**

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

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 1]  
**我们完善chatservice.hpp**  
新增代码

#include "friendmodel.hpp"
    //添加好友业务
    void addFriend(const TcpConnectionPtr &conn, json &js, Timestamp time);
    
    FriendModel _friendModel;

完整代码

#ifndef CHATSERVICE_H
    #define CHATSERVICE_H
    
    #include <muduo/net/TcpConnection.h>
    #include <unordered_map>//一个消息ID映射一个事件处理 
    #include <functional>
    #include <mutex>
    using namespace std;
    using namespace muduo;
    using namespace muduo::net;
    
    #include "usermodel.hpp"
    #include "offlinemessagemodel.hpp"
    #include "friendmodel.hpp"
    #include "json.hpp"
    using json = nlohmann::json;
    
    //表示处理消息的事件回调方法类型，事件处理器，派发3个东西 
    using MsgHandler = std::function<void(const TcpConnectionPtr &conn, json &js, Timestamp)>;
    
    //聊天服务器业务类
    class ChatService
    {
        
    public:
        //获取单例对象的接口函数
        static ChatService *instance();
        //处理登录业务
        void login(const TcpConnectionPtr &conn, json &js, Timestamp time);
        //处理注册业务
        void reg(const TcpConnectionPtr &conn, json &js, Timestamp time);
        //一对一聊天业务
        void oneChat(const TcpConnectionPtr &conn, json &js, Timestamp time);
        //添加好友业务
        void addFriend(const TcpConnectionPtr &conn, json &js, Timestamp time);
        //处理客户端异常退出
        void clientCloseException(const TcpConnectionPtr &conn);
        //服务器异常，业务重置方法
        void reset();
        //获取消息对应的处理器
        MsgHandler getHandler(int msgid);
    private:
        ChatService();//单例 
    
        //存储消息id和其对应的业务处理方法，消息处理器的一个表，写消息id对应的处理操作 
        unordered_map<int, MsgHandler> _msgHandlerMap;
        //存储在线用户的通信连接，用户的id， TcpConnectionPtr
        unordered_map<int, TcpConnectionPtr> _userConnMap;
        //定义互斥锁，保证_userConnMap的线程安全
        mutex _connMutex;
        //数据操作类对象
        UserModel _userModel;
        OfflineMsgModel _offlineMsgModel;   
        FriendModel _friendModel;
    
    };
    
    #endif

**我们完善chatservice.cpp**  
新增代码

_msgHandlerMap.insert({
        ADD_FRIEND_MSG, std::bind(&ChatService::addFriend, this, _1, _2, _3)});
    //添加好友业务 格式： msgid id friendid
    void ChatService::addFriend(const TcpConnectionPtr &conn, json &js, Timestamp time)
    {
        
        int userid = js["id"].get<int>();//当前用户的id 
        int friendid = js["friendid"].get<int>();//添加好友的id 
    
        //存储好友信息
        _friendModel.insert(userid, friendid);
    }
    //处理登录业务  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
            {
        
                //登录成功，记录用户连接信息，要考虑线程安全，因为多线程访问 
                {
        
                    lock_guard<mutex> lock(_connMutex);
                    _userConnMap.insert({
        id, conn});
                }//加个作用域，出了这个右括号就自动解锁
                
                //登录成功，更新用户状态信息 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();
                //查询该用户是否有离线消息
                vector<string> vec = _offlineMsgModel.query(id);//查当用户id 
                if (!vec.empty())//不为空 
                {
        
                    response["offlinemsg"] = vec;//json库可以和容器之间序列化和反序列化 
                    //读取该用户的离线消息后，把该用户的所有离线消息删除掉
                    _offlineMsgModel.remove(id);
                }
                //查询该用户的好友信息并返回
                vector<User> userVec = _friendModel.query(id);
                if (!userVec.empty())
                {
        
                    vector<string> vec2;
                    for (User &user : userVec)
                    {
        
                        json js;
                        js["id"] = user.getId();
                        js["name"] = user.getName();
                        js["state"] = user.getState();
                        vec2.push_back(js.dump());
                    }
                    response["friends"] = vec2;
                }
                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());
        }
    }

完整代码

#include "chatservice.hpp"
    #include "public.hpp"
    #include <muduo/base/Logging.h>//muduo的日志 
    #include <vector>
    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)});
        _msgHandlerMap.insert({
        ONE_CHAT_MSG, std::bind(&ChatService::oneChat, this, _1, _2, _3)});   
        _msgHandlerMap.insert({
        ADD_FRIEND_MSG, std::bind(&ChatService::addFriend, this, _1, _2, _3)});
    }
    
    //服务器异常，业务重置方法
    void ChatService::reset()
    {
        
        //把online状态的用户，设置成offline
        _userModel.resetState();
    }
    
    //获取消息对应的处理器
    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
            {
        
                //登录成功，记录用户连接信息，要考虑线程安全，因为多线程访问 
                {
        
                    lock_guard<mutex> lock(_connMutex);
                    _userConnMap.insert({
        id, conn});
                }//加个作用域，出了这个右括号就自动解锁
                
                //登录成功，更新用户状态信息 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();
                //查询该用户是否有离线消息
                vector<string> vec = _offlineMsgModel.query(id);//查当用户id 
                if (!vec.empty())//不为空 
                {
        
                    response["offlinemsg"] = vec;//json库可以和容器之间序列化和反序列化 
                    //读取该用户的离线消息后，把该用户的所有离线消息删除掉
                    _offlineMsgModel.remove(id);
                }
                //查询该用户的好友信息并返回
                vector<User> userVec = _friendModel.query(id);
                if (!userVec.empty())
                {
        
                    vector<string> vec2;
                    for (User &user : userVec)
                    {
        
                        json js;
                        js["id"] = user.getId();
                        js["name"] = user.getName();
                        js["state"] = user.getState();
                        vec2.push_back(js.dump());
                    }
                    response["friends"] = vec2;
                }
                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字符串 
        }
    }
    
    //处理客户端异常退出
    void ChatService::clientCloseException(const TcpConnectionPtr &conn)
    {
        
        User user;
        {
        
            lock_guard<mutex> lock(_connMutex);
            for (auto it = _userConnMap.begin(); it != _userConnMap.end(); ++it)//用迭代器 
            {
        
                if (it->second == conn)
                {
        
                    //从map表删除用户的链接信息
                    user.setId(it->first);
                    _userConnMap.erase(it);
                    break;
                }
            }
        }
        //更新用户的状态信息
        if (user.getId() != -1)
        {
        
            user.setState("offline");
            _userModel.updateState(user);
        }
    }
    
    //一对一聊天业务
    void ChatService::oneChat(const TcpConnectionPtr &conn, json &js, Timestamp time)
    {
        
        int toid = js["toid"].get<int>();//获取对方的id号 
    
        {
        
            lock_guard<mutex> lock(_connMutex);//访问连接信息表，要保证线程安全 
            auto it = _userConnMap.find(toid);//查找对方id号 
            if (it != _userConnMap.end())//找到了 
            {
        
                //toid在线，转发消息  服务器主动推送消息给toid用户
                it->second->send(js.dump());
                return;
            }
        }
    
        //toid不在线，存储离线消息
        _offlineMsgModel.insert(toid, js.dump());
    }
    
    //添加好友业务 格式： msgid id friendid
    void ChatService::addFriend(const TcpConnectionPtr &conn, json &js, Timestamp time)
    {
        
        int userid = js["id"].get<int>();//当前用户的id 
        int friendid = js["friendid"].get<int>();//添加好友的id 
    
        //存储好友信息
        _friendModel.insert(userid, friendid);
    }

**我们在include的server的model下创建文件：friendmodel.hpp**

#ifndef FRIENDMODEL_H
    #define FRIENDMODEL_H
    
    #include "user.hpp"
    #include <vector>
    using namespace std;
    
    //维护好友信息的操作接口方法
    class FriendModel
    {
        
    public:
        //添加好友关系
        void insert(int userid, int friendid);
    
        //返回用户好友列表
        vector<User> query(int userid);
    };
    
    #endif

**我们做下面这两张表的联合查询就可以了。**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 2]  
**我们在src下的server下的model下创建friendmodel.cpp**

#include "friendmodel.hpp"
    #include "db.hpp"
    
    //添加好友关系
    void FriendModel::insert(int userid, int friendid)
    {
        
        //1.组装sql语句
        char sql[1024] = {
        0};
        sprintf(sql, "insert into friend values(%d, %d)", userid, friendid);
    
        MySQL mysql;
        if (mysql.connect())
        {
        
            mysql.update(sql);
        }
    }
    
    //返回用户好友列表
    vector<User> FriendModel::query(int userid)
    {
        
        // 1.组装sql语句
        char sql[1024] = {
        0};
    
        sprintf(sql, "select a.id,a.name,a.state from user a inner join friend b on b.friendid = a.id where b.userid=%d", userid);
    
        vector<User> vec;
        MySQL mysql;
        if (mysql.connect())
        {
        
            MYSQL_RES *res = mysql.query(sql);
            if (res != nullptr)
            {
        
                //把userid的好友请求的用户信息都存入数组中 
                MYSQL_ROW row;
                while((row = mysql_fetch_row(res)) != nullptr)
                {
        
                    User user;
                    user.setId(atoi(row[0]));
                    user.setName(row[1]);
                    user.setState(row[2]);
                    vec.push_back(user);
                }
                mysql_free_result(res);
                return vec;
            }
        }
        return vec;
    }

**编译成功**  
![在这里插入图片描述][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]  
**我们先看看mysql数据库的user表有什么？**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 5]

**我先登录上linzeyu用户**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 6]  
**然后linzeyu添加zhangsan为好友**

{
        "msgid":6,"id":1,"friendid":2}

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 7]  
**我们打开mysql数据库看一下**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 8]  
**然后现在linzeyu用户先退出，然后重新登录**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 9]  
**显示出好友列表了！**

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70]: /images/20220829/f4274b3cd1ae49af804b96ad6a48a947.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 1]: /images/20220829/bf661d7b33854b8eb2cd31039d114f63.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 2]: /images/20220829/e22ca29d906b429c9aec2bde563be14a.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 3]: /images/20220829/4f20c1b02e864032919decea2cbe0699.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 4]: /images/20220829/da270c0e0f974518a73391f9cce2e773.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 5]: /images/20220829/878ab16a3c75480595b0da92fdde4bfa.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 6]: /images/20220829/18d35078b88e4e3a93365e15ff400df7.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 7]: /images/20220829/2ae56295320e4f09981f0b8bcf7aec77.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 8]: /images/20220829/bdcf6806838b4a6ca8ebd3579a59aa7b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xJTlpFWVU2NjY_size_16_color_FFFFFF_t_70 9]: /images/20220829/6128e0d81f664930900cd1b78268c11f.png