当前位置:   article > 正文

C++网络编程实践:使用C++11基于epoll技术实现一个超大并发TCP服务器_c++ epoll高并发服务器详细代码

c++ epoll高并发服务器详细代码

理解epoll

epoll 是一种 I/O 复用技术,它允许一个线程有效地管理多个文件描述符(在本例中为套接字),而无需为每个连接创建单独的线程。这是通过事件驱动的方式来实现的,epoll 能够通知你哪些套接字已经准备好进行读写操作。

在传统的基于线程的模型中,每当有一个新的连接,服务器就会创建一个新的线程来处理这个连接。这种方法在连接数量较少时工作良好,但在高并发场景下会遇到线程上下文切换开销大、系统资源消耗过多等问题。

相比之下,epoll 允许你在一个线程中处理数千甚至数百万个连接,而不会因为线程管理带来额外的开销。epoll 通过注册事件(如 EPOLLIN 表示可读,EPOLLOUT 表示可写)来监控多个套接字。当某些套接字准备好时,epoll_wait 函数会立即返回这些套接字的列表,然后服务器可以依次处理这些事件,而无需创建新线程。

使用epoll的优势

使用 epoll 进行网络编程相比于传统的 selectpoll 方法具有显著的优势,尤其在处理大量并发连接的场景下更为明显

  1. 高效率epoll 使用更高效的数据结构和算法,例如基于红黑树的实现,使得它在处理大量文件描述符时效率远高于 selectpoll。它能够快速查找和更新事件状态,减少了系统调用的次数和上下文切换的开销。

  2. 扩展性epoll 支持水平触发(Level Triggered, LT)和边缘触发(Edge Triggered, ET)两种模式。ET 模式下,epoll 只在事件首次发生时报告,这可以减少不必要的事件报告,提高效率。LT 模式则允许重复报告事件直到被处理。

  3. 节省资源epoll 不需要为每一个文件描述符维护一个内核数据结构,而是使用一个文件描述符来管理多个连接,这样减少了内存使用和系统资源的消耗。

  4. 无连接数限制select 的最大限制是受 FD_SETSIZE 的约束,而 epoll 没有硬性的连接数限制,理论上可以处理成千上万乃至更多数量的并发连接。

  5. 精确的超时控制epoll 提供了更精确的超时控制,这对于需要高精度超时的网络应用尤为重要。

  6. 低延迟和高吞吐量epoll 的事件通知机制允许快速响应网络事件,降低延迟并提高整体吞吐量。

  7. 事件驱动模型epoll 基于事件驱动,这意味着只有当套接字上的事件真正发生时才会有通知,这避免了轮询所有连接所带来的开销。

  8. 支持多路复用epoll 可以同时处理多个套接字的读写事件,使得服务器可以有效地处理大量并发请求,而无需为每个连接创建额外的线程或进程。

  9. 低内存拷贝开销: 在处理事件时,epoll 可以直接访问内核中的数据结构,减少了用户空间和内核空间之间不必要的内存拷贝。

  10. 灵活的通知机制epoll 允许应用程序注册不同类型的事件,包括读事件、写事件等,以及错误条件和挂起事件。

代码样例 

下面用epoll技术实现一个TCP服务器,具体方法和思路看注释

  1. #include <iostream>
  2. #include <string.h>
  3. #include <unistd.h>
  4. #include <sys/socket.h> // socket, bind, listen, accept
  5. #include <netinet/in.h> // sockaddr_in
  6. #include <arpa/inet.h> // inet_addr
  7. #include <fcntl.h> // fcntl
  8. #include <errno.h>
  9. #include <sys/epoll.h> // epoll
  10. #include <unordered_map>
  11. #include <string>
  12. #include <vector>
  13. #define MAX_EVENTS 3000
  14. class TCPServer {
  15. public:
  16. //构造函数
  17. TCPServer(int port) : listenSocket(-1), port(port){}
  18. ~TCPServer() {
  19. if (listenSocket != -1)
  20. close(listenSocket);
  21. }
  22. void start() {
  23. initSocket();
  24. setupEpoll();
  25. listen(listenSocket, SOMAXCONN);
  26. std::cout << "Server started on port " << port << std::endl;
  27. run();
  28. }
  29. private:
  30. int listenSocket;
  31. int epollFd;
  32. int port;
  33. int msgCount;
  34. /**
  35. * 初始化监听套接字
  36. * 该函数创建一个监听套接字,并将其绑定到指定的IP地址和端口上。
  37. * 如果创建或绑定套接字时发生错误,程序将输出错误信息并退出。
  38. */
  39. void initSocket() {
  40. // 创建一个流式套接字
  41. listenSocket = socket(AF_INET, SOCK_STREAM, 0);
  42. if (listenSocket == -1) {
  43. perror("Error creating socket");
  44. exit(EXIT_FAILURE);
  45. }
  46. // 设置套接字选项,允许重复使用地址
  47. // 这在重启服务时特别有用,可以避免因为地址还处于TIME_WAIT状态而无法绑定
  48. int optval = 1;
  49. setsockopt(listenSocket, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
  50. // 初始化sockaddr_in结构体,用于指定套接字要绑定的地址和端口
  51. struct sockaddr_in addr;
  52. memset(&addr, 0, sizeof(addr));
  53. addr.sin_family = AF_INET;
  54. addr.sin_port = htons(port);
  55. addr.sin_addr.s_addr = htonl(INADDR_ANY);
  56. // 将套接字绑定到指定的地址和端口上
  57. if (bind(listenSocket, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
  58. perror("Error binding socket");
  59. close(listenSocket);
  60. exit(EXIT_FAILURE);
  61. }
  62. }
  63. /**
  64. * 演示epoll的使用
  65. * 初始化epoll监听结构体
  66. * 本函数用于创建epoll实例,并将监听socket添加到epoll中,以非阻塞方式监听客户端连接请求。
  67. *
  68. * @note 如果创建epoll实例或添加监听socket到epoll中失败,程序将退出。
  69. */
  70. void setupEpoll() {
  71. // 创建epoll实例
  72. epollFd = epoll_create1(0);
  73. if (epollFd == -1) {
  74. // 如果创建失败,输出错误信息,关闭监听socket,并退出程序
  75. perror("Error creating epoll instance");
  76. close(listenSocket);
  77. exit(EXIT_FAILURE);
  78. }
  79. struct epoll_event ev;
  80. // 设置epoll事件类型为可读事件和边缘触发
  81. ev.events = EPOLLIN | EPOLLET;
  82. // 将监听socket关联到epoll事件中
  83. ev.data.fd = listenSocket;
  84. if (epoll_ctl(epollFd, EPOLL_CTL_ADD, listenSocket, &ev) == -1) {
  85. // 如果添加监听socket到epoll失败,输出错误信息,关闭监听socket和epoll实例,并退出程序
  86. perror("Error adding listen socket to epoll");
  87. close(listenSocket);
  88. close(epollFd);
  89. exit(EXIT_FAILURE);
  90. }
  91. // 将监听socket设置为非阻塞模式
  92. fcntl(listenSocket, F_SETFL, O_NONBLOCK);
  93. }
  94. /**
  95. * 主循环,负责监听和处理客户端连接。
  96. * 该循环将持续运行,直到发生无法恢复的错误。
  97. * 它通过epoll_wait监控客户端连接和已建立的连接上的活动。
  98. */
  99. void run() {
  100. while (true) {
  101. // 准备接收epoll_wait返回的事件。
  102. std::vector<struct epoll_event> events(MAX_EVENTS);
  103. // 调用epoll_wait阻塞,直到有事件发生或超时。超时时间为-1,表示无限等待。
  104. int numEvents = epoll_wait(epollFd, events.data(), MAX_EVENTS, -1);
  105. // 检查epoll_wait调用是否失败。
  106. if (numEvents == -1) {
  107. perror("Error in epoll_wait");
  108. break;
  109. }
  110. // 遍历发生的事件,区分监听套接字和客户端套接字的事件。
  111. for (int i = 0; i < numEvents; ++i) {
  112. // 如果事件来源是监听套接字,则处理新的客户端连接请求。
  113. if (events[i].data.fd == listenSocket) {
  114. handleNewConnection();
  115. }
  116. else {
  117. // 如果事件来源是已建立的客户端套接字,则处理客户端的请求或数据。
  118. handleClient(events[i].data.fd);
  119. }
  120. }
  121. }
  122. }
  123. /**
  124. * 接收新连接
  125. */
  126. void handleNewConnection() {
  127. int clientSocket = accept(listenSocket, nullptr, nullptr);
  128. if (clientSocket == -1) {
  129. if (errno != EAGAIN && errno != EWOULDBLOCK) {
  130. perror("Error accepting new connection");
  131. }
  132. return;
  133. }
  134. //添加到epoll中
  135. struct epoll_event ev;
  136. ev.events = EPOLLIN | EPOLLET;
  137. ev.data.fd = clientSocket;
  138. if (epoll_ctl(epollFd, EPOLL_CTL_ADD, clientSocket, &ev) == -1) {
  139. perror("Error adding client socket to epoll");
  140. close(clientSocket);
  141. return;
  142. }
  143. //设置非阻塞模式
  144. fcntl(clientSocket, F_SETFL, O_NONBLOCK);
  145. struct sockaddr_in addr;
  146. socklen_t len = sizeof(addr);
  147. getpeername(clientSocket, (struct sockaddr*)&addr, &len);
  148. std::string address = inet_ntoa(addr.sin_addr);
  149. std::cout << "Client connected " << address << std::endl;
  150. }
  151. /**
  152. * 处理客户端消息
  153. */
  154. void handleClient(int clientSocket) {
  155. ssize_t bytesRead = read(clientSocket, buffer, sizeof(buffer) - 1);
  156. if (bytesRead == -1) {
  157. if (errno != EAGAIN && errno != EWOULDBLOCK) {
  158. perror("Error reading from client");
  159. }
  160. return -1;
  161. }else if(bytesRead == 0){
  162. close(clientSocket);
  163. epoll_ctl(epollFd, EPOLL_CTL_DEL, clientSocket, nullptr);
  164. std::cout << "Client Disconnected " << std::endl;
  165. return 0;
  166. }
  167. std::cout << "Received data " << bytesRead << std::endl;
  168. //回复一些数据
  169. //write(clientSocket, buffer, len);
  170. }
  171. };
  172. int main(int argc, char* argv[]) {
  173. int PORT = 8080;
  174. if (argc >= 2) {
  175. PORT = std::atoi(argv[1]);
  176. }
  177. TCPServer server(PORT);
  178. server.start();
  179. return 0;
  180. }

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/878262
推荐阅读
相关标签
  

闽ICP备14008679号