在 C 语言中使用“select”函数

简介

在 C 语言中，使用“select”函数可以实现异步的 I/O 操作。

其基本用法如下：

#include <sys/select.h>

int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);

其中，“nfds”是有效的文件描述符集合长度；

“readfds”、“writefds”、“exceptfds”分别是指向可读、可写、有异常等事件集合的指针；

“timeout”是该函数的超时时间。

当调用“select”函数时，该函数会一直阻塞，直到发生指定的事件之一。一旦事件发生，函数就会返回，此时可以使用“FD_ISSET”宏来分别判断各个事件的状态。

示例代码

下面是一个示例代码，使用“select”函数来监听多个文件描述符：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/select.h>

int main()
{
    int fd1, fd2;
    fd_set read_fds;
    int max_fd;

    fd1 = open("file1", O_RDONLY);
    fd2 = open("file2", O_RDONLY);

    if (fd1 < 0 || fd2 < 0) {
        fprintf(stderr, "Failed to open files\n");
        exit(EXIT_FAILURE);
    }

    max_fd = (fd1 > fd2) ? fd1 : fd2;

    while (1) {
        FD_ZERO(&read_fds);
        FD_SET(fd1, &read_fds);
        FD_SET(fd2, &read_fds);

        if (select(max_fd + 1, &read_fds, NULL, NULL, NULL) < 0) {
            perror("select");
            exit(EXIT_FAILURE);
        }

        if (FD_ISSET(fd1, &read_fds)) {
            printf("File1 is ready to read\n");
            // 处理文件1的读操作
        }

        if (FD_ISSET(fd2, &read_fds)) {
            printf("File2 is ready to read\n");
            // 处理文件2的读操作
        }
    }

    return 0;
}

其中，“FD_ZERO”宏用于初始化指定的事件集合；

“FD_SET”宏用于将指定的文件描述符添加到指定的事件集合中；

“select”函数用于阻塞并监听被添加到事件集合中的文件描述符，直到其中一个文件准备好读取，函数才会返回；

“FD_ISSET”宏用于判断文件描述符是否是指定事件集合中的一部分，并返回一个布尔值。

总结

使用“select”函数可以在 C 语言中实现异步的 I/O 操作。

代码片段：

#include <sys/select.h>

int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);

FD_ZERO(&read_fds);
FD_SET(fd1, &read_fds);
FD_SET(fd2, &read_fds);

if (select(max_fd + 1, &read_fds, NULL, NULL, NULL) < 0) {
    perror("select");
    exit(EXIT_FAILURE);
}

if (FD_ISSET(fd1, &read_fds)) {
    printf("File1 is ready to read\n");
    // 处理文件1的读操作
}

if (FD_ISSET(fd2, &read_fds)) {
    printf("File2 is ready to read\n");
    // 处理文件2的读操作
}