提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
文章目录
前言
搭建一个网络服务器,在内部提供各个业务接口即可。
在业务处理函数中,每次请求过来找到对应的信道,通过信道句柄调用前边封装好的处理接口进行处理,最后返回响应即可。
管理的字段
服务器需要管理的字段,其中需要搭建一个tcp服务器。然后就是我们业务所需的句柄,一个是虚拟机管理句柄,消费者管理句柄,连接管理句柄和线程池句柄。
cpp
class Server
{
private:
using MessagePtr = std::shared_ptr<google::protobuf::Message>;
muduo::net::TcpServer _server;
muduo::net::EventLoop _baseloop;
ProtobufCodecPtr _codec; // 协议处理器 对收到的请求进行protobuf协议处理
ProtobufDispatcher _dispatcher; // 请求分发器
VirtualHost::ptr _host;
ConsumerManager::ptr _consumer_manager;
ConnectionManager::ptr _connection_manager;
ThreadPool::ptr _pool;
}
我们需要为服务器注册业务请求处理函数。muduo库是支持protobuf协议的处理的。
cpp
// 注册请求处理函数
_dispatcher.registerMessageCallback<openChannelRequest>(std::bind(&Server::OnOpenChannel,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<closeChannelRequest>(std::bind(&Server::OnClodeChannle,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<declareExchangeRequest>(std::bind(&Server::OnDeclareExchange,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<deleteExchangeRequest>(std::bind(&Server::OnDeleteExchange,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<declareQueueRequest>(std::bind(&Server::OnDeclareQueue,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<deleteQueueRequest>(std::bind(&Server::OnDeleteQueue,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<queueBindRequest>(std::bind(&Server::onQueueBind,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<queueUnBindRequest>(std::bind(&Server::onQueueUnBind,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<basicPublishRequest>(std::bind(&Server::onBasicPublish,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<basicAckRequest>(std::bind(&Server::onBasicAck,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<basicConsumeRequest>(std::bind(&Server::onBasicConsume,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
_dispatcher.registerMessageCallback<basicCancelRequest>(std::bind(&Server::onBasicCancel,this,std::placeholders::_1,std::placeholders::_2,std::placeholders::_3));
}
当新连接建立时,我们需调用连接句柄创建一个新连接。
而连接断开时,需要删除连接对象。
cpp
void onConnection(const muduo::net::TcpConnectionPtr &conn)
{
if (conn->connected() == true)
{
LOG_INFO << "新连接建立了";
_connection_manager->newConnection(_host,_consumer_manager,_codec,conn,_pool);
}
else
{
LOG_INFO << "连接断开了";
_connection_manager->delConnection(conn);
}
}
而其他注册的业务处理函数也比较简答,大体流程就是通过连接找到对应的连接管理对象,然后通过请求中的rid字段找道连接管理中的对应信道。调用信道中封装好的处理接口进行处理即可。
cpp
void OnDeclareExchange(const muduo::net::TcpConnectionPtr &conn,const declareExchangeRequestPtr &message,muduo::Timestamp){
Connection::ptr mconn = _connection_manager->getConnection(conn);
if(mconn.get() == nullptr)
{
DLOG("声明交换机时,没有找到连接对应的Connection对象!");
conn->shutdown();
return;
}
Channel::ptr cp = mconn->getChannel(message->cid());
if(cp.get() == nullptr)
{
DLOG("声明交换机时,没有找到信道!");
conn->shutdown();
return;
}
return cp->declareExchange(message);
}
这里有两个比较特殊,一个打开信道,一个是关闭信道
这两个操作是通过连接找到连接管理对象,然后调用连接管理对象提供的打开信道和关闭信道操作进行处理。
cpp
void OnOpenChannel(const muduo::net::TcpConnectionPtr &conn,const openChannelRequestPtr &message,muduo::Timestamp){
Connection::ptr mconn = _connection_manager->getConnection(conn);
if(mconn.get() == nullptr)
{
DLOG("打开信道时,没有找到连接对应的Connection对象!");
conn->shutdown();
return;
}
return mconn->openChannel(message);
}
//关闭信道
void OnClodeChannle(const muduo::net::TcpConnectionPtr &conn,const closeChannelRequestPtr &message,muduo::Timestamp){
Connection::ptr mconn = _connection_manager->getConnection(conn);
if(mconn.get() == nullptr)
{
DLOG("关闭信道时,没有找到连接对应的Connection对象!");
conn->shutdown();
return;
}
return mconn->closeChannel(message);
}
在构造函数的时候,由于队列消费者是以队列为单位管理的,所以我们要获取已经存在的队列,来进行队列消费者的初始化。
cpp
//消费者是按照队列为单元进行管理的,针对历史消息中的所有队列,需要初始化队列的消费者管理结构QueueConsumer
std::unordered_map<std::string, mq::MsgQueue::ptr> mqmp = _host->allQueue();
for(auto &mq : mqmp)
{
_consumer_manager->initQueueConsumer(mq.first);
}