395-点对点聊天业务代码和测试

在一对一的聊天业务中，发送过来的json数据都包含了哪些字段？
首先我们完善一下public.hpp
增加字段 ：ONE_CHAT_MSG,//聊天消息

#ifndef PUBLIC_H
#define PUBLIC_H

/*
server和client的公共文件
*/
enum EnMsgType
{
    LOGIN_MSG = 1,//登录消息
    LOGIN_MSG_ACK,//登录响应消息
    REG_MSG,//注册消息
    REG_MSG_ACK,// 注册响应消息
    ONE_CHAT_MSG,//聊天消息
};

#endif

当我们登录都是用id号登录的，登录成功之后，聊天的话会发送：

msgid
id:1
from:"linzeyu"
to:3
msg:"你好“

我要先找你这个用户在不在线，如果在线，直接找到你这个用户的对应的connection（所有用户的connection都存起来了），如果不在线，存在离校消息表，等你上线了，这个消息通知到你。

点对点聊天业务代码

首先，我们完善chatservice.hpp
新增代码：

//一对一聊天业务
void oneChat(const TcpConnectionPtr &conn, json &js, Timestamp time);

然后我们完善chatservice.cpp
新增代码如下

 _msgHandlerMap.insert({ONE_CHAT_MSG, std::bind(&ChatService::oneChat, this, _1, _2, _3)});
//一对一聊天业务
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不在线，存储离线消息
}

完整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)});
    _msgHandlerMap.insert({ONE_CHAT_MSG, std::bind(&ChatService::oneChat, 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
        {
            //登录成功，记录用户连接信息，要考虑线程安全，因为多线程访问 
            {
                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();
            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不在线，存储离线消息
}

编译成功

测试

注册的消息：{"msgid":3,"name":"zhangsan","password":"123"}
登录的消息：{"msgid":1,"id":1,"password":"123456"}
我们打开2个终端，启动
我们先注册一个新用户

注册成功，id号为2
我们打开mysql数据库查看一下

现在我们就有两个用户了。

我们关掉之前的终端。
我们开启3个终端。


我们先登录zhangsan用户

然后我在第3个终端，登录 linzeyu用户

我们在数据库中查看一下

状态都变成online了！

现在我们让linzeyu给zhangsan发送hello!
我们在linzeyu终端输入

{"msgid":5,"id":1,"from":"linzeyu","toid":2,"msg":"hello!"} 

接下来我们让zhangsan回复linzeyu
我们在zhangsan的终端输入

{"msgid":5,"id":2,"from":"zhangsan","toid":1,"msg":"ok"}