socket并发解决方案 java

相关链接：TCP连接与释放、网络编程——C++实现socket通信(TCP)

相关函数：

服务端：
socket()
bind()
listen()
poll()  高并发poll模式
accept()
read() 或 recv()等
write() 或 send()等
close()

客户端：
socket()
connect()
write() 或 send()等
read() 或 recv()等
close()

着重说明下poll函数用法。

跟select功能类似，可以设置的同时监听上限会更多,poll效率更高,调用完poll函数之后不会清空监听的事件集合.
int poll(struct pollfd *fds, nfds_t nfds, int timeout);		
	-fds: 是一个struct pollfd结构类型的数组，用于存放需要检测其状态的socket描述符。结构体类型定义如下：
	struct pollfd {
		int fd;	//文件描述符
		short events;	//等待的需要监听的事件类型,常用取值为POLLIN(监听读)/POLLOUT(写)/POLLERR(异常)。如fds[0].events = POLLIN
		short revents;	//实际发生了的事件，也就是返回结果。值的范围同events: POLLIN/POLLOUT/POLLERR
	};

	-nfds: nfds_t类型的参数，用于标记数组fds中的结构体元素的总数量；
	-timeout: 是poll函数调用阻塞的时间，单位：毫秒。传值-1表示阻塞监听，0表示不阻塞立即返回，>0表示阻塞等待timeout的时间

返回值：>0：数组fds中准备好读、写或出错状态的那些socket描述符的总数量，fds数组中有状态的fd的revents被赋值传出，可以通过跟POLLIN/POLLOUT/POLLERR等标志通过位与&来判断，如if(fds[n].revents & POLLIN)
	=0:数组fds中没有任何socket描述符准备好读、写，或出错，revents会被清空
	=-1：poll函数调用失败，同时会自动设置全局变量errno.

注意：每当服务端连接断开后，进入TIME_WAIT状态，等待2msl时间之后才能重新使用IP和端口，否则在bind时就会报错。要解决这个问题可以在程序开始时调用端口复用函数setsockopt。原型如下：

//int setsockopt(int sockfd, int level, int optname,const void *optval, socklen_t optlen);
    /*　sockfd：标识一个套接口的描述字。
　　    level：选项定义的层次；支持SOL_SOCKET、IPPROTO_TCP、IPPROTO_IP和IPPROTO_IPV6。
　　    optname：需设置的选项。
　　    optval：指针，指向存放选项值的缓冲区
　　    optlen：optval缓冲区长度。
　　    返回值：  成功返回0，失败返回 -1.  */
　　    

实际调用：
 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));

废话不多说，上源码！

实现的功能：客户端C向服务端S发送一串字符数据，S端会对字符串做转大写操作然后回发给C端。直接在咱们Tcp_Server.cpp基础上修改代码

服务端Poll_Server.cpp

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <netinet/in.h>
#include <ctype.h>
#include <poll.h>	//poll头文件

#define MAXSIZE 1024
#define IP_ADDR "127.0.0.1"
#define IP_PORT 8888

int main()
{
	int i_listenfd, i_connfd;
	struct sockaddr_in st_sersock;
	char msg[MAXSIZE];
	int nrecvSize = 0;
	
	int index = 0;	//记录fd数组中最大fd对应的下标
	struct pollfd pofds[MAXSIZE];	//结构体数组

	for(n : pofds)	//将所有数组中的fd设为-1，方便以后填充
	{
		n.fd = -1;
	}

	if((i_listenfd = socket(AF_INET, SOCK_STREAM, 0) ) < 0)	//建立socket套接字
	{
		printf("socket Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	memset(&st_sersock, 0, sizeof(st_sersock));
	st_sersock.sin_family = AF_INET;  //IPv4协议
	st_sersock.sin_addr.s_addr = htonl(INADDR_ANY);	//INADDR_ANY转换过来就是0.0.0.0，泛指本机的意思，也就是表示本机的所有IP，因为有些机子不止一块网卡，多网卡的情况下，这个就表示所有网卡ip地址的意思。
	st_sersock.sin_port = htons(IP_PORT);

	if(bind(i_listenfd,(struct sockaddr*)&st_sersock, sizeof(st_sersock)) < 0) //将套接字绑定IP和端口用于监听
	{
		printf("bind Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	if(listen(i_listenfd, 20) < 0)	//设定可同时排队的客户端最大连接个数
	{
		printf("listen Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}
	printf("listen fd: %d\n", i_listenfd);

	pofds[index].fd = i_listenfd;	//先赋值
	pofds[index].events = POLLIN;
	printf("======waiting for client's request======\n");
	//准备接受客户端连接
	while(1)
	{
		int nCount = poll(pofds, index+1, -1);	//阻塞监听
		printf("----------poll监听到可读事件计数:%d\n",nCount);

		for(int i = 0; i < MAXSIZE; i++)
		{
			if(nCount == 0)
			{
				break;
			}
			if(!(pofds[i].revents & POLLIN))
			{
				continue;	//不在监听事件中则跳过
			}
			printf("----------即将处理监听到的 pofds[%d]: %d\n", i, pofds[i].fd);
			nCount--;	//每处理一次就自减1
			if(pofds[i].fd == i_listenfd)	//监听到有客户端连接
			{
				if((i_connfd = accept(i_listenfd, (struct sockaddr*)NULL, NULL)) < 0)	//阻塞等待客户端连接
				{
					printf("accept Error: %s (errno: %d)\n", strerror(errno), errno);
				//	continue;
				}	
				else
				{
					printf("Client[%d], welcome!\n", i_connfd);
				}

				for(int n = 0; n < MAXSIZE; n++)
				{
					if(pofds[n].fd == -1)	//将新客户端fd加入数组中
					{
						pofds[n].fd = i_connfd;
						pofds[n].events = POLLIN;
						index < n ? index = n : true ;
						printf("将新客户端fd加入数组中. fd:%d, index:%d\n", pofds[n].fd, index);
						break; 
					}
				}
			
			}
			else	//监听到已连接的客户端发来的数据
			{
				//接受客户端发来的消息并作处理(小写转大写)后回写给客户端
				memset(msg, 0 ,sizeof(msg));
				if((nrecvSize = read(pofds[i].fd, msg, MAXSIZE)) < 0)
				{
					printf("accept Error: %s (errno: %d)\n", strerror(errno), errno);
					continue;
				}
				else if( nrecvSize == 0)	//read返回0代表对方已close断开连接。
				{
					printf("client has disconnected!\n");
					if(index == i)	//如果是最大的下标的客户端退出，则index-1
					{
						index--;
					}
					pofds[i].fd = -1;	//清除数组中相应位置
					close(pofds[i].fd);  
					
					continue;
				}
				else
				{
					printf("recvMsg:%s", msg);
					for(int i=0; msg[i] != '\0'; i++)
					{
						msg[i] = toupper(msg[i]);
					}
					if(write(pofds[i].fd, msg, strlen(msg)+1) < 0)
					{
						printf("accept Error: %s (errno: %d)\n", strerror(errno), errno);
					}

				}
			}
		}
	}//while
	close(i_listenfd);

	return 0;
}

客户端Poll_Client.cpp (直接用咱们Tcp_Client.cpp就可以)

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

#define MAXSIZE 1024
#define IP_ADDR "127.0.0.1"
#define IP_PORT 8888

int i_sockfd = -1;

void SigCatch(int sigNum)	//信号捕捉函数(捕获Ctrl+C)
{
	if(i_sockfd != -1)
	{
		close(i_sockfd);
	}
	printf("Bye~! Will Exit...\n");
	exit(0);
}

int main()
{
	struct sockaddr_in st_clnsock;
	char msg[1024];
	int nrecvSize = 0;

	signal(SIGINT, SigCatch);	//注册信号捕获函数

	if((i_sockfd = socket(AF_INET, SOCK_STREAM, 0) ) < 0)	//建立套接字
	{
		printf("socket Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	memset(&st_clnsock, 0, sizeof(st_clnsock));
	st_clnsock.sin_family = AF_INET;  //IPv4协议
	//IP地址转换(直接可以从物理字节序的点分十进制 转换成网络字节序)
	if(inet_pton(AF_INET, IP_ADDR, &st_clnsock.sin_addr) <= 0)
	{
		printf("inet_pton Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}
	st_clnsock.sin_port = htons(IP_PORT);	//端口转换(物理字节序到网络字节序)

	if(connect(i_sockfd, (struct sockaddr*)&st_clnsock, sizeof(st_clnsock)) < 0)	//主动向设置的IP和端口号的服务端发出连接
	{
		printf("connect Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	printf("======connect to server, sent data======\n");

	while(1)	//循环输入，向服务端发送数据并接受服务端返回的数据
	{
		fgets(msg, MAXSIZE, stdin);
		printf("will send: %s", msg);
		if(write(i_sockfd, msg, MAXSIZE) < 0)	//发送数据
		{
			printf("write Error: %s (errno: %d)\n", strerror(errno), errno);
			exit(0);
		}

		memset(msg, 0, sizeof(msg));
		if((nrecvSize = read(i_sockfd, msg, MAXSIZE)) < 0)	//接受数据
		{
			printf("read Error: %s (errno: %d)\n", strerror(errno), errno);
		}
		else if(nrecvSize == 0)
		{
			printf("Service Close!\n");
		}
		else
		{
			printf("Server return: %s\n", msg);
		}

	}
	return 0;
}