[集群聊天服务器]----(二)利用muduo网络库实现网络模块ChatServer

本文主要是介绍[集群聊天服务器]----(二)利用muduo网络库实现网络模块ChatServer，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

muduo 是由陈硕大佬个人开发的 TCP 网络库，有关于muduo网络库本人使用C++对muduo库的核心部分进行了重构，并且对每一个模块进行剖析介绍（详情见muduo网络库专栏），在使用muduo编写网络模块ChatServer之前，一定要对muduo网络库有一定的基本了解并且要对muduo网络库的基本使用有一定的了解。muduo 的线程模型为「one loop per thread + threadPool」模型。其中mainReactor只专注于监听新连接的到来，通过Acceptor接收新连接，然后将新连接派发到subReactor上进行连接的维护。同时多个Reactor可以并行运行在多核 CPU 中，增加服务效率。

启动服务端

muduo 网络库中使用 TcpServer 编写网络服务器，接收客户端发起连接，并且保存着用户注册的对于不同事件发生时要调用的回调函数。所以本项目中使用 TcpServer 来构造聊天服务器 ChatServer，并向 TcpServer 注册不同的回调函数来完成相关业务逻辑。server端的main.cpp中，我们首先进行了user状态的重置，并且进行了启动服务器代码编写：

#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)
{if (argc < 3){cerr << "command invalid! example: ./ChatServer 127.0.0.1 6000" << endl;exit(-1);}// 解析通过命令行参数传递的ip和portchar *ip = argv[1];uint16_t port = atoi(argv[2]);signal(SIGINT, resetHandler);EventLoop loop;InetAddress addr(ip, port);ChatServer server(&loop, addr, "ChatServer");server.start();loop.loop();return 0;
}

解析通过命令行参数传递的ip和port
通过信号处理函数，捕捉ctrl+c，即服务端异常退出状态，并对user的状态信息进行重置
创建mianloop，传入地址
ChatServer server(&loop, addr, "ChatServer");创建server对象
启动服务端loop，监听客户端状态

构造聊天服务器 ChatServer

在server端的main.cpp中，我们调用了ChatServer 创建一个server对象，在ChatServer中，我们主要注册了连接相关以及读写事件相关的回调函数，监听客户端的连接以及相关业务需求：

#ifndef CHATSERVER_H
#define CHATSERVER_H#include <muduo/net/TcpServer.h>
#include <muduo/net/EventLoop.h>
using namespace muduo;
using namespace muduo::net;class ChatServer
{
public: ChatServer(EventLoop *loop,const InetAddress& listenAddr,const string& nameArg);void start();
private://上报链接相关信息的回调函数void onConnection(const TcpConnectionPtr&);//上报读写事件相关信息的回调函数void onMessage(const TcpConnectionPtr&,Buffer *,Timestamp);TcpServer _server; //组合的muduo库，实现服务器功能的类对象EventLoop *_loop;   //指向事件循环对象的指针
};#endif

调用muduo网络库TcpServer模块，创建了 _server对象，并构建了事件循环对象的指针_loop；
构造函数中，调用了muduo网络库TcpServer模块，注册了连接以及读写事件回调函数，并设置线程数量

ChatServer::ChatServer(EventLoop *loop,const InetAddress &listenAddr,const string &nameArg): _server(loop, listenAddr, nameArg), _loop(loop)
{//注册连接回调_server.setConnectionCallback(std::bind(&ChatServer::onConnection,this,_1));_server.setMessageCallback(std::bind(&ChatServer::onMessage,this,_1,_2,_3));_server.setThreadNum(4);}

上报连接相关信息的回调函数

void ChatServer::onConnection(const TcpConnectionPtr& conn)
{//客户端断开连接if(!conn->connected()){ChatService::instance()->clientCloseException(conn);conn->shutdown();}
}

当 ChatServer 接收到连接相关事件时，会调用我们写ChatServer::onConnection函数。若是客户端连接断开的事件，会关闭连接。
上报读写事件相关信息的回调函数

void ChatServer::onMessage(const TcpConnectionPtr& conn,Buffer *buffer,Timestamp time)
{string buf = buffer->retrieveAllAsString();//数据的反序列化json js = json::parse(buf);auto msgHandler = ChatService::instance()->getHandler(js["msgid"].get<int>());//回调消息绑定好的事件处理器，来执行相应的业务处理msgHandler(conn,js,time);}

调用muduo网络库中buffer的retrieveAllAsString()函数，将读写事件buffer转换为string类型，并进行反序列化parse
将业务模块和网络模块解耦。当处理客户端读写事件时，我们有很多的类型。比如，好友一对一聊天，群组间消息，登录，注册信息，离线消息等等。针对不同的信息，使用不同的字段来标识，然后调用不同的处理函数，写出来的代码可能是：

if (msgid == Login) {LoginHandler();
} else if (msgid == Register) {RegisterHandler();
} else if (...)

这相当于在网络模块中写上了业务模块的代码，这样不仅会显得代码逻辑混乱，并且会暴露太多业务模块的代码。这时，我们就需要做一个解耦，网络模块正常收发数据，其他的业务逻辑写在业务模块内。因此，创建一个 ChatService 类提供不同的回调服务，使用function保存不同的回调函数，并且使用 Json 解析数据时得到数据类型，然后直接调用对应的函数。不同的消息类型，在public.hpp中进行了定义。

#ifndef PUBLIC_H
#define PUBLIC_Henum EnMsgType
{LOGIN_MSG = 1, //登录消息LOGIN_MSG_ACK, //登录响应消息LOGINOUT_MSG,  //退出相应消息REG_MSG,  //注册消息REG_MSG_ACK, //注册响应消息ONE_CHAT_MSG,//聊天消息ADD_FRIEND_MSG, //添加好友消息CREATE_GROUP_MSG, // 创建群组ADD_GROUP_MSG, // 加入群组GROUP_CHAT_MSG, // 群聊天
};#endif