397-服务器异常退出处理代码和测试

最新推荐文章于 2022-10-22 16:51:40 发布

林林林ZEYU

最新推荐文章于 2022-10-22 16:51:40 发布

阅读量474

点赞数 1

分类专栏： C++实现校园集群聊天服务器文章标签：数据库 mysql linux

本文链接：https://blog.csdn.net/LINZEYU666/article/details/119485027

版权

C++实现校园集群聊天服务器专栏收录该内容

30 篇文章 39 订阅

订阅专栏

在之前，我完成看客户端异常退出的处理代码，客户端异常退出是因为它没有发出json字符串，是通过连接异常断开进行用户的connection的删除和用户表状态的更改。但是服务器异常退出了，user表的用户的状态还是online，这个问题还没有解决。因为我们Ctrl C强制退出服务器的时候，服务器根本没有机会去数据库里更改用户的状态信息。

服务器异常退出处理代码

我们完善src下的server下的main.cpp

#include "chatserver.hpp"
#include "chatservice.hpp"
#include <iostream>
#include <signal.h>
using namespace std;

//处理服务器ctrl+c结束后，进行重置user的状态信息
void resetHandler(int)
{
    ChatService::instance()->reset();//调用重置 
    exit(0);
}

int main(int argc, char **argv)
{
 
    signal(SIGINT, resetHandler);

    EventLoop loop;
    InetAddress addr("127.0.0.1", 6000);
    ChatServer server(&loop, addr, "ChatServer");

    server.start();
    loop.loop();

    return 0;
}

我们完善一下include下的server下的chatservice.hpp
新增代码

void clientCloseException(const TcpConnectionPtr &conn);
//服务器异常，业务重置方法

完整代码

#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 "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 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;   

};

#endif

我们完善一下src下的server下的chatservice.cpp
新增代码

//服务器异常，业务重置方法
void ChatService::reset()
{
    //把online状态的用户，设置成offline
    _userModel.resetState();
}

完整代码

#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)});   

}

//服务器异常，业务重置方法
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);
            }
            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());
}

我们完善usermodel.hpp

#ifndef USERMODEL_H
#define USERMODEL_H

#include "user.hpp"

//User表的数据操作类
class UserModel {
public:
    //User表的增加方法
    bool insert(User &user);

    //根据用户号码查询用户信息
    User query(int id);

    //更新用户的状态信息
    bool updateState(User user);

    //重置用户的状态信息
    void resetState();
};

#endif

我们完善usermodel.cpp
新增代码

//重置用户的状态信息
void UserModel::resetState()
{
    //1.组装sql语句
    char sql[1024] = "update user set state = 'offline' where state = 'online'";

    MySQL mysql;//定义mysql对象 
    if (mysql.connect())//连接成功 
    {
        mysql.update(sql);//更新 
    }
}

完整代码

#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;
}

//重置用户的状态信息
void UserModel::resetState()
{
    //1.组装sql语句
    char sql[1024] = "update user set state = 'offline' where state = 'online'";

    MySQL mysql;//定义mysql对象 
    if (mysql.connect())//连接成功 
    {
        mysql.update(sql);//更新 
    }
}

编译成功
在这里插入图片描述

测试

我们先查看mysql数据库中的user表的状态信息
在这里插入图片描述
现在我们打开2个终端，启动

我们做一个登录操作，登录linzeyu用户

登录成功。我们在mysql数据库查看一下用户linzeyu的状态
在这里插入图片描述
然后我们现在在服务端执行ctrl c强制退出

现在我们再去mysql数据库查看一下用户linzeyu的状态
在这里插入图片描述
测试成功！！！

林林林ZEYU

关注

1
点赞
踩
0

收藏

觉得还不错? 一键收藏
打赏
0
评论
397-服务器异常退出处理代码和测试

在之前，我完成看客户端异常退出的处理代码，客户端异常退出是因为它没有发出json字符串，是通过连接异常断开进行用户的connection的删除和用户表状态的更改。但是服务器异常退出了，user表的用户的状态还是online，这个问题还没有解决。因为我们Ctrl C强制退出服务器的时候，服务器根本没有机会去数据库里更改用户的状态信息。服务器异常退出处理代码我们完善src下的server下的main.cpp#include "chatserver.hpp"#include "chatservice.hp
复制链接

扫一扫