先做一个简单实例:
#include <arpa/inet.h>
#include <assert.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#include <unistd.h>
#include <iostream>
#include "../epollctl.hpp"
void handlerClient(int clientFd) {
std::string msg;
if (not EchoServer::RecvMsg(clientFd, msg)) {
return;
}
EchoServer::SendMsg(clientFd, msg);
}
int main(int argc, char *argv[]) {
if (argc != 3) {
std::cout << "invalid input" << std::endl;
std::cout << "example: ./Epoll 0.0.0.0 1688" << std::endl;
return -1;
}
int sockFd = EchoServer::CreateListenSocket(argv[1], atoi(argv[2]), false);
if (sockFd < 0) {
return -1;
}
epoll_event events[2048];
int epollFd = epoll_create(1024);
if (epollFd < 0) {
perror("epoll_create failed");
return -1;
}
EchoServer::Conn conn(sockFd, epollFd, false);
EchoServer::SetNotBlock(sockFd);
EchoServer::AddReadEvent(&conn);
while (true) {
int num = epoll_wait(epollFd, events, 2048, -1);
if (num < 0) {
perror("epoll_wait failed");
continue;
}
for (int i = 0; i < num; i++) {
EchoServer::Conn *conn = (EchoServer::Conn *)events[i].data.ptr;
if (conn->Fd() == sockFd) {
EchoServer::LoopAccept(sockFd, 2048, [epollFd](int clientFd) {
EchoServer::Conn *conn = new EchoServer::Conn(clientFd, epollFd, false);
EchoServer::AddReadEvent(conn); // 监听可读事件,保持fd为阻塞IO
EchoServer::SetTimeOut(conn->Fd(), 0, 500000); // 设置读写超时时间为500ms
});
continue;
}
handlerClient(conn->Fd());
EchoServer::ClearEvent(conn);
delete conn;
}
}
return 0;
}
方法调用的链路过程如下:
- 程序首先在
main
函数中检查命令行参数数量是否正确,如果不正确则输出提示信息并返回。 - 调用
EchoServer::CreateListenSocket
创建监听套接字。 - 调用
epoll_create
创建 epoll 实例。 - 创建
EchoServer::Conn
对象,并设置套接字为非阻塞模式,添加可读事件。 - 进入无限循环,通过
epoll_wait
等待事件发生。 - 当有事件发生时,遍历事件。如果是监听套接字的事件,调用
EchoServer::LoopAccept
处理新的连接,包括创建新的连接对象、添加可读事件和设置超时时间。 - 如果是客户端连接的事件,调用
handlerClient
处理客户端的消息收发。 - 处理完事件后,调用
EchoServer::ClearEvent
清除事件,并删除连接对象。