//所有内容均来自linux高性能服务器编程这本书.
利用epoll中的EPOLLONESHOT实现一段时间内只能有一个线程在处理一个SOKCET 防止该线程刚读取完数据准备处理 又来了一个数据 启动另外一个线程来处理 ONESHOT只会触发一次 所以当该工作进程处理完当前的事件之后,应该重置EPOLL_CTL_MOD 该FD状态 以便于下一次能够再次触发
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
// linux下使用结构体一定要在前面加入struct这个字 不然是没有效果的
#define MAX_EVENT_NUMBER 1024
#define BUFFER_SIZE 1024
// 由于之前没有什么敲linux下代码的经验
// 这次debug调式了很久的时间 需要注意的点就是如果是应用一个结构体 那么就需要在前面加入struct关键字 对于指针的传递一定要写&地址符号 (在实际编程中很少这样写 导致频繁出错) !!
typedef struct // 定义一个结构体
{
int epollfd;
int socket;
}fds;
int setnoblock(int fd)
{
int old_option = fcntl(fd,F_GETFL);
int new_option = old_option | O_NONBLOCK;
fcntl(fd,F_SETFL,new_option);
return old_option;
}
void addfd(int epollfd,int fd,bool oneshot)
{
struct epoll_event event;
event.data.fd = fd;
event.events = EPOLLIN | EPOLLET;
if(oneshot)
{
event.events |= EPOLLONESHOT;
}
epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&event);
setnoblock(fd);
}
void reset_oneshot(int epollfd,int fd)
{
struct epoll_event event;
event.data.fd = fd;
event.events = EPOLLIN | EPOLLET | EPOLLONESHOT;
epoll_ctl(epollfd,EPOLL_CTL_MOD,fd,&event);
}
void* work(void* arg) // 主线程已经帮我连接了 为只是负责处理这个事件
{
int sockfd = (( fds* )arg)->socket;
int epollfd = (( fds* )arg)->epollfd;
printf("start new thread to receive msg on %d\n",socket);
char buf[BUFFER_SIZE];
memset(buf,0, BUFFER_SIZE);
while(1)
{
int ret = recv(sockfd,buf,BUFFER_SIZE - 1,0);
if(ret == 0)
{
close(socket);
printf("foreiner close the connect\n");
break;
}
else if(ret < 0)
{
if(errno == EAGAIN)
{
reset_oneshot(epollfd,socket);
printf("all revc and reset socket");
break;
}
}
else
{
printf("the msg is %s\n",buf);
sleep(5); // 模拟处理过程
}
printf("end thread receiving data on fd: %d\n",socket);
}
}
int main()
{
const char* ip = "";
int port = 80;
struct sockaddr_in address;
bzero(&address,sizeof address);
address.sin_family = AF_INET;
address.sin_port = htons(port);
inet_pton(AF_INET,ip,&address.sin_addr);
int listenfd = socket(PF_INET,SOCK_STREAM,0);
assert(listenfd >= 0);
int ret = bind(listenfd,(struct sockaddr *)&address,sizeof address);
assert(ret != -1);
ret = listen(listenfd,5);
assert(ret != -1);
struct epoll_event events[MAX_EVENT_NUMBER];
int epollfd = epoll_create(5);
addfd(epollfd,listenfd,false);
while(1)
{
int ret = epoll_wait(epollfd,events,MAX_EVENT_NUMBER,-1);
if(ret < 0)
{
printf("epoll failed");break;
}
for(int i = 0; i < ret ; i++)
{
int sockfd = events[i].data.fd;
if(sockfd == listenfd)
{
struct sockaddr_in client;
socklen_t len = sizeof(client);
int t = accept(sockfd,(struct sockaddr*)&client,&len);
addfd(epollfd,t,true);
}
else if(events[i].events & EPOLLIN)
{
pthread_t thread;
fds fds_for_new_work;
fds_for_new_work.epollfd = epollfd;
fds_for_new_work.socket = events[i].data.fd;
pthread_create(&thread,NULL,work,(void *)&fds_for_new_work); // 传递给工作线程的函数
}
else
{
printf("someting else happen\n");
}
}
}
close(listenfd);
return 0;
}
利用I/O复用 可以实现的高级功能有:
1.非阻塞的connect
同时发出多个链接 。用select监听有无读写数据
2.聊天室程序
多个用户同时在线进行聊天 poll I/O复用实现多用户在线聊天
客户端程序:
3.同时处理TCP和UDP服务
本文地址:https://blog.csdn.net/qq_41468712/article/details/107254109