poll() - IO 多路复用，告别傻乎乎轮询！

2024-01-30 17:17:42

IO 多路复用，让网络应用飞起来

在网络编程中，我们经常需要处理大量客户端连接，比如 web 服务器、聊天室、游戏服务器等。传统的处理方式是采用轮询，即不断检查每个 socket 是否有数据可读。这种方式简单易懂，但效率低下，尤其是当需要处理大量连接时。

IO 多路复用是一种更高效的处理方式。它可以同时监听多个 socket，并且只处理已经准备好数据的 socket。这大大减少了系统开销，提高了网络应用的性能和可扩展性。

poll()，IO 多路复用的利器

poll() 函数是 Linux 系统中常用的 IO 多路复用函数之一。它可以同时监听多个 socket，并且将准备好数据的 socket 放入一个就绪列表中。应用程序只需遍历这个就绪列表，即可处理所有已经准备好数据的 socket。

poll() 函数的原型如下：

#include <poll.h>

int poll(struct pollfd *fds, nfds_t nfds, int timeout);

其中：

fds：指向一个 pollfd 结构数组的指针，其中每个元素代表一个要监听的 socket。
nfds：fds 数组的长度，即要监听的 socket 的数量。
timeout：超时时间，单位是毫秒。如果超时时间为 0，则 poll() 函数将一直阻塞，直到有数据可读。

poll() 函数的使用示例

以下是一个使用 poll() 函数的示例：

#include <poll.h>

int main() {
    // 创建一个 pollfd 数组，其中每个元素代表一个要监听的 socket
    struct pollfd fds[2];
    fds[0].fd = socket(AF_INET, SOCK_STREAM, 0);
    fds[1].fd = socket(AF_INET, SOCK_STREAM, 0);

    // 设置 pollfd 数组的事件类型
    fds[0].events = POLLIN;
    fds[1].events = POLLIN;

    // 使用 poll() 函数监听 socket
    int ret = poll(fds, 2, -1);

    // 检查 poll() 函数的返回值
    if (ret < 0) {
        // 发生错误
        perror("poll");
        exit(1);
    } else if (ret == 0) {
        // 超时
        printf("Timeout\n");
    } else {
        // 有数据可读
        for (int i = 0; i < 2; i++) {
            if (fds[i].revents & POLLIN) {
                // socket i 有数据可读
                // 处理数据
            }
        }
    }

    return 0;
}