项目(百万并发网络通信架构)12.2---内存管理之（为服务端添加智能指针）

最新推荐文章于 2024-08-05 00:23:05 发布
最新推荐文章于 2024-08-05 00:23:05 发布
阅读量1.3k 收藏 5
点赞数 1
分类专栏： 项目(百万并发网络通信架构)
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_41453285/article/details/105975568
版权

一、C++智能指针概述

二、为服务端添加智能指针

  • 在前一篇文章中我们为服务端添加了内存池，重写了new和delete运算符。在本文中，我们将服务端程序中的普通类型指针大部分更改为智能指针shared_ptr
  • shared_ptr的内部也会使用new和delete，因此shared_ptr的内部也会使用到我们的内存池，因为new和delete被我们重写了

三、代码如下

  • 在之前文章中的所有代码中，我们只需要更改EasyTcpServer.hpp，其余不需要更改

EasyTcpServer.hpp代码如下

  • 代码如下，不能保证完全没有错误
#ifndef _EasyTcpClient_hpp_
#define _EasyTcpClient_hpp_

#ifdef _WIN32
#define FD_SETSIZE 10240
#define WIN32_LEAN_AND_MEAN
#define _WINSOCK_DEPRECATED_NO_WARNINGS //for inet_pton()
#define _CRT_SECURE_NO_WARNINGS
#include <windows.h>
#include <WinSock2.h>
#pragma comment(lib, "ws2_32.lib")
#else
#include <unistd.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/select.h>
//在Unix下没有这些宏，为了兼容，自己定义
#define SOCKET int
#define INVALID_SOCKET  (SOCKET)(~0)
#define SOCKET_ERROR            (-1)
#endif


#ifndef RECV_BUFF_SIZE
#define RECV_BUFF_SIZE 10240*5        //接收缓冲区的大小
#endif // !RECV_BUFF_SIZE

#ifndef SEND_BUFF_SIZE
#define SEND_BUFF_SIZE RECV_BUFF_SIZE //发送缓冲区的大小
#endif // !SEND_BUFF_SIZE

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <vector>
#include <mutex>
#include <atomic>
#include <functional> 
#include <map>
#include <memory>
#include "MessageHeader.hpp"
#include "CELLTimestamp.hpp"
#include "CELLTask.hpp"
//using namespace std;

class CellServer;

//客户端数据类型，用来封装一个客户端
class ClientSocket
{
public:
	ClientSocket(SOCKET sockfd = INVALID_SOCKET) :_sock(sockfd), _lastRecvPos(0), _lastSendPos(0) {
		memset(_recvMsgBuff, 0, sizeof(_recvMsgBuff));
		memset(_sendMsgBuff, 0, sizeof(_sendMsgBuff));
	}

	SOCKET sockfd() { return _sock; }
	char   *recvMsgBuff() { return _recvMsgBuff; }
	int    getRecvLastPos() { return _lastRecvPos; }
	void   setRecvLastPos(int pos) { _lastRecvPos = pos; }

	char   *sendMsgBuff() { return _sendMsgBuff; }
	int    getSendLastPos() { return _lastSendPos; }
	void   setSendLastPos(int pos) { _lastSendPos = pos; }

	int    SendData(std::shared_ptr<DataHeader>& header);
private:
	SOCKET _sock;                         //客户端socket

	char   _recvMsgBuff[RECV_BUFF_SIZE];  //消息接收缓冲区
	int    _lastRecvPos;                  //消息接收缓冲区的数据尾部位置

	char   _sendMsgBuff[SEND_BUFF_SIZE];  //消息发送缓冲区
	int    _lastSendPos;                  //消息发送缓冲区的数据尾部位置
};

//网络事件接口
class INetEvent {
public:
	virtual void OnClientJoin(std::shared_ptr<ClientSocket>& pClient) = 0;     //客户端加入事件
	virtual void OnClientLeave(std::shared_ptr<ClientSocket>& pClient) = 0;    //客户端离开事件
	virtual void OnNetMsg(CellServer* pCellServer, std::shared_ptr<ClientSocket>& pClient, DataHeader* header) = 0; //接收到客户端消息事件
	virtual void OnNetRecv(std::shared_ptr<ClientSocket>& pClient) = 0;        //recv函数执行事件
};

//发送消息的任务
class CellSendMsg2ClientTask :public CellTask
{
public:
	CellSendMsg2ClientTask(std::shared_ptr<ClientSocket>& pClient, std::shared_ptr<DataHeader>& header)
		:_pClient(pClient), _pHeader(header) {}

	void doTask()
	{
		_pClient->SendData(_pHeader);
	}
private:
	std::shared_ptr<ClientSocket> _pClient; //发送给哪个客户端
	std::shared_ptr<DataHeader>   _pHeader; //要发送的数据的头指针
};

//处理客户的从服务器类
class CellServer
{
public:
	CellServer(SOCKET sock = INVALID_SOCKET) :_sock(sock), maxSock(_sock), _pthread(nullptr), _pNetEvent(nullptr), _clients_change(false) {
		memset(&_fdRead_bak, 0, sizeof(_fdRead_bak));
	}
	~CellServer() { CloseSocket(); }
public:
	bool   isRun() { return _sock != INVALID_SOCKET; }
	void   CloseSocket();
public:
	size_t getClientCount() { return _clients.size() + _clientsBuff.size(); } //返回当前客户端的数量
	void   setEventObj(INetEvent* event) { _pNetEvent = event; }              //设置事件对象，此处绑定的是EasyTcpServer
public:
	bool   Onrun();
	void   AddClient(std::shared_ptr<ClientSocket>& pClient)  //讲客户端加入到客户端连接缓冲队列中
	{
		//自解锁
		std::lock_guard<std::mutex> lock(_mutex);
		//_mutex.lock(); 当然也可以使用互斥锁
		_clientsBuff.push_back(pClient);
		//_mutex.unlock();
	}
	int    RecvData(std::shared_ptr<ClientSocket>& pClient);     //接收数据
	void   OnNetMessage(std::shared_ptr<ClientSocket>& pClient, DataHeader* header);//处理网络消息
	void   Start() {
		//启动当前服务线程
		//创建一个线程，线程执行函数为Onrun()，其实可以不传递this，但是为了更安全，可以传递this给Onrun()
		_pthread = new std::thread(std::mem_fn(&CellServer::Onrun), this);

		//启动任务管理
		_taskServer.Start();
	}
public:
	void addSendTask(std::shared_ptr<ClientSocket>& pClient, std::shared_ptr<DataHeader>& header)
	{
		auto task = std::make_shared<CellSendMsg2ClientTask>(pClient, header);
		_taskServer.addTask((CellTaskPtr&)task);
	}
private:
	SOCKET                     _sock;         //服务端的套接字
	std::map<SOCKET, std::shared_ptr<ClientSocket>> _clients; //真正存储客户端
	std::vector<std::shared_ptr<ClientSocket>> _clientsBuff;  //存储客户端连接缓冲队列，之后会被加入到_clients中去
	SOCKET                     maxSock;       //当前最大的文件描述符值，select的参数1要使用
											  //char _recvBuff[RECV_BUFF_SIZE];           //接收缓冲区
	std::mutex                 _mutex;        //互斥锁
	std::thread*               _pthread;      //当前子服务端执行的线程
	INetEvent*                 _pNetEvent;
	fd_set                     _fdRead_bak;   //用来保存当前的fd_set
	bool                       _clients_change;//当前是否有新客户端加入进来
private:
	CellTaskServer _taskServer;
};

//服务器主类
class EasyTcpServer :public INetEvent
{
public:
	EasyTcpServer() :_sock(INVALID_SOCKET), _msgCount(0), _recvCount(0), _clientCount(0) {}
	virtual ~EasyTcpServer() { CloseSocket(); }
public:
	void InitSocket();  //初始化socket
	int  Bind(const char* ip, unsigned short port);    //绑定端口号
	int  Listen(int n); //监听端口号
	SOCKET Accept();    //接收客户端连接
	void addClientToCellServer(std::shared_ptr<ClientSocket>& pClient);//将新客户加入到CellServer的客户端连接缓冲队列中
	void Start(int nCellServer);                      //创建从服务器，并运行所有的从服务器。(参数为从服务器的数量)
	void CloseSocket(); //关闭socket
	bool isRun() { return _sock != INVALID_SOCKET; }  //判断当前服务端是否在运行
	bool Onrun();       //处理网络消息

	void time4msg();    //每1秒统计一次收到的数据包的数量
public:
	//客户端加入事件(这个是线程安全的，因为其只会被主服务器(自己)调用)
	virtual void OnClientJoin(std::shared_ptr<ClientSocket>& pClient)override { _clientCount++; }
	
	//客户端离开事件，这个里面做的事情比较简单，只是将当前客户端的数量--（如果从服务器不止一个，那么此函数不是线程安全的，因为这个函数会被多个从服务器调用的）
	virtual void OnClientLeave(std::shared_ptr<ClientSocket>& pClient)override { _clientCount--; }
	
	//接收到客户端消息事件，将数据包的数量增加（如果从服务器不止一个，那么此函数不是线程安全的，因为这个函数会被多个从服务器调用的）
	//参数1，代表哪一个CellServer来处理这个消息的
	virtual void OnNetMsg(CellServer* pCellServer, std::shared_ptr<ClientSocket>& pClient, DataHeader* header)override { _msgCount++; }
	
	//recv事件
	virtual void OnNetRecv(std::shared_ptr<ClientSocket>& pClient)override { _recvCount++; }
private:
	SOCKET                     _sock;        //服务端套接字
	std::vector<std::shared_ptr<CellServer>>   _cellServers; //存放从服务端对象
	CELLTimestamp              _tTime;       //计时器
	std::atomic_int            _clientCount; //客户端的数量(这里采用一个原子操作，没什么特殊原因，使用玩玩罢了,下同)
	std::atomic_int            _msgCount;    //表示服务端接收到客户端数据包的数量
	std::atomic_int            _recvCount;   //recv()函数执行的次数
};

int ClientSocket::SendData(std::shared_ptr<DataHeader>& header) {
	int ret = SOCKET_ERROR;

	//要发送的数据的长度
	int nSendLen = header->dataLength;
	const char* pSendData = (const char*)header.get();

	//在下面第一个if之后，如果nSendLen仍然大于SEND_BUFF_SIZE，那么就需要继续执行if，继续发送数据
	while (true)
	{
		//如果当前"要发送的数据的长度+缓冲区数据结尾位置"之和总的缓冲区大小，说明缓冲区满了，那么将整个缓冲区都发送出去
		if (_lastSendPos + nSendLen >= SEND_BUFF_SIZE)
		{
			//计算缓冲区还剩多少空间
			int nCopyLen = SEND_BUFF_SIZE - _lastSendPos;
			//然后将nCopyLen长度的pSendData数据拷贝到缓冲区中
			memcpy(_sendMsgBuff + _lastSendPos, pSendData, nCopyLen);
			//计算剩余数据位置
			pSendData += nCopyLen;
			//计算剩余数据长度
			nSendLen -= nCopyLen;
			ret = send(_sock, _sendMsgBuff, SEND_BUFF_SIZE, 0);
			//缓冲区数据尾部置为0
			_lastSendPos = 0;

			if (ret = SOCKET_ERROR)
				return -1;
		}
		//如果发送缓冲区还没满，那么将这条消息放到缓冲区中，而不直接发送
		else
		{
			memcpy(_sendMsgBuff + _lastSendPos, pSendData, nSendLen);
			//更新缓冲区数据尾部
			_lastSendPos += nSendLen;
			//直接退出，不进行while
			break;
		}
	}

	return ret;
}

void EasyTcpServer::InitSocket()
{
#ifdef _WIN32
	WORD ver = MAKEWORD(2, 2);
	WSADATA dat;
	WSAStartup(ver, &dat);
#endif

	//建立socket
	_sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (INVALID_SOCKET == _sock) {
		std::cout << "Server：创建socket成功" << std::endl;
	}
	else {
		std::cout << "Server：创建socket成功" << std::endl;
	}
}

int EasyTcpServer::Bind(const char* ip, unsigned short port)
{
	if (!isRun())
		InitSocket();

	//初始化服务端地址
	struct sockaddr_in _sin = {};
#ifdef _WIN32
	if (ip)
		_sin.sin_addr.S_un.S_addr = inet_addr(ip);
	else
		_sin.sin_addr.S_un.S_addr = INADDR_ANY;
#else
	if (ip)
		_sin.sin_addr.s_addr = inet_addr(ip);
	else
		_sin.sin_addr.s_addr = INADDR_ANY;
#endif
	_sin.sin_family = AF_INET;
	_sin.sin_port = htons(port);

	//绑定服务端地址
	int ret = ::bind(_sock, (struct sockaddr*)&_sin, sizeof(_sin));
	if (SOCKET_ERROR == ret) {
		if (ip)
			std::cout << "Server：绑定地址(" << ip << "," << port << ")失败!" << std::endl;
		else
			std::cout << "Server：绑定地址(INADDR_ANY," << port << ")失败!" << std::endl;
	}
	else {
		if (ip)
			std::cout << "Server：绑定地址(" << ip << "," << port << ")成功!" << std::endl;
		else
			std::cout << "Server：绑定地址(INADDR_ANY," << port << ")成功!" << std::endl;
	}
	return ret;
}

void EasyTcpServer::CloseSocket()
{
	if (_sock != INVALID_SOCKET)
	{
#ifdef _WIN32
		closesocket(_sock);
		WSACleanup();
#else
		close(_sock);
#endif
		_sock = INVALID_SOCKET;
	}
}

int EasyTcpServer::Listen(int n)
{
	//监听网络端口
	int ret = listen(_sock, n);
	if (SOCKET_ERROR == ret)
		std::cout << "Server：监听网络端口失败!" << std::endl;
	else
		std::cout << "Server：监听网络端口成功!" << std::endl;
	return ret;
}

SOCKET EasyTcpServer::Accept()
{
	//用来保存客户端地址
	struct sockaddr_in _clientAddr = {};
	int nAddrLen = sizeof(_clientAddr);
	SOCKET _cSock = INVALID_SOCKET;

	//接收客户端连接
#ifdef _WIN32
	_cSock = accept(_sock, (struct sockaddr*)&_clientAddr, &nAddrLen);
#else
	_cSock = accept(_sock, (struct sockaddr*)&_clientAddr, (socklen_t*)&nAddrLen);
#endif
	if (INVALID_SOCKET == _cSock) {
		std::cout << "Server：接收到无效客户端!" << std::endl;
	}
	else {
		//通知其他已存在的所有客户端，有新的客户端加入
		//NewUserJoin newUserInfo(static_cast<int>(_cSock));
		//SendDataToAll(&newUserInfo);

		//将新连接的客户端加入到从服务器的客户端缓冲队列中
		std::shared_ptr<ClientSocket> newClient = std::make_shared<ClientSocket>(_cSock);
		addClientToCellServer(newClient);
		OnClientJoin(newClient); //相应客户端加入事件，其函数会将客户端的数量++

		//std::cout << "Server：接受到新的客户端(" << _clients.size() << ")连接,IP=" << inet_ntoa(_clientAddr.sin_addr)
		//	<< ",Socket=" << static_cast<int>(_cSock) << std::endl;
	}
	return _cSock;
}

void EasyTcpServer::addClientToCellServer(std::shared_ptr<ClientSocket>& pClient)
{
	//在_cellServers中寻找，哪一个CellServer其处理的客户数量最少，那么就将新客户加入到这个CellServer对象中去
	auto pMinServer = _cellServers[0];
	for (auto pCellServer : _cellServers)
	{
		if (pMinServer->getClientCount() > pCellServer->getClientCount())
		{
			pMinServer = pCellServer;
		}
	}
	pMinServer->AddClient(pClient);
}

void EasyTcpServer::Start(int nCellServer)
{
	for (int i = 0; i < nCellServer; ++i)
	{
		auto ser = std::make_shared<CellServer>(_sock);
		_cellServers.push_back(ser);
		ser->setEventObj(this);
		ser->Start();
	}
}

bool EasyTcpServer::Onrun()
{
	if (isRun())
	{
		time4msg(); //统计当前接收到的数据包的数量

		fd_set fdRead;
		//fd_set fdWrite;
		//fd_set fdExp;
		FD_ZERO(&fdRead);
		//FD_ZERO(&fdWrite);
		//FD_ZERO(&fdExp);
		FD_SET(_sock, &fdRead);
		//FD_SET(_sock, &fdWrite);
		//FD_SET(_sock, &fdExp);

		struct timeval t = { 0,0 };
		int ret = select(_sock + 1, &fdRead, nullptr, nullptr, &t);
		if (ret < 0)
		{
			std::cout << "Server：select出错！" << std::endl;
			//select出错，那么就不能再继续运行select，出错之后，调用CloseSocket()，
			//关闭所有服务端、及客所有户端套接字，那么isRun()就会返回false，从而终止server.cpp程序运行
			CloseSocket();
			return false;
		}
		if (FD_ISSET(_sock, &fdRead))//如果一个客户端连接进来，那么服务端的socket就会变为可读的，此时我们使用accept来接收这个客户端
		{
			FD_CLR(_sock, &fdRead);
			Accept();
			return true;
		}
		return true;
	}
	return false;
}

void EasyTcpServer::time4msg()
{
	auto t = _tTime.getElapsedSecond();
	if (t >= 1.0)
	{
		//msgCount,_recvCount为什么要除以t：
		//因为我们要每1秒钟打印一次接收到的数据包，如果这个函数调用的时候时间差大于1秒，那么可以将recvCount/t，获得比较平均的数据包数量/recv执行次数
		printf("time<%lf>,thread numer<%d>,client number<%d>,msgCount<%d>,recvCount<%d>\n",
			t, _cellServers.size(), static_cast<int>(_clientCount), static_cast<int>(_msgCount / t), static_cast<int>(_recvCount / t));
		_recvCount = 0;
		_msgCount = 0;
		_tTime.update();
	}
}

void CellServer::CloseSocket()
{
	if (_sock != INVALID_SOCKET)
	{
#ifdef _WIN32
		//将所有的客户端套接字关闭
		for (auto iter : _clients)
		{
			closesocket(iter.second->sockfd());
		}
		//关闭服务端套接字
		closesocket(_sock);

		//因为这是从服务器，所以就不要清理套接字环境了，放置主服务器的套接字环境被清除
		//WSACleanup();
#else
		for (auto iter : _clients)
		{
			close(iter.second->sockfd());
		}
		close(_sock);
#endif
		_clients.clear();
		_sock = INVALID_SOCKET;
		delete _pthread;
	}
}

bool CellServer::Onrun()
{
	while (isRun())
	{
		//如果客户端缓冲队列_clientsBuff中有新客户，那么就将其加入到_clients中
		if (_clientsBuff.size() > 0)
		{
			//自解锁lock_guard，作用域结束后自动释放锁，因此if执行结束之后，_mutex就被释放了
			std::lock_guard<std::mutex> lock(_mutex);
			for (auto pClient : _clientsBuff)
			{
				_clients[pClient->sockfd()] = pClient;
			}
			_clientsBuff.clear();
			_clients_change = true;
		}

		//如果没有客户，那么休眠一秒然后继续下一次循环
		if (_clients.empty())
		{
			std::chrono::milliseconds t(1);
			std::this_thread::sleep_for(t);
			continue;
		}

		fd_set fdRead;
		FD_ZERO(&fdRead);
		//在主线程的select中已经对主服务端的_sock进行查询了，所以从服务器就不需要再将_sock加入到fd_set中了，否则两个地方同时操作会出错
		//FD_SET(_sock, &fdRead);

		//根据_fdRead_change判断是否有新客户端加入，如果有那么就进行新的FD_SET
		if (_clients_change)
		{
			_clients_change = false;
			for (auto iter : _clients)
			{
				FD_SET(iter.second->sockfd(), &fdRead);
				if (maxSock < iter.second->sockfd())
					maxSock = iter.second->sockfd();
			}
			//将更新后的fd_set保存到_fdRead_bak中
			memcpy(&_fdRead_bak, &fdRead, sizeof(_fdRead_bak));
		}
		//否则直接拷贝，不用再循环FD_SET了
		else
			memcpy(&fdRead, &_fdRead_bak, sizeof(_fdRead_bak));


		//从服务器一般只用来收取数据，所以这里设置为阻塞的也可以
		//struct timeval t = { 1,0 };
		int ret = select(maxSock + 1, &fdRead, nullptr, nullptr, nullptr);
		if (ret < 0)
		{
			std::cout << "Server：select出错！" << std::endl;
			CloseSocket();
			return false;
		}

#ifdef _WIN32
		//如果是WIN下运行，fd_set拥有fd_count与fd_array成员
		//我们可以遍历fd_set，然后从中获取数据，不需要使用FD_ISSET了
		for (int n = 0; n < fdRead.fd_count; n++)
		{
			auto iter = _clients.find(fdRead.fd_array[n]);
			//如果RecvData出错，那么就将该客户端从_client中移除
			if (-1 == RecvData(iter->second))
			{
				if (_pNetEvent)
					_pNetEvent->OnClientLeave(iter->second); //通知主服务器有客户端退出
				//delete iter->second;
				_clients.erase(iter->first);
				_clients_change = true; //这个要设置为true，因为有客户端退出了，需要重新进行FD_SET
			}
		}
#else
		//如果在UNIX下，fd_set无fd_count与fd_array成员，我们只能遍历_clients数组
		//遍历_clients map容器中所有的客户端，然后从中获取数据
		for (auto iter : _clients)
		{
			//因为_clients是一个map，因此每次iter返回一个pair,其first成员为key(SOCKET)，value成员为value(ClientSocket)
			if (FD_ISSET(iter.second->sockfd(), &fdRead))
			{
				//如果RecvData出错，那么就将该客户端从_client中移除
				if (-1 == RecvData(iter.second))
				{
					if (_pNetEvent)
						_pNetEvent->OnClientLeave(iter.second); //通知主服务器有客户端退出
					_clients.erase(iter.first);
					_clients_change = true; //原因同上
				}
			}
		}
#endif // _WIN32

	}
	return false;
}

int CellServer::RecvData(std::shared_ptr<ClientSocket>& pClient)
{
	char *_recvBuff = pClient->recvMsgBuff() + pClient->getRecvLastPos();
	//先将数据接收到_recvBuff缓冲区中
	int _nLen = recv(pClient->sockfd(), _recvBuff, RECV_BUFF_SIZE - pClient->getRecvLastPos(), 0);
	_pNetEvent->OnNetRecv(pClient);
	if (_nLen < 0) {
		//std::cout << "recv函数出错！" << std::endl;
		return -1;
	}
	else if (_nLen == 0) {
		//std::cout << "客户端<Socket=" << pClient->sockfd() << ">：已退出！" << std::endl;
		return -1;
	}

	//（不需要这一步了）将数据从_recvBuff中拷贝到客户端的缓冲区中
	//memcpy(pClient->msgBuff() + pClient->getLastPos(), _recvBuff, _nLen);

	pClient->setRecvLastPos(pClient->getRecvLastPos() + _nLen);
	//判断客户端的缓冲区中数据结尾的位置，如果有一个DataHeader的大小那么就可以对数据进行处理
	while (pClient->getRecvLastPos() >= sizeof(DataHeader))
	{
		//获取缓冲区的首指针
		DataHeader* header = (DataHeader*)pClient->recvMsgBuff();
		//如果当前缓冲区中数据结尾的位置大于等于一个数据包的大小，那么就对数据进行处理
		if (pClient->getRecvLastPos() >= header->dataLength)
		{
			//先保存剩余未处理消息缓冲区的长度
			int nSize = pClient->getRecvLastPos() - header->dataLength;
			//处理网络消息
			OnNetMessage(pClient, header);
			//处理完成之后，将_recvMsgBuff中剩余未处理部分的数据前移
			memcpy(pClient->recvMsgBuff(), pClient->recvMsgBuff() + header->dataLength, nSize);
			pClient->setRecvLastPos(nSize);
		}
		else {
			//消息缓冲区剩余数据不够一条完整消息
			break;
		}
	}
	return 0;
}

void CellServer::OnNetMessage(std::shared_ptr<ClientSocket>& pClient, DataHeader* header)
{
	//调用主服务OnNetMsg事件
	_pNetEvent->OnNetMsg(this, pClient, header);

	switch (header->cmd)
	{
	case CMD_LOGIN: //如果是登录
	{
		//Login *login = (Login*)header;
		//std::cout << "服务端：收到客户端<Socket=" << pClient->sockfd() << ">的消息CMD_LOGIN，用户名：" << login->userName << "，密码：" << login->PassWord << std::endl;

		//此处可以判断用户账户和密码是否正确等等（省略）

		//返回登录的结果给客户端
		auto ret = std::make_shared<LoginResult>();
		//在_taskServer()内部封装一个发送消息任务，然后执行该发送任务
		this->addSendTask(pClient, (std::shared_ptr<DataHeader>)ret);
	}
	break;
	case CMD_LOGOUT:  //如果是退出
	{
		//Logout *logout = (Logout*)header;
		//std::cout << "服务端：收到客户端<Socket=" << pClient->sockfd() << ">的消息CMD_LOGOUT，用户名：" << logout->userName << std::endl;

		//返回退出的结果给客户端
		auto ret = std::make_shared<LogoutResult>();
		this->addSendTask(pClient, (std::shared_ptr<DataHeader>)ret);
	}
	break;
	default:  //如果有错误
	{
		//std::cout << "服务端：收到客户端<Socket=" << pClient->sockfd() << ">的未知消息消息" << std::endl;

		//返回错误给客户端，DataHeader默认为错误消息
		auto ret = std::make_shared<DataHeader>();
		this->addSendTask(pClient, ret);
	}
	break;
	}
}

#endif

四、测试

server.cpp测试程序如下

#include "EasyTcpServer.hpp"
#include "MessageHeader.hpp"
#include "Alloctor.h"

int main()
{
	EasyTcpServer server;
	server.Bind("192.168.0.105", 4567);
	server.Listen(5);
	server.Start(4);

	while (server.isRun())
	{
		server.Onrun();
	}

	server.CloseSocket();
	std::cout << "服务端停止工作!" << std::endl;

	getchar();  //防止程序一闪而过
	return 0;
}

client.cpp测试程序如下

  • 代码中1000个客户端连接服务器，每次发送10条数据包
#include "EasyTcpClient.hpp"
#include "CELLTimestamp.hpp"
#include <thread>
#include <atomic>

bool g_bRun = false;
const int cCount = 1000;      //客户端的数量
const int tCount = 4;          //线程的数量
std::atomic_int sendCount = 0; //send()函数执行的次数
std::atomic_int readyCount = 0;//代表已经准备就绪的线程数量
EasyTcpClient* client[cCount]; //客户端的数组

void cmdThread();
void sendThread(int id);

int main()
{
	g_bRun = true;

	//UI线程，可以输入命令
	std::thread t(cmdThread);
	t.detach();

	//启动发送线程
	for (int n = 0; n < tCount; ++n)
	{
		std::thread t(sendThread, n + 1);
		t.detach();
	}

	//每1秒中打印一次信息(其中包括send()函数的执行次数)
	CELLTimestamp tTime;
	while (true)
	{
		auto t = tTime.getElapsedSecond();
		if (t >= 1.0)
		{
			printf("time<%lf>,thread numer<%d>,client number<%d>,sendCount<%d>\n",
				t, tCount, cCount, static_cast<int>(sendCount / t));
			sendCount = 0;
			tTime.update();
		}
		Sleep(1);
	}

	return 0;
}

void cmdThread()
{
	char cmdBuf[256] = {};
	while (true)
	{
		std::cin >> cmdBuf;
		if (0 == strcmp(cmdBuf, "exit"))
		{
			g_bRun = false;
			break;
		}
		else {
			std::cout << "命令不识别，请重新输入" << std::endl;
		}
	}
}

void sendThread(int id)
{
	/*
		下面这几个变量是为了平均每个线程创建的客户端的数量：
			例如，本次测试时客户端数量为1000，线程数量为4，那么每个线程应该创建250个客户端
				线程1：c=250，begin=0,end=250
				线程2：c=250，begin=250,end=500
				线程3：c=250，begin=500,end=750
				线程4：c=250，begin=750,end=1000
	*/
	int c = cCount / tCount;
	int begin = (id - 1)*c;
	int end = id*c;

	for (int n = begin; n < end; ++n) //创建客户端
	{
		client[n] = new EasyTcpClient;
	}
	for (int n = begin; n < end; ++n) //让每个客户端连接服务器
	{
		client[n]->ConnectServer("192.168.0.105", 4567);
	}
	printf("Thread<%d>,Connect=<begin=%d, end=%d>\n", id, (begin + 1), end);

	
	//将readyCount，然后判断readyCount是否达到了tCount
	//如果没有，说明所有的线程还没有准备好，那么就等待所有线程都准备好一起返回发送数据
	readyCount++;
	while (readyCount < tCount)
	{
		std::chrono::microseconds t(10);
		std::this_thread::sleep_for(t);
	}

	//这里定义为数组，可以随根据需求修改客户端单次发送给服务端的数据包数量
	const int nNum = 10;
	Login login[nNum];
	for (int n = 0; n < nNum; ++n)
	{
		strcpy(login[n].userName, "dongshao");
		strcpy(login[n].PassWord, "123456");
	}
	//在外面定义nLen，就不用每次在for循环中SendData时还要去sizeof计算一下login的大小
	int nLen = sizeof(login);
	//循环向服务端发送消息

	while (g_bRun)
	{
		for (int n = begin; n < end; ++n)
		{
			if (client[n]->SendData(login, nLen) != SOCKET_ERROR)
			{
				sendCount++;
			}
			client[n]->Onrun();
		}
		
	}

	//关闭客户端
	for (int n = begin; n < end; ++n)
	{
		client[n]->CloseSocket();
		delete client[n];
	}

	printf("thread:all clients close the connection!\n");
}

  • 结果如下所示：与之前未更改之前的差不多一样

五、备注

  • 因为程序中的shared_ptr内部申请与释放的内存就来自于我们的内存池，所以我们如果想要程序申请的内存都来自于我们的内存可以更改MemoryAlloctor的模板参数2来更改内存池的大小
//内存池管理工具(单例模式)
class MemoryMgr
{
private:
	//可以随意更改MemoryAlloctor的模板参数2来更改内存池的大小
	MemoryAlloctor<64, 4000000> _mem64;
	MemoryAlloctor<128, 1000000> _mem128;
	MemoryAlloc* _szAlloc[MAX_MEMORY_SIZE + 1];
};
  • 如果想要知道程序是否使用了系统的内存，那么可以在MemoryAlloc的allocMemory()函数中添加一个打印信息，如果使用了那么就会打印这句话，说明我们某块内存池用完了，开始向系统申请内存了
//内存池
class MemoryAlloc
{
	void *allocMemory(size_t nSize) //申请内存
	{
		//如果内存池使用完了，那么_pHeader应该指向某个内存块的最后nullptr尾指针
		//此时调用malloc向系统申请内存，不向我们的内存池进行内存申请
		if (_pHeader == nullptr) {
			//打印这条消息，可以知道我们的程序使用了
			printf("MemoryAlloc::allocMem: %llx,id=%d, size=%d\n", pReturn, pReturn->nID, nSize);
		}
	}
};

董哥的黑板报
关注
专栏目录
c++ 高并发网路通信架构实现资料.rar
01-08
c++ 高并发网路通信架构实现 socket 全栈 跨平台 开发。
项目(百万并发网络通信架构)10.4---服务端数据收发的性能瓶颈
董哥的黑板报
04-24 984
前言 现在我们把服务端和客户端的代码修改一下，修改为两者互相发送与接收数据 一、服务端最终代码 主要就是修改CellServer::OnNetMessage()中的代码，使其在接收到数据之后返回数据给客户端 #ifndef _EasyTcpClient_hpp_ #define _EasyTcpClient_hpp_ #ifdef _WIN32 #define FD_SETSIZE 10...
封装Server类和Client类
weixin_33691700的博客
06-12 178
服务器端： EasyTcpServer.hpp #ifndef _EasyTcpServer_hpp_ #define _EasyTcpServer_hpp_ #ifdef _WIN32 #define WIN32_LEAN_AND_MEAN #define _WINSOCK_DEPRECATED_NO_WARNINGS #include<WinSock2.h> ...
C++之auto的使用
zyc2018的博客
06-25 1万+
https://blog.csdn.net/huang_xw/article/details/8760403 C++11中引入的auto主要有两种用途：自动类型推断和返回值占位。 1. 自动类型推断 auto自动类型推断，用于从初始化表达式中推断出变量的数据类型。 #include <vector> #include <map> using nam...
EasyTCP 服务器消息接收与发送分离
wwxy1995的博客
07-10 592
在EasyTCP服务器发送数据的时候，不是直接调用send函数发送，而是设计一个发送缓冲区，每次从这个发送缓冲区中发送 //发送数据 int SendData(DataHeader* header) { int ret = SOCKET_ERROR; //要发送的数据长度 int nSendLen = header->dataLength; //要发送的数据 c...
华清远见-重庆中心-JAVA高级阶段知识点梳理
weixin_52590383的博客
12-04 617
Java高级
Chapter 12&13-高效并发
最新发布
m0_51710682的博客
08-05 852
对于64位的数据类型（long和double），允许虚拟机将没有被volatile修饰的64位数据的读写操作划分为两次32位的操作来进行，即允许虚拟机实现自行选择是否要保证64位数据类型的load、store、read和write这四个操作的原子性，这就是所谓的“long和double的非原子性协定”（Non-Atomic Treatment of double and long Variables）。当你的数据需要在不同代码块间共享，但这些代码块总是在同一线程中运行时，可以使用线程本地存储。
大型网站技术架构-第3篇 案例
dkjkls的博客
04-24 826
读书笔记 摘自：《大型网站技术架构：核心原理与案例分析-李智慧》9 淘宝网的架构演化案例分析9.1 淘宝网的业务发展历程业务驱动着技术不得不往前走9.2 淘宝网技术架构演化Linux+Apache+MySQL+PHP（LAMP）架构作为一个刚刚起步的小网站，使用开源、免费、简单的技术产品搭建网站是明智之举，可谓一举多得：免费的技术降低网站的成本，成熟的开源技术可以从开源社区获取文档和技术支持；网站发
Linux多线程服务端编程：使用muduo C++网络库
08-17
《Linux多线程服务端编程：使用muduo C++网络库》主要讲述采用现代C++在x86-64 Linux上编写多线程TCP网络服务程序的主流常规技术，重点讲解一种适应性较强的多线程服务器的编程模型，即one loop per thread。...
大数据技术之_19_Spark学习_06_Spark 源码解析 + Spark 通信架构、脚本解析、standalone 模式启动、提交流程 + Spark Shuffle 过程、内存管理与分配、部署
黑泽君
05-02 1003
大数据技术之_19_Spark学习_06_Spark 源码解析 + Spark 通信架构、脚本解析、standalone 模式启动、提交流程 + Spark Shuffle 过程 + Spark 内存管理与分配 + Spark 部署模式
2023 华为 Datacom-HCIE 真题题库 09/12--含解析
mxl00z的博客
05-28 3064
2023 华为 Datacom-HCIE 真题题库 09--含解析
Python安全攻防-从入门到入狱
热门推荐
Yasso的博客
01-31 1万+
Python安全攻防-从入门到入狱
EasyTcp 服务器完善版
wwxy1995的博客
07-09 814
服务器主函数 #include "EasyTcpServer.hpp" #include<thread> bool g_bRun = true; void cmdThread() { while (true) { char cmdBuf[256] = {}; scanf("%s", cmdBuf); if (0 == strcmp(cmdBuf, "exit")...
EasyTcp服务器端编程分离客户端连接与消息处理业务
wwxy1995的博客
07-09 356
服务器主程序函数，非常清晰 #include "EasyTcpServer.hpp" #include<thread> bool g_bRun = true; void cmdThread() { while (true) { char cmdBuf[256] = {}; scanf("%s", cmdBuf); if (0 == strcmp(cmdBuf, "...
项目(百万并发网络通信架构)11.2---将消息的接收与发送分离
董哥的黑板报
04-26 1891
一、前言 在前面的文章中，我们的服务端在接收到客户端的数据之后，对数据处理完会给客户端回送一个消息，这种做法实际上是低效的，因为在数据recv()接收之后再去调用send()，那么当send()阻塞或速度过慢时，会导致当前线程无法继续向前执行或执行较慢，那么会影响后面recv()的再次调用，从而降低服务端的性能，而且会使缓冲区膨胀 二、将消息的接收与发送分离方案 我们将recv()之后给客户端...
项目(百万并发网络通信架构)10.2---recv()函数的极限测试
董哥的黑板报
04-24 1844
待续
项目(百万并发网络通信架构)1---Windows下Tcp Socket的基本结构（客户端、服务端）
董哥的黑板报
04-02 1668
一、服务端代码 #define WIN32_LEAN_AND_MEAN #define _WINSOCK_DEPRECATED_NO_WARNINGS //for inet_pton() #include <windows.h> #include <WinSock2.h> #include <iostream> #pragma comment(lib...
评论 7
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

董哥的黑板报

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值