集群聊天服务器：网络模块ChatServer

chatServer.hpp

#ifndef CHATSERVER_H
#define CHATSERVER_H

#include <muduo/net/TcpServer.h>
#include <muduo/net/EventLoop.h>
#include <string>
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

chatServer.cpp

#include "chatServer.hpp"
#include "chatService.hpp"
#include "json.hpp"
#include <functional>
using namespace std;
using namespace placeholders;
using json = nlohmann::json;

// 初始化聊天服务器对象
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::start()
{
    
    _server.start();
}

// 上报链接相关信息的回调函数
void ChatServer::onConnection(const TcpConnectionPtr& conn)
{
    
    // 客户端断开链接
    if (!conn->connected())
    {
    
        // 处理异常断开？
        ChatService::instance()->clientCloseException(conn);
        conn->shutdown();
    }
}

// 上报读写事件相关信息的回调函数
void ChatServer::onMessage(const TcpConnectionPtr& conn,
                Buffer* buffer,
                Timestamp time)
{
    
    string buf = buffer->retrieveAllAsString();
    // 数据的反序列化
    json js = json::parse(buf);

    // 达到的目的：完全解耦网络模块的代码和业务模块的代码
    // 通过js["msgid"]（一个ID绑定一个模块）获取-》业务handler-》 conn js time
    auto msgHandler = ChatService::instance()->getHandler(js["msgid"].get<int>());
    // 回调消息绑定好的事件处理器，来执行相应的业务处理
    msgHandler(conn, js, time);
}