TcpServer 服务器优化之后,加了多线程,对心跳包进行优化

TcpServer 服务器优化之后,加了多线程,对心跳包进行优化

TcpServer.h

#ifndef TCPSERVER_H
#define TCPSERVER_H#include <iostream>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <vector>
#include <map>
#include <string>
#include <ctime>// 引入静态链接库
#pragma comment(lib, "ws2_32.lib")
#define HEARTBEATTIME  1000class TcpServer {
public:TcpServer();~TcpServer();// 启动服务器,监听指定端口bool start(int port);// 停止服务器void stop();// 发送数据给指定客户端int sendData(SOCKET clientSocket, const char* data, int dataLength);// 处理服务器业务逻辑,通常在循环中调用void handle();//链接static DWORD WINAPI ThreadAccept(LPVOID lpParam);//接收数据static DWORD WINAPI ThreadRecvData(LPVOID lpParam);//心跳包static DWORD WINAPI ThreadHeartBeat(LPVOID lpParam);
public:std::vector<SOCKET> socketsToRemove;BOOL  m_bExit;//程序是否关闭BOOL m_bHeartBeat;//是否启用心跳包int heartbeatInterval;  // 心跳包间隔时间(秒)
private:SOCKET listenSocket;std::vector<SOCKET> clientSockets;std::map<SOCKET, std::time_t> clientLastHeartbeatTime;// 设置套接字为非阻塞模式bool setSocketNonBlocking(SOCKET socket);// 接受新的客户端连接void acceptNewClients();// 接收客户端数据void receiveClientData();// 发送心跳包给客户端,并检测客户端响应void sendHeartbeatsAndCheck();// 移除已断开连接的客户端void removeDisconnectedClients(std::vector<SOCKET> &socketsToRemove);
};#endif

TcpServer.cpp

#include "TcpServer.h"// 构造函数,初始化相关成员变量
TcpServer::TcpServer() : listenSocket(INVALID_SOCKET),
heartbeatInterval(5), m_bExit(false), m_bHeartBeat(false)
{WSADATA wsaData;int result = WSAStartup(MAKEWORD(2, 2), &wsaData);if (result != 0) {std::cerr << "WSAStartup failed: " << result << std::endl;}
}// 析构函数,关闭套接字并清理WinSock环境
TcpServer::~TcpServer() 
{stop();WSACleanup();
}// 启动服务器,监听指定端口
bool TcpServer::start(int port)
{listenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if (listenSocket == INVALID_SOCKET){std::cerr << "Socket creation failed: " << WSAGetLastError() << std::endl;return false;}if (!setSocketNonBlocking(listenSocket)) {std::cerr << "Failed to set listen socket non-blocking" << std::endl;closesocket(listenSocket);return false;}sockaddr_in serverAddr;serverAddr.sin_family = AF_INET;serverAddr.sin_addr.s_addr = INADDR_ANY;serverAddr.sin_port = htons(port);int result = bind(listenSocket, (sockaddr*)&serverAddr, sizeof(serverAddr));if (result == SOCKET_ERROR){std::cerr << "Bind failed: " << WSAGetLastError() << std::endl;closesocket(listenSocket);return false;}result = listen(listenSocket, SOMAXCONN);if (result == SOCKET_ERROR){std::cerr << "Listen failed: " << WSAGetLastError() << std::endl;closesocket(listenSocket);return false;}return true;
}// 停止服务器
void TcpServer::stop()
{if (listenSocket != INVALID_SOCKET) {closesocket(listenSocket);listenSocket = INVALID_SOCKET;}for (SOCKET clientSocket : clientSockets) {closesocket(clientSocket);}clientSockets.clear();clientLastHeartbeatTime.clear();
}// 设置套接字为非阻塞模式
bool TcpServer::setSocketNonBlocking(SOCKET socket) 
{u_long iMode = 1;int result = ioctlsocket(socket, FIONBIO, &iMode);if (result == SOCKET_ERROR){std::cerr << "ioctlsocket failed: " << WSAGetLastError() << std::endl;return false;}return true;
}// 发送数据给指定客户端
int TcpServer::sendData(SOCKET clientSocket, const char* data, int dataLength) 
{if (clientSocket == INVALID_SOCKET) {std::cerr << "Invalid client socket, cannot send data" << std::endl;return SOCKET_ERROR;}int totalBytesSent = 0;while (totalBytesSent < dataLength) {int bytesSent = ::send(clientSocket, data + totalBytesSent, dataLength - totalBytesSent, 0);if (bytesSent == SOCKET_ERROR){if (WSAGetLastError() == WSAEWOULDBLOCK) {// 暂时无法发送,等待下次尝试continue;}return SOCKET_ERROR;}totalBytesSent += bytesSent;}return totalBytesSent;
}// 接受新的客户端连接
void TcpServer::acceptNewClients()
{SOCKET newClientSocket = accept(listenSocket, NULL, NULL);if (newClientSocket == INVALID_SOCKET){if (WSAGetLastError() != WSAEWOULDBLOCK) {std::cerr << "Accept failed: " << WSAGetLastError() << std::endl;}return;}else{std::cout << "Accept success: " << newClientSocket << std::endl;}if (!setSocketNonBlocking(newClientSocket)) {std::cerr << "Failed to set client socket non-blocking" << std::endl;closesocket(newClientSocket);return;}clientSockets.push_back(newClientSocket);clientLastHeartbeatTime[newClientSocket] = std::time(nullptr);
}// 接收客户端数据
void TcpServer::receiveClientData()
{for (size_t i = 0; i < clientSockets.size(); ++i){SOCKET clientSocket = clientSockets[i];char buffer[1024];int bytesReceived = ::recv(clientSocket, buffer, sizeof(buffer), 0);if (bytesReceived == SOCKET_ERROR) {if (WSAGetLastError() == WSAEWOULDBLOCK) {// 暂时无数据可读,继续检查下一个客户端continue;}}else{buffer[bytesReceived] = '\0';std::string receivedData(buffer);// 在这里可以根据接收到的数据进行具体业务逻辑处理,比如解析命令等std::cout << "Received from client " << clientSocket << ": " << receivedData << std::endl;clientLastHeartbeatTime[clientSocket] = std::time(nullptr);std::string heartbeatData = buffer;heartbeatData+="    recvok:";int sentBytes = sendData(clientSocket, heartbeatData.c_str(), heartbeatData.length());}}
}// 发送心跳包给客户端,并检测客户端响应
void TcpServer::sendHeartbeatsAndCheck()
{const char* heartbeatData = "HEARTBEAT";  // 简单的心跳包内容,可自定义int dataLength = strlen(heartbeatData);for (auto& clientPair : clientLastHeartbeatTime){SOCKET clientSocket = clientPair.first;std::time_t& lastHeartbeatTime = clientPair.second;std::time_t currentTime = std::time(nullptr);if (currentTime - lastHeartbeatTime > heartbeatInterval) {// 超过心跳间隔时间没收到心跳响应,认为客户端连接异常socketsToRemove.push_back(clientSocket);continue;}int sentBytes = sendData(clientSocket, heartbeatData, dataLength);if (sentBytes == SOCKET_ERROR){// 发送心跳包失败,认为客户端连接可能有问题socketsToRemove.push_back(clientSocket);continue;}}
}// 移除已断开连接的客户端(更新函数定义,无参数)
void TcpServer::removeDisconnectedClients(std::vector<SOCKET>&socketsToRemove)
{for (SOCKET socketToRemove : socketsToRemove){auto it = std::find(clientSockets.begin(), clientSockets.end(), socketToRemove);if (it != clientSockets.end()) {std::cout << "Remove :"<< * it << std::endl;clientSockets.erase(it);clientLastHeartbeatTime.erase(socketToRemove);}}
}//接收链接线程
DWORD WINAPI TcpServer::ThreadAccept(LPVOID lpParam)
{TcpServer* t_Server = static_cast<TcpServer*>(lpParam);while (t_Server->m_bExit==false){t_Server->acceptNewClients();}return 0;
}
//接收数据
DWORD WINAPI TcpServer::ThreadRecvData(LPVOID lpParam)
{TcpServer* t_Server = static_cast<TcpServer*>(lpParam);while (t_Server->m_bExit == false){t_Server->receiveClientData();}return 0;
}//心跳包
DWORD WINAPI TcpServer::ThreadHeartBeat(LPVOID lpParam)
{TcpServer* t_Server = static_cast<TcpServer*>(lpParam);while (t_Server->m_bExit == false){Sleep(HEARTBEATTIME);t_Server->sendHeartbeatsAndCheck();if (t_Server->heartbeatInterval > 0){t_Server->removeDisconnectedClients(t_Server->socketsToRemove);}}return 0;
}
// 处理服务器业务逻辑,通常在循环中调用
void TcpServer::handle() 
{//创建4个线程,分别进行接收链接  接收数据 发送数据 发送心跳包// 创建线程,传入当前对象指针作为参数,线程启动函数为 SendHeartbeatHANDLE acceptThreadHandle = CreateThread(NULL, 0, ThreadAccept, this, 0, NULL);if (acceptThreadHandle == NULL){std::cerr << "Create accept thread failed: " << GetLastError() << std::endl;}else{std::cerr << "Create accept thread success: " << acceptThreadHandle << std::endl;}HANDLE recvDatatThreadHandle = CreateThread(NULL, 0, ThreadRecvData, this, 0, NULL);if (acceptThreadHandle == NULL){std::cerr << "Create recvData thread failed: " << GetLastError() << std::endl;}else{std::cerr << "Create recvData thread success: " << recvDatatThreadHandle << std::endl;}if (m_bHeartBeat == true){HANDLE heartBeatThreadHandle = CreateThread(NULL, 0, ThreadHeartBeat, this, 0, NULL);if (acceptThreadHandle == NULL){std::cerr << "Create heartBeat thread failed: " << GetLastError() << std::endl;}else{std::cerr << "Create heartBeat thread success: " << heartBeatThreadHandle << std::endl;}}}

main.cpp

#include "TcpServer.h"int main()
{TcpServer server;server.heartbeatInterval = 30;server.m_bHeartBeat = true;if (server.start(8080)) {while (true) {server.handle();// 可以在这里添加适当的延时,避免过于频繁地循环处理,消耗过多CPU资源Sleep(100);break;}}else{std::cout << "server initiatefail" << std::endl;}Sleep(1000000);return 0;
}

在这里插入图片描述
在这里插入图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/889208.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

python进阶-05-利用Selenium来实现动态爬虫

python进阶-05-利用Selenium来实现动态爬虫 一.说明 这是python进阶部分05&#xff0c;我们上一篇文章学习了Scrapy来爬取网站&#xff0c;但是很多网站需要登录才能爬取有用的信息&#xff0c;或者网站的静态部分是一个空壳&#xff0c;内容是js动态加载的,或者人机验证&…

Linux —— vim 编辑器

一、什么是vim vim是一个功能强大、高度可定制的文本编辑器。以下是对vim编辑器的具体介绍&#xff1a; 历史背景&#xff1a;vim最初由Bram Moolenaar在1991年开发&#xff0c;作为vi编辑器的增强版&#xff0c;增加了许多新的特性和改进。它继承了vi的基本编辑功能和键盘快捷…

高效率同步降压转换器 - YB2416D: 实现快速充电和高效能供电的利器

概述: YB2416是一款输入耐压超过40V&#xff0c;在4.5V~30V输入电压条件下正常工作&#xff0c;并且能够实现精确恒压以及恒流的同步降压型DC-DC转换器。 内部集成80m2的上管和40m2的下管&#xff0c;无需外部肖特基二极管&#xff0c;可连续输出3A电流。输出3A电流时系统转换…

Repo管理

文章目录 前言Repo介绍清单仓库清单仓库的组成 初始化Repo同步远程仓库Repo实际应用 前言 我们知道&#xff0c;Git是用来管理某一个仓库&#xff0c;那当一个项目用到了多个仓库时&#xff0c;怎么来同步管理这些仓库呢&#xff1f;这个时候就可以引入Repo管理。 Repo介绍 …

神经网络的起源与工作原理

神经网络起源&#xff1a;一个生物神经网络是由一组化学上相连或功能上相关的神经元组成。一个神经元可能与许多其他神经元相连&#xff0c;网络中的神经元和连接的总数可能很广泛。连接&#xff0c;称为突触&#xff0c;通常是从轴突到树突形成的&#xff0c;尽管树突和其他连…

Qwen2.5-7B-Instruct vLLM 部署调用

Qwen2.5-7B-Instruct vLLM 部署调用 vLLM 简介 vLLM 框架是一个高效的大语言模型推理和部署服务系统&#xff0c;具备以下特性&#xff1a; 高效的内存管理&#xff1a;通过 PagedAttention 算法&#xff0c;vLLM 实现了对 KV 缓存的高效管理&#xff0c;减少了内存浪费&…

解决 Mac(M1/M2)芯片,使用node 14版本

前言 nvm 在安装 Node.js v14.21.3 时&#xff0c;报错&#xff1a; nvm install 14 Downloading and installing node v14.21.3... Downloading https://nodejs.org/dist/v14.21.3/node-v14.21.3-darwin-arm64.tar.xz... curl: (56) The requested URL returned error: 404Bin…

TesseractOCR-GUI:基于WPF/C#构建TesseractOCR简单易用的用户界面

前言 前篇文章使用Tesseract进行图片文字识别介绍了如何安装TesseractOCR与TesseractOCR的命令行使用。但在日常使用过程中&#xff0c;命令行使用还是不太方便的&#xff0c;因此今天介绍一下如何使用WPF/C#构建TesseractOCR简单易用的用户界面。 普通用户使用 参照上一篇教…

【ETCD】【源码阅读】configurePeerListeners() 函数解析

configurePeerListeners 是 ETCD 的一个核心函数&#xff0c;用于为集群中节点之间的通信配置监听器&#xff08;Peer Listener&#xff09;。这些监听器主要负责 Raft 协议的消息传递、日志复制等功能。函数返回一个包含所有监听器的列表。 函数签名 func configurePeerList…

uniapp改成用vue起项目

目的&#xff1a;让项目按照vue的打包流程跑流水线 1.按照uniapp官网教程执行 2.执行第二条命令时报错 ERROR Failed to get response from true/vue-cli-version-marker 3.解决方式 报错可能跟yarn有关&#xff0c;然后切换成npm 找到自己本地电脑的这个文件 按照截图修…

【SH】微信小程序调用EasyDL零门槛AI开发平台的图像分类研发笔记

文章目录 微信小程序字符串字符串模板字符串拼接 上传图片编写JS代码编写wxml代码编写wxss代码 GET请求测试编写测试代码域名不合法问题 GET和POST请求测试编写JS代码编写wxml代码编写wxss代码 效果展示 微信小程序字符串 字符串模板 这是ES6引入的特性&#xff0c;允许你通过…

[小白系列]Ubuntu安装教程-安装prometheus和Grafana

Docker安装prometheus 拉取镜像 docker pull prom/prometheus 配置文件prometheus.yml 在/data/prometheus/建立prometheus.yml配置文件。&#xff08;/data/prometheus/可根据自己需要调整&#xff09; global:scrape_interval: 15s # By default, scrape targets ev…

[大数据]Hudi编译集成

1. Hudi概述 1.1 Hudi简介 What is Apache Hudi Apache Hudi is the next generation streaming data lake platform. Apache Hudi brings core warehouse and database functionality directly to a data lake. Hudi provides tables, transactions, efficient upserts/dele…

windows下 mysql开启 binlog日志

一、查看是否开启 binlog -- 方式一 show binary logs;-- 方式二 show VARIABLES like log_bin 说明没有开启 方式一 &#xff1a;you are not using binary logging 方式二&#xff1a;log_bin off 二、编辑 my.ini 配置文件 默认安装地点位于&#xff1a;C:\ProgramDat…

Java-22 深入浅出 MyBatis - 手写ORM框架3 手写SqlSession、Executor 工作原理

点一下关注吧&#xff01;&#xff01;&#xff01;非常感谢&#xff01;&#xff01;持续更新&#xff01;&#xff01;&#xff01; 大数据篇正在更新&#xff01;https://blog.csdn.net/w776341482/category_12713819.html 目前已经更新到了&#xff1a; MyBatis&#xff…

Android 逆向/反编译/Hook修改应用行为 基础实现

前言&#xff1a;本文通过一个简单的情景案例实现安卓逆向的基本操作 一、情景描述 本文通过一个简单的情景案例来实现安卓逆向的基本操作。在这个案例中所使用的项目程序是我自己的Demo程序&#xff0c;不会造成任何的财产侵害&#xff0c;本文仅作为日常记录及案例分享。实…

IDEA创建Spring Boot项目配置阿里云Spring Initializr Server URL【详细教程-轻松学会】

1.首先打开idea选择新建项目 2.选择Spring Boot框架(就是选择Spring Initializr这个) 3.点击中间界面Server URL后面的三个点更换为阿里云的Server URL Idea中默认的Server URL地址&#xff1a;https://start.spring.io/ 修改为阿里云Server URL地址&#xff1a;https://star…

基于MATLAB的信号处理工具:信号分析器

信号&#xff08;或时间序列&#xff09;是与特定时间相关的一系列数字或测量值&#xff0c;不同的行业和学科将这一与时间相关的数字序列称为信号或时间序列。生物医学或电气工程师会将其称为信号&#xff0c;而统计学家或金融定量分析师会使用时间序列这一术语。例如&#xf…

Plugin - 插件开发03_Spring Boot动态插件化与热加载

文章目录 Pre方案概览使用插件的好处流程CodePlugin 定义Plugin 实现Plugin 使用方动态加载插件类加载器注册与卸载插件配置文件启动类测试验证 小结 Pre 插件 - 通过SPI方式实现插件管理 插件 - 一份配置&#xff0c;离插件机制只有一步之遥 插件 - 插件机制触手可及 Plug…

ECharts柱状图-阶梯瀑布图,附视频讲解与代码下载

引言&#xff1a; 在数据可视化的世界里&#xff0c;ECharts凭借其丰富的图表类型和强大的配置能力&#xff0c;成为了众多开发者的首选。今天&#xff0c;我将带大家一起实现一个柱状图图表&#xff0c;通过该图表我们可以直观地展示和分析数据。此外&#xff0c;我还将提供…