linux中处理非阻塞 Socket 的常见方法

在 Linux 客户端中，使用非阻塞的 Socket 可以提高程序的响应性，允许在发送或接收数据时继续执行其他任务。非阻塞 Socket 通常需要使用异步 I/O 操作或轮询机制来实现。以下是一些处理非阻塞 Socket 的常见方法：

1. 使用 select 函数

select 函数允许你监视多个文件描述符，判断它们是否处于可读、可写或异常状态，从而进行相应的处理。

#include <sys/select.h> fd_set read_fds, write_fds, except_fds; struct timeval timeout; // 初始化文件描述符集 FD_ZERO(&read_fds); FD_ZERO(&write_fds); FD_ZERO(&except_fds); // 将需要监视的文件描述符加入集合 FD_SET(socket_fd, &read_fds); FD_SET(socket_fd, &write_fds); FD_SET(socket_fd, &except_fds); // 设置超时时间 timeout.tv_sec = 5; timeout.tv_usec = 0; // 使用 select 进行轮询 int result = select(socket_fd + 1, &read_fds, &write_fds, &except_fds, &timeout); if (result > 0) { // 文件描述符处于可读、可写或异常状态，进行相应处理 } else if (result == 0) { // 超时，无事件发生 } else { // 出现错误 }

2. 使用 poll 函数

poll 函数与 select 类似，但更灵活，允许监视大量的文件描述符。

#include <poll.h> struct pollfd fds[1]; int timeout = 5000; // 超时时间，单位为毫秒 // 设置需要监视的文件描述符 fds[0].fd = socket_fd; fds[0].events = POLLIN | POLLOUT | POLLERR; // 使用 poll 进行轮询 int result = poll(fds, 1, timeout); if (result > 0) { // 文件描述符处于可读、可写或异常状态，进行相应处理 } else if (result == 0) { // 超时，无事件发生 } else { // 出现错误 }

3. 使用异步 I/O

Linux 提供了异步 I/O 操作，可以通过 aio_read 和 aio_write 等函数来进行异步读写操作。

#include <aio.h> struct aiocb aio; // 初始化 aiocb 结构 bzero(&aio, sizeof(struct aiocb)); aio.aio_fildes = socket_fd; aio.aio_buf = buffer; aio.aio_nbytes = buffer_size; // 发起异步读操作 if (aio_read(&aio) == -1) { // 处理错误 } // 等待异步操作完成 while (aio_error(&aio) == EINPROGRESS); // 检查异步读的结果 ssize_t bytesRead = aio_return(&aio);

4. 使用 fcntl 设置非阻塞模式

通过 fcntl 函数可以设置文件描述符的属性，包括将其设置为非阻塞模式。

#include <fcntl.h> int flags = fcntl(socket_fd, F_GETFL, 0); fcntl(socket_fd, F_SETFL, flags | O_NONBLOCK);

这种方法适用于需要在整个程序中使用非阻塞 Socket 的情况。

以上方法可以单独使用，也可以结合使用，具体选择取决于程序的需求。在使用非阻塞 Socket 时，需要仔细处理返回的状态以及错误条件，确保程序正确地处理数据和异常情况。