TCP并发服务器模型（2）

四、IO复用循环服务器

将系统的处理能力集中在核心业务上，降低并发处理单元的数量。

建立两个线程：一个处理客户端的连接请求 thread_connect，一个处理客户端的业务处理请求 thread_handle。

accept 成功后，thread_connect 将accept 返回的 fd 放入客户端连接状态表，此表与 thread_handle 共享。

thread_handle 建立 fd 集合，select 对 fd 集合进行超时等待；当业务请求到来，thread_handle 进行处理；处理完毕，客户端退出的时候，将 fd 从客户端连接状态表中取出。

代码实现：

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <time.h>
#include <string.h>
#include <stdio.h>
#include <pthread.h>
 #include <sys/select.h>
#define BUFFLEN 1024
#define SERVER_PORT 8888
#define BACKLOG 5
#define CLIENTNUM 1024/*最大支持客户端数量*/

/*可连接客户端的文件描述符数组*/
int connect_host[CLIENTNUM];
int connect_number =  0;
static void *handle_request(void *argv)
{	
	time_t now;		/*时间*/
	char buff[BUFFLEN];/*收发数据缓冲区*/
	int n = 0;
	
	int maxfd = -1;/*最大侦听文件描述符*/
	fd_set scanfd;	/*侦听描述符集合*/
	struct   timeval   timeout; 	/*超时*/
	timeout.tv_sec     =   1; /*   阻塞1秒后超时返回   */     
	timeout.tv_usec   =   0;     
    
	int i = 0;
	int err  = -1;
	for(;;)
	{   
		/*最大文件描述符值初始化为-1*/		
		maxfd = -1;
		FD_ZERO(&scanfd);/*清零文件描述符集合*/
		for(i=0;i<CLIENTNUM;i++)/*将文件描述符放入集合*/
		{
			if(connect_host[i] != -1)/*合法的文件描述符*/
			{
				FD_SET(connect_host[i], &scanfd);/*放入集合*/
				if(maxfd < 	connect_host[i])/*更新最大文件描述符值*/
				{
					maxfd = connect_host[i];
				}
			}
		}
		/*select等待*/
		err = select(maxfd + 1, &scanfd, NULL, NULL, &timeout) ;		
		switch(err)
		{
			case 0:/*超时*/
				break;
			case -1:/*错误发生*/
				break;
			default:/*有可读套接字文件描述符*/
				if(connect_number<=0)
					break;
				for(i = 0;i<CLIENTNUM;i++)
				{
					/*查找激活的文件描述符*/
					if(connect_host[i] != -1)
					if(FD_ISSET(connect_host[i],&scanfd))   
					{  
						memset(buff, 0, BUFFLEN);/*清零*/
						n = recv(connect_host[i], buff, BUFFLEN,0);/*接收发送方数据*/
						if(n > 0 && !strncmp(buff, "TIME", 4))/*判断是否合法接收数据*/
						{
							memset(buff, 0, BUFFLEN);/*清零*/
							now = time(NULL);/*当前时间*/
							sprintf(buff, "%24s\r\n",ctime(&now));/*将时间拷贝入缓冲区*/
							send(connect_host[i], buff, strlen(buff),0);/*发送数据*/
						}
						/*更新文件描述符在数组中的值*/
						connect_host[i] = -1;
						connect_number --;	/*客户端计数器减1*/	
						/*关闭客户端*/
						close(connect_host[i]);							
					} 
				}
				break; 	
		}		  
	} 
	
	return NULL;
}

static void *handle_connect(void *argv)
{	
	int s_s = *((int*)argv) ;/*获得服务器侦听套接字文件描述符*/
	int s_c = -1;/*连接客户端文件描述符*/
	struct sockaddr_in from;
	int len = sizeof(from);
	/*接收客户端连接*/
	for(;;)
	{
		int i = 0;
		int s_c = accept(s_s, (struct sockaddr*)&from, &len);/*接收客户端的请求*/
		printf("a client connect, from:%s\n",inet_ntoa(from.sin_addr));
		/*查找合适位置，将客户端的文件描述符放入*/				
		for(i=0;i<CLIENTNUM;i++)
		{
			if(connect_host[i] == -1)/*找到*/
			{
				/*放入*/
				connect_host[i]= s_c;
				
				/*客户端计数器加1*/
				connect_number ++;
				/*继续轮询等待客户端连接*/
				break;						
			}	
		}		
	}	
	return NULL;
}

int main(int argc, char *argv[])
{
	int s_s;	/*服务器套接字文件描述符*/
	struct sockaddr_in local;	/*本地地址*/	
	int i = 0;
	memset(connect_host, -1, CLIENTNUM);
	
	/*建立TCP套接字*/
	s_s = socket(AF_INET, SOCK_STREAM, 0);
	
	/*初始化地址接哦股*/
	memset(&local, 0, sizeof(local));/*清零*/
	local.sin_family = AF_INET;/*AF_INET协议族*/
	local.sin_addr.s_addr = htonl(INADDR_ANY);/*任意本地地址*/
	local.sin_port = htons(SERVER_PORT);/*服务器端口*/
	
	/*将套接字文件描述符绑定到本地地址和端口*/
	int err = bind(s_s, (struct sockaddr*)&local, sizeof(local));
	err = listen(s_s, BACKLOG);/*侦听*/
	
	pthread_t  thread_do[2];/*线程ID*/
	/*创建线程处理客户端连接*/
	pthread_create(&thread_do[0],/*线程ID*/
					NULL,/*属性*/
					handle_connect,/*线程回调函数*/
					(void*)&s_s);		/*线程参数*/
	/*创建线程处理客户端请求*/					
	pthread_create(&thread_do[1],/*线程ID*/
					NULL,/*属性*/
					handle_request,/*线程回调函数*/
					NULL);		/*线程参数*/
	/*等待线程结束*/
	for(i=0;i<2;i++)
		pthread_join(thread_do[i], NULL);
	
	close(s_s);
	
	return 0;		
}