IO复用（Reactor模式和Preactor模式）——用epoll来提高服务器并发能力

#include <iostream>
#include <sys/socket.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>  
#include <stdlib.h>  
#include <string.h>  
#include <errno.h>

using namespace std;

#define MAXLINE 5
#define OPEN_MAX 100
#define LISTENQ 20
#define SERV_PORT 5000
#define INFTIM 1000

int main(int argc, char* argv[])
{
    int listen_fd, connfd_fd, socket_fd, epfd, nfds;
    ssize_t n;
    char line[MAXLINE];
    socklen_t clilen;

    //声明epoll_event结构体的变量,ev用于注册事件,数组用于回传要处理的事件
    struct epoll_event ev,events[20];
    //生成用于处理accept的epoll专用的文件描述符
    epfd=epoll_create(5);
    struct sockaddr_in clientaddr;
    struct sockaddr_in serveraddr;
    listen_fd = socket(AF_INET, SOCK_STREAM, 0);
    //设置与要处理的事件相关的文件描述符
    ev.data.fd = listen_fd;
    //设置要处理的事件类型
    ev.events=EPOLLIN|EPOLLET;
    //注册epoll事件
    epoll_ctl(epfd,EPOLL_CTL_ADD,listen_fd,&ev);
    
    memset(&serveraddr, 0, sizeof(serveraddr));  
    serveraddr.sin_family = AF_INET;  
    serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
    serveraddr.sin_port = htons(SERV_PORT);
    
    if (bind(listen_fd, (struct sockaddr*)&serveraddr, sizeof(serveraddr)) == -1)
    {  
        printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);  
        exit(0);  
    }  
    
    if (listen(listen_fd, LISTENQ) == -1)
    {  
        exit(0);  
    }   
    
    for ( ; ; ) 
    {
        //等待epoll事件的发生
        nfds = epoll_wait(epfd,events,20,500);
        //处理所发生的所有事件
        for (int i = 0; i < nfds; ++i)
        {
            if (events[i].data.fd == listen_fd)//如果新监测到一个SOCKET用户连接到了绑定的SOCKET端口，建立新的连接。

            {
                connfd_fd = accept(listen_fd,(sockaddr *)&clientaddr, &clilen);
                if (connfd_fd < 0){
                    perror("connfd_fd < 0");
                    exit(1);
                }
                char *str = inet_ntoa(clientaddr.sin_addr);
                cout << "accapt a connection from " << str << endl;
                //设置用于读操作的文件描述符
                ev.data.fd = connfd_fd;
                //设置用于注测的读操作事件
                ev.events = EPOLLIN|EPOLLET;
                //注册ev
                epoll_ctl(epfd,EPOLL_CTL_ADD,connfd_fd,&ev);
            }
            else if (events[i].events&EPOLLIN)//如果是已经连接的用户，并且收到数据，那么进行读入。
            {            
                memset(&line,'\0', sizeof(line));  
                if ( (socket_fd = events[i].data.fd) < 0)
                    continue;
                if ( (n = read(socket_fd, line, MAXLINE)) < 0) {
                    if (errno == ECONNRESET) {
                        close(socket_fd);
                        events[i].data.fd = -1;
                    } else
                        std::cout<<"readline error"<<std::endl;
                } else if (n == 0) {
                    close(socket_fd);
                    events[i].data.fd = -1;
                }                
                cout << line << endl;
                //设置用于写操作的文件描述符
                ev.data.fd = socket_fd;
                //设置用于注测的写操作事件
                ev.events = EPOLLOUT|EPOLLET;
                //修改socket_fd上要处理的事件为EPOLLOUT
                //epoll_ctl(epfd,EPOLL_CTL_MOD,socket_fd,&ev);
            }
            else if (events[i].events&EPOLLOUT) // 如果有数据发送
            {
                socket_fd = events[i].data.fd;
                write(socket_fd, line, n);
                //设置用于读操作的文件描述符
                ev.data.fd = socket_fd;
                //设置用于注测的读操作事件
                ev.events = EPOLLIN|EPOLLET;
                //修改socket_fd上要处理的事件为EPOLIN
                epoll_ctl(epfd,EPOLL_CTL_MOD,socket_fd,&ev);
            }
        }
    }
    return 0;
}