重写muduo网络库：终章TcpServer

2023-10-31 08:30
文章标签 重写 muduo 网络 tcpserver 终章

本文主要是介绍重写muduo网络库：终章TcpServer，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

终章TcpServer

回顾muduo网络库的简单使用：

#include <mymuduo/TcpServer.h>
#include <mymuduo/Logger.h>
#include <functional>
#include <string>class EchoServer
{
public:EchoServer(EventLoop* loop, const InetAddress &addr,const std::string &name):loop_(loop),server_(loop,addr,name){//注册回调函数server_.setMessageCallback(std::bind(&EchoServer::onMessage, this, std::placeholders::_1, std::placeholders::_2, std::placeholders::_3));server_.setConnectionCallback(std::bind(&EchoServer::onConnection, this, std::placeholders::_1));//设置线程数量server_.setThreadNum(3);}void start(){server_.start();}
private:void onConnection(const TcpConnectionPtr& conn){if(conn->connected()){LOG_INFO("Connect UP : %s", conn->peerAddress().toIpPort().c_str());}else{LOG_INFO("Connect DOWN : %s", conn->peerAddress().toIpPort().c_str());}}void onMessage(const TcpConnectionPtr& conn,Buffer* buf,Timestamp time){std::string msg = buf->retrieveAllAsString();conn->send(msg);conn->shutdown();}EventLoop* loop_;TcpServer server_;
};int main()
{EventLoop loop;InetAddress addr(8000);EchoServer server(&loop, addr, "EchoServer-01");server.start();loop.loop();return 0;
}

用户先定义一个EventLoop事件循环和Socket地址，用他们构造一个TcpServer对象，然后就是设置回调函数和线程数量，接着就是启动线程池，开启循环监听。

而在构造TcpServer对象的时候做了什么呢？
在这里插入图片描述
首先就是接收了一个mainLoop，用接收的mainLoop和Socket地址创建了一个Acceptor，然后就是给Acceptor绑定了一个回调，这个回调就是在accept的处理流程，得到一个connfd时，就会触发这个回调。
在这里插入图片描述
这个回调会从线程池中获取一个ioLoop，对这个新连接的connfd打包成TcpConnection；设置用户设置的回调，回调的传递流程时从用户=》TcpServer=》TcpConnection=》Channel=》Poller；然后就runInLoop唤醒ioLoop所在的线程起来处理。

重写TcpServer.h：

#pragma once#include "noncopyable.h"
#include "EventLoop.h"
#include "Acceptor.h"
#include "InetAddress.h"
#include "Callbacks.h"
#include "EventLoopThreadPool.h"
#include "Buffer.h"
#include "TcpConnection.h"#include <string>
#include <functional>
#include <atomic>
#include <unordered_map>class TcpServer : noncopyable
{
public:using ThreadInitCallback = std::function<void(EventLoop*)>;enum Option{kNoReusePort,kReusePort,};TcpServer(EventLoop* loop, const InetAddress& listenAddr, const std::string& nameArg, Option option = kNoReusePort);~TcpServer();void setThreadInitCallback(const ThreadInitCallback &cb) { threadInitCallback_ = cb; }void setConnectionCallback(const ConnectionCallback &cb) { connectionCallback_ = cb; }void setMessageCallback(const MessageCallback &cb) { messageCallback_ = cb; }void setWriteCompleteCallback(const WriteCompleteCallback &cb) { writeCompleteCallback_ = cb; }//设置subloop线程数量void setThreadNum(int numThreads);//开启服务监听void start();private:void newConnection(int sockfd, const InetAddress &peerAddr);void removeConnection(const TcpConnectionPtr& conn);void removeConnectionInLoop(const TcpConnectionPtr& conn);using ConnectionMap = std::unordered_map<std::string, TcpConnectionPtr>;EventLoop *loop_; //用户定义的loopconst std::string ipPort_;const std::string name_;std::unique_ptr<Acceptor> acceptor_;    //运行在mainloop，任务就是监听新连接事件std::shared_ptr<EventLoopThreadPool> threadPool_;  //one loop per threadConnectionCallback connectionCallback_; //新连接回调MessageCallback messageCallback_;   //有读写消息时的回调WriteCompleteCallback writeCompleteCallback_;   //消息发送完成以后的回调ThreadInitCallback threadInitCallback_; //loop线程初始化的回调std::atomic_int started_;int nextConnId_;ConnectionMap connections_;  //保存所有的连接
};

重写TcpServer.cc：

#include "TcpServer.h"
#include "Logger.h"
#include "TcpConnection.h"#include <strings.h>EventLoop* CheckLoopNotNull(EventLoop *loop)
{if(loop == nullptr){LOG_FATAL("%s:%s:%d mainloop is null \n", __FILE__, __FUNCTION__, __LINE__);}return loop;
}TcpServer::TcpServer(EventLoop* loop, const InetAddress& listenAddr, const std::string& nameArg, Option option):loop_(CheckLoopNotNull(loop)),ipPort_(listenAddr.toIpPort()),name_(nameArg),acceptor_(new Acceptor(loop, listenAddr, option = kReusePort)),threadPool_(new EventLoopThreadPool(loop, name_)),connectionCallback_(),messageCallback_(),nextConnId_(1),started_(0)
{//当有新用户连接时，会执行TcpServer::newConnection回调acceptor_->setNewConnectionCallback(std::bind(&TcpServer::newConnection, this,std::placeholders::_1, std::placeholders::_2));
}//设置subloop线程数量
void TcpServer::setThreadNum(int numThreads)
{threadPool_->setThreadNum(numThreads);
}//开启服务监听
void TcpServer::start()
{if(started_++ == 0) //防止一个TcpServer对象被start多次{threadPool_->start(threadInitCallback_);loop_->runInLoop(std::bind(&Acceptor::listen, acceptor_.get()));}
}//有一个新的客户端的连接，acceptor会执行这个回调操作
void TcpServer::newConnection(int sockfd, const InetAddress& peerAddr)
{//轮询算法，选择一个subLoop，来管理channelEventLoop* ioLoop = threadPool_->getNextLoop();char buf[64] = {0};snprintf(buf, sizeof buf, "-%s#%d", ipPort_.c_str(), nextConnId_);++nextConnId_;std::string connName = name_ + buf;LOG_INFO("TcpServer::newConnection [%s] - new connection [%s] from %s \n",name_.c_str(), connName.c_str(), peerAddr.toIpPort().c_str());//通过sockfd获取其绑定的本机的IP地址和端口信息sockaddr_in local;::bzero(&local, sizeof local);socklen_t addrLen = sizeof local;if(::getsockname(sockfd, (sockaddr*)&local, &addrLen) < 0){LOG_ERROR("sockets::getLocalAddr");}InetAddress localAddr(local);//根据连接成功的sockfd，创建TcpConnection连接对象TcpConnectionPtr conn(new TcpConnection(ioLoop, connName, sockfd, localAddr, peerAddr));connections_[connName] = conn;//下面回调都是用户设置给Tcpserver =》TcpConnection = 》 channel =》 poller =》notify channel调用回调conn->setConnectionCallback(connectionCallback_);conn->setMessageCallback(messageCallback_);conn->setWriteCompleteCallback(writeCompleteCallback_);//设置了如何关闭连接的回调conn->setCloseCallback(std::bind(&TcpServer::removeConnection, this, std::placeholders::_1));ioLoop->runInLoop(std::bind(&TcpConnection::connectEstablished, conn));
}void TcpServer::removeConnection(const TcpConnectionPtr& conn)
{loop_->runInLoop(std::bind(&TcpServer::removeConnectionInLoop, this, conn));
}void TcpServer::removeConnectionInLoop(const TcpConnectionPtr& conn)
{LOG_INFO("TcpServer::removeConnectionInLoop [%s] - connection %s\n",name_.c_str(), conn->name().c_str());connections_.erase(conn->name());EventLoop* ioLoop = conn->getLoop();    ioLoop->queueInLoop(std::bind(&TcpConnection::connectDestroyed, conn));
}TcpServer::~TcpServer()
{for(auto &item : connections_){TcpConnectionPtr conn(item.second); //这个局部的shared_ptr智能指针对象，出右括号，可以自动释放newitem.second.reset();//销毁连接conn->getLoop()->runInLoop(std::bind(&TcpConnection::connectDestroyed, conn));}
}

这篇关于重写muduo网络库：终章TcpServer的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/314083。

相关文章

Linux 网络编程 --- 应用层

Linux 网络编程 --- 应用层

一、自定义协议和序列化反序列化 代码： 序列化反序列化实现网络版本计算器 二、HTTP协议 1、谈两个简单的预备知识 https://www.baidu.com/ --- 域名 --- 域名解析 --- IP地址 http的端口号为80端口，https的端口号为443 url为统一资源定位符。CSDNhttps://mp.csdn.net/mp_blog/creation/editor
阅读更多...
ASIO网络调试助手之一：简介

ASIO网络调试助手之一：简介

多年前，写过几篇《Boost.Asio C++网络编程》的学习文章，一直没机会实践。最近项目中用到了Asio，于是抽空写了个网络调试助手。 开发环境： Win10 Qt5.12.6 + Asio(standalone) + spdlog 支持协议： UDP + TCP Client + TCP Server 独立的Asio（http://www.think-async.com）只包含了头文件，不依
阅读更多...
poj 3181 网络流，建图。

poj 3181 网络流，建图。

题意： 农夫约翰为他的牛准备了F种食物和D种饮料。 每头牛都有各自喜欢的食物和饮料，而每种食物和饮料都只能分配给一头牛。 问最多能有多少头牛可以同时得到喜欢的食物和饮料。 解析： 由于要同时得到喜欢的食物和饮料，所以网络流建图的时候要把牛拆点了。 如下建图： s -> 食物 -> 牛1 -> 牛2 -> 饮料 -> t 所以分配一下点： s  =  0， 牛1= 1~
阅读更多...
poj 3068 有流量限制的最小费用网络流

poj 3068 有流量限制的最小费用网络流

题意： m条有向边连接了n个仓库，每条边都有一定费用。 将两种危险品从0运到n-1，除了起点和终点外，危险品不能放在一起，也不能走相同的路径。 求最小的费用是多少。 解析： 抽象出一个源点s一个汇点t，源点与0相连，费用为0，容量为2。 汇点与n - 1相连，费用为0，容量为2。 每条边之间也相连，费用为每条边的费用，容量为1。 建图完毕之后，求一条流量为2的最小费用流就行了
阅读更多...
poj 2112 网络流+二分

poj 2112 网络流+二分

题意： k台挤奶机，c头牛，每台挤奶机可以挤m头牛。 现在给出每只牛到挤奶机的距离矩阵，求最小化牛的最大路程。 解析： 最大值最小化，最小值最大化，用二分来做。 先求出两点之间的最短距离。 然后二分匹配牛到挤奶机的最大路程，匹配中的判断是在这个最大路程下，是否牛的数量达到c只。 如何求牛的数量呢，用网络流来做。 从源点到牛引一条容量为1的边，然后挤奶机到汇点引一条容量为m的边
阅读更多...
配置InfiniBand (IB) 和 RDMA over Converged Ethernet (RoCE) 网络

配置InfiniBand (IB) 和 RDMA over Converged Ethernet (RoCE) 网络

配置InfiniBand (IB) 和 RDMA over Converged Ethernet (RoCE) 网络 服务器端配置 在服务器端，你需要确保安装了必要的驱动程序和软件包，并且正确配置了网络接口。 安装 OFED 首先，安装 Open Fabrics Enterprise Distribution (OFED)，它包含了 InfiniBand 所需的驱动程序和库。 sudo
阅读更多...
【机器学习】高斯网络的基本概念和应用领域

【机器学习】高斯网络的基本概念和应用领域

引言 高斯网络（Gaussian Network）通常指的是一个概率图模型，其中所有的随机变量（或节点）都遵循高斯分布 文章目录 引言一、高斯网络（Gaussian Network）1.1 高斯过程（Gaussian Process）1.2 高斯混合模型（Gaussian Mixture Model）1.3 应用1.4 总结 二、高斯网络的应用2.1 机器学习2.2 统计学2.3
阅读更多...
网络学习-eNSP配置NAT

网络学习-eNSP配置NAT

NAT实现内网和外网互通 #给路由器接口设置IP地址模拟实验环境<Huawei>system-viewEnter system view, return user view with Ctrl+Z.[Huawei]undo info-center enableInfo: Information center is disabled.[Huawei]interface gigabit
阅读更多...
Golang 网络爬虫框架gocolly/colly（五）

Golang 网络爬虫框架gocolly/colly（五）

gcocolly+goquery可以非常好地抓取HTML页面中的数据，但碰到页面是由Javascript动态生成时，用goquery就显得捉襟见肘了。解决方法有很多种： 一，最笨拙但有效的方法是字符串处理，go语言string底层对应字节数组，复制任何长度的字符串的开销都很低廉，搜索性能比较高； 二，利用正则表达式，要提取的数据往往有明显的特征，所以正则表达式写起来比较简单，不必非常严谨； 三，使
阅读更多...
Golang网络爬虫框架gocolly/colly（四）

Golang网络爬虫框架gocolly/colly（四）

爬虫靠演技，表演得越像浏览器，抓取数据越容易，这是我多年爬虫经验的感悟。回顾下个人的爬虫经历，共分三个阶段：第一阶段，09年左右开始接触爬虫，那时由于项目需要，要访问各大国际社交网站，Facebook，myspace，filcker，youtube等等，国际上叫得上名字的社交网站都爬过，大部分网站提供restful api，有些功能没有api，就只能用http抓包工具分析协议，自己爬；国内的优酷、
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2