（5）基于多路复用的服务端

基于多路复用的服务端

缺陷分析与改进方案

• 目前服务端的瓶颈分析
  – 服务器在处于等待连接和客户端数据的时候将会处于阻塞状态（accept和recv在调用后会阻塞当前线程，即在等待连接时无法处理下一个客户端的数据，在处理客户端数据时，无法处理客户端的连接请求），无法处理其他事件，无法发挥服务器的最大性能；
  – 代码示例：

while(1)
{
	client = accept(server,(struct sockaddr*)&caddr,&asize);//直接阻塞，等待客户端连接
	printf("client is connected\n");
	do
	{
		r = recv(client, buf, sizeof(buf), 0);//再次阻塞，等待客户端数据
		if(r > 0)
		{
			printf("Receive : %s\n",buf);
			
			if(strcmp("quit",buf) != 0){
				len = send(client, buf, r, 0);
			}else{
				break;
			}
		}
	}while(r > 0);
	close(client);
}

• 解决方案：阻塞变轮询
  – 1、监听连接事件： 通过select函数监听服务端文件描述符server_fd；
  – 2、获取client_fd： 当连接事件发生时，调用accept函数接受连接，返回得到客户端文件描述符client_fd，用于客户端数据收发；
  – 3、监听接收事件： 将client_fd加入监听范围，监听数据接收事件；
  – 4、循环查看： 轮询查看各个被监听的文件描述符是否有事件发生。

服务端优化细节刨析

• 实现逻辑：通过监视事件发生的标识位来响应事件，事件先发生，服务器再响应，这样可以减少进入阻塞的时间。

while(1)
{
	rset = reads;
	num = select(max+1, &rset, 0, 0, &timeout);
	if(num > 0)
	{
		int i = 0;
		for(i = 1; i <= max; i++)//轮询[1,max],0是输入输出，不加入轮询
		{
			if(FD_ISSET(i,&rset))//判断是不是第i号文件描述符上的事件
			{
				if(i == server)
				{
					//服务端的事件：
					//接收客户端的连接，并将客户端的fd加入监视范围
				}
				else
				{
					//客户端的事件：
					//读取客户端的数据（通过i文件描述符）
				}
			}
		}
	}
}

• 实现关键：动态调整需要监视的文件描述符
  – 当接收到客户端连接时，将客户端文件描述符加入监听变量（fd_set）中；
  – 当发现客户端断开时，在监听变量（fd_set）中剔除客户端文件描述符；

if(client > -1)
{
	FD_SET(client,&reads);//将客户端fd在reads中对应的位做标记
	max = (client > max) ? client:max;//判断新fd值有没有大于当前轮询最大值，有则替换
	printf("client:%d\n",client);
}

if(r == -1)
{
	FD_CLR(i,&reads);//不再监视该客户端fd（i）
	close(i);//释放资源
}

服务端轮询框架搭建

• 大致流程：
  – 1、先配置协议，拿到服务端fd；
  – 2、绑定服务端IP地址和端口号；
  – 3、监视文件描述符对应的事件是否发生；
  – 4、事件发生后，通过fd判断是否事件类型，服务端事件则接入新客户端，客户端事件则处理数据请求；

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>

// 处理服务端事件
int server_handler(int server)
{
	struct sockaddr_in addr = {0};
	socklen_t asize = sizeof(addr);
	
	return accept(server, (struct sockaddr*)&addr, &asize);
}

// 处理客户端事件
int client_handler(int client)
{
	char buf[32] = {0};
	int ret = read(client, buf, sizeof(buf)-1);
	
	if(ret > 0)
	{
		buf[ret] = 0;
		printf("Receive: s%\n",buf);
		
		if(strcmp(buf, "quit") != 0)
		{
			ret = write(client, buf, ret);
		}
		else
		{
			ret = -1;
		}
	}
}

int main()
{
	int server = 0;
	struct sockaddr_in saddr = {0};
	int max = 0;//监视文件描述符的范围
	int num = 0;//接收select函数的返回值
	fd_set reads = {0};//对感兴趣fd做标记
	fd_set temps = {0};
	struct timeval timeout = {0};//超时值

	server = socket(PF_INET,SOCK_STREAM,0);//server:文件描述符
	if(server == -1){
		printf("server socket error\n");
		return -1;
	}
	
	saddr.sin_family = AF_INET;
	saddr.sin_addr.s_addr = htonl(INADDR_ANY);
	saddr.sin_port = htons(8888);
	
	if(bind(server,(struct sockaddr*)&saddr,sizeof(saddr)) == -1){
		printf("server bind error\n");
		return -1;
	}
	
	if(listen(server,1) == -1){
		printf("server listen error \n");
	}
	
	printf("server start success\n");

	FD_ZERO(&reads);//清零
	FD_SET(server, &reads);//标记服务端fd对应位
	
	max = server;
	
	while(1)
	{
		temps = reads;
		timeout.tv_sec = 0;
		timeout.tv_usec = 10000;

		num = select(max+1, &temps, 0, 0, &timeout);//轮询[0,max+1),num表示有事件发生的文件描述符的数量
	
		if(num > 0)
		{
			for(int i = 1; i<=max; i++)
			{
				if(FD_ISSET(i,&temps))//找到产生事件的fd（FD_ISSET用于判断指定fd是否产生事件）
				{
					if(i == server)//判断是服务端事件还是客户端事件
					{
						//服务端事件：连接事件，处理新增客户端的接入请求
						int client = server_handler(server);
						if(client > -1)
						{
							FD_SET(client, &reads);
							max = (client > max) ? client : max;
							printf("accept client : %d\n",client);
						}
					}
					else
					{
						//客户端事件：接收事件，处理已接入的客户端的数据请求
						int r = client_handler(i);
						if(r == -1)
						{
							FD_CLR(i, &reads);
							close(i);
						}
					}
				}
			}
		}
		
	}
}