目录
I/O复用技术
select函数
设置文件描述符
指定监视范围
设置超时
I/O复用服务器端的实现
由服务器创建多个进程来实现并发的做法有时会带来一些问题,比如:内存上的开销、CPU的大量占用等,这些因素会消耗掉服务器端有限的计算资源、进而影响程序之间的执行效率。那么,有没有方法可以在不创建额外进程的条件下实现并发呢?当然有,那就是I/O复用技术。
I/O复用技术
I/O复用指的是通过单个线程记录一个或多个I/O流的状态,并对不同状态下的I/O流进行协调,使进程不阻塞于某个特定的I/O调用过程中,从而将有限资源最大化利用。
引用自一段情景材料,方便大家能更好地理解这个技术背后的思想:
假设你是一个机场的空管,你需要管理到你机场的所有的航线,包括进港,出港,有些航班需要放到停机坪等待,有些航班需要去登机口接乘客。
你会怎么做?
最简单的做法,就是你去招一大批空管员,然后每人盯一架飞机,从进港,接客,排位,出港,航线监控,直至交接给下一个空港,全程监控。
那么问题就来了:
- 很快你就发现空管塔里面聚集起来一大票的空管员,交通稍微繁忙一点,新的空管员就已经挤不进来了。
- 空管员之间需要协调,屋子里面就1, 2个人的时候还好,几十号人以后,基本上就成菜市场了。
- 空管员经常需要更新一些公用的东西,比如起飞显示屏,比如下一个小时后的出港排期,最后你会很惊奇的发现,每个人的时间最后都花在了抢这些资源上。
现实上我们的空管同时管几十架飞机稀松平常的事情, 他们怎么做的呢?他们用这个东西:
这个东西叫 flight progress strip,每一个块代表一个航班,不同的槽代表不同的状态,然后一个空管员可以管理一组这样的块(一组航班),而他的工作,就是在航班信息有新的更新的时候,把对应的块放到不同的槽子里面。这个东西现在还没有淘汰哦,只是变成电子的了而已。是不是觉得一下子效率高了很多,一个空管塔里可以调度的航线可以是前一种方法的几倍到几十倍。
如果你把每一个航线当成一个 Sock(I/O 流),空管当成你的服务端 Sock 管理代码的话:
- 第一种方法就是最传统的多进程并发模型:每进来一个新的 I/O 流会分配一个新的进程管理。
- 第二种方法就是 I/O 多路复用:单个线程,通过记录跟踪每个 I/O 流(sock)的状态,来同时管理多个 I/O 流 。
参考资料:IO 多路复用是什么意思? - 罗志宇
select函数
select是I/O复用技术中比较经典且常用到的一个函数(还有poll、epoll),在使用上,需要引入<sys/select.h>和<sys/time.h>两个头文件。
#include <sys/select.h>
#include <sys/time.h>
int select(int maxfd , fd_set * readset , fd_set * writeset , fd_set * exceptset , const struct timeval * timeout);
//成功时返回大于0的值,失败时返回-1。其余情况为返回发生事件(监视项)的文件描述符数量
/* 监视项 */
// 1.接收数据的套接字
// 2.传输数据(无阻塞)的套接字
// 3.发生异常的套接字
/* 参数含义 */
// maxfd: 监视对象文件描述符数量
// readset: 将所有关注"是否存在待读取数据"的文件描述符注册到fd_set型变量,并传递其地址值。
// writeset: 将所有关注"是否可传输无阻塞数据"的文件描述符注册到fd_set型变量,并传递其地址值。
// exceptset: 将所有关注"是否发生异常"的文件描述符注册至fd_set型变量,并传递其地址值。
// timeout: 调用 select 函数后,为防止陷入无限阻塞的状态,传递超时(time-out)消息。
使用select函数时,一般遵循以下流程步骤:
设置文件描述符
在调用select函数之前,我们需要声明监视事件,并用数据类型为fd_set的变量来记录监视事件下的每个文件描述符的状态。如图所示,fd_set以数组的形式记录每个文件描述符的状态:
以图中fd0、fd2为例,值为0,代表着所指向的文件描述符不是被监视对象;反之,fd1和fd3值为1,则是被监视对象。
在对fd_set数组操作时,有些宏可以帮助我们简化代码的编写工作,下列所示为与对fd_set操作相关的宏:
- FD_ZERO(fd_set * fdset): 将fd_set变量的所有位初始化为0 。
- FD_SET(int fd , fd_set * fdset): 在参数fdset指向的变量中注册文件描述符fd的信息。
- FD_CLR(int fd , fd_set * fdset): 从参数fdset指向的变量中清除文件描述符fd的信息。
- FD_ISSET(int fd , fd_set * fdset): 若参数fdset指向的变量中包含文件描述符fd的信息,则返回"真",用于对select调用结果的验证。
宏对应的操作含义如图所示:
指定监视范围
即设置select函数中的第一个参数 maxfd,其值用来标记限制对监视事件中的文件描述符最大监视数。
设置超时
当监视的文件描述符未发生变化时,select函数会导致进程发生阻塞。为了避免这种情况发生,我们可以通过设置超时(select函数中的最后一个参数timeout)来防止这种情况的发生。
timeout的结构体定义如下:
struct timeval
{
long tv_sec; // 秒
long tv_usec; // 毫秒
}
//timeval.tv_sec=5,timeval.tv_usec=500; 代表设置超时等待周期为5秒500毫秒
I/O复用服务器端的实现
io_echoserver.cpp
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/select.h>
#define BUF_SIZE 1024
void Sender_error(char *message);
int main(int argc, char *argv[])
{
int port, serv_sock, clnt_sock;
struct sockaddr_in serv_adr, clnt_adr;
struct timeval timeout;
fd_set reads, cpy_reads;
socklen_t adr_sz;
int fd_max, str_len, fd_num, i;
char buf[BUF_SIZE];
//可以改为直接传参(即使用argv数组),这里主要方便验证和展示
printf("Please input the port of socket that you want to create:\n");
scanf("%d", &port);
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
if (serv_sock == -1)
{
Sender_error((char *)"Sock creation error");
}
else
{
// 注意:serv_sock初始化成功后值为0
fd_max = serv_sock;
}
memset(&serv_adr, 0, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_adr.sin_port = htons(port);
if (bind(serv_sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
{
Sender_error((char *)"Bind() error");
}
if (listen(serv_sock, 5) == -1)
{
Sender_error((char *)"Listen error");
}
// 对监测项的文件描述符作初始化赋0操作
FD_ZERO(&reads);
// 注册serv_sock套接字信息至reads变量中
FD_SET(serv_sock, &reads);
while (1)
{
// cpy_reads用来记录文件描述符变化
cpy_reads = reads;
// 设置超时等待周期为5s
timeout.tv_sec = 5;
timeout.tv_usec = 0;
// 只关注接收数据的套接字,不对传输数据、出现异常的套接字进行监视
// fd_num用来记录发生监视事件的文件描述符数量
if ((fd_num = select(fd_max + 1, &cpy_reads, 0, 0, &timeout)) == -1)
{
break;
}
if (fd_num == 0)
{
continue;
}
for (i = 0; i < fd_max + 1; i++)
{
// 判断cpy_reads中是否含有文件描述符i的信息
if (FD_ISSET(i, &cpy_reads))
{
// 若当前只有服务器端套接字,则尝试接收客户端请求
if (i == serv_sock)
{
adr_sz = sizeof(clnt_adr);
clnt_sock = accept(serv_sock, (struct sockaddr *)&clnt_adr, &adr_sz);
// 将客户端的套接字文件描述符信息注册至reads中
FD_SET(clnt_sock, &reads);
if (fd_max < clnt_sock)
{
// 增加监视数上限(因为有新的客户端套接字加入)
fd_max = clnt_sock;
}
printf("Connected client: %d \n", clnt_sock);
}
else
{
// 接收数据
str_len = read(i, buf, BUF_SIZE);
// 无数据,则关闭对应套接字
if (str_len == 0)
{
// 清除reads变量中文件描述符i的信息
FD_CLR(i, &reads);
close(i);
printf("Closed client: %d \n", i);
}
else
{
write(i, buf, str_len);
}
}
}
}
}
close(serv_sock);
return 0;
}
void Sender_error(char *message)
{
puts(message);
exit(1);
}
运行结果:文章来源:https://www.toymoban.com/news/detail-483139.html
文章来源地址https://www.toymoban.com/news/detail-483139.html
到了这里,关于【TCP/IP】利用I/O复用技术实现并发服务器 - select函数的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!