50-使用 epoll 改写服务器

使用 epoll 改写服务器，会让我们的程序变得更加简单、高效。

1. 程序路径

代码托管在 gitos 上，请使用下面的命令获取：

git clone https://git.oschina.net/ivan_allen/unp.git

如果你已经 clone 过这个代码了，请使用 git pull 更新一下。本节程序所使用的程序路径是 unp/program/echo/multiplexing_epoll_server。

2. 回忆 epoll 系列函数

主要有三个，分别是 epoll_create、epoll_ctl、epoll_wait. 有关 epoll 的具体用法，请参考《Linux 环境编程学习笔记》第十二章内容。

#include <sys/epoll.h>

int epoll_create(int size);
int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout);

3. 使用 epoll 改写服务器

同样的，这里只贴出伪代码。

void server_routine() {
   // ...
   int epfd;
   struct epoll_event evts[5]; // 接收 epoll_wait 的返回值
   struct epoll_event ev;
   listen(listenfd);

   epfd = epoll_create(1); // 参数只要大于 0 就行了

   // 添加 listenfd 到 epoll
   ev.data.fd = listenfd;
   ev.events = EPOLLIN;
   epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &evt);

   while(1) {
     // 监听事件
     nready = epoll_wait(epfd, evts, 5， 0);
     for (i = 0; i < nready, ++i) {
       // 查看是否有新的连接
       if (evts[i].data.fd == listenfd && (evts[i].events & EPOLLIN)) {
         sockfd = accept(listenfd);
         ev.data.fd = sockfd;
         ev.events = EPOLLIN;
         // epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &evt);
         // 此行原来有误，应该是添加 sockfd 而非 listenfd.感谢网友 Neymar 指出，修改如下:
         epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &evt);
       }
       // 与客户端交互数据
       else if (evts[i].events & EPOLLIN) {
         ret = doServer(evts[i].data.fd);
         if (ret <= 0) {
           // 客户端关闭，则将其从 epoll 中删除
           epoll_ctl(epfd, EPOLL_CTL_DEL, evts[i].data.fd, NULL);
           close(evts[i].data.fd);
         }
       }
     }
   }
}

4. 程序运行结果

图1 运行结果

5. 总结

• 掌握 epoll 在网络编程中的应用