前面两篇介绍用进程方式和线程方式实现并发服务,其实里面调用send,read,accept函数都会导致阻塞,而linux的select函数可以使我们在程序中同时监听多个文件描述符的读写状态。程序会停在select这里等待,直到被监视的文件描述符有某一个或多个发生了状态改变。select()的机制中提供一fd_set的数据结构,实际上是一long类型的数组, 每一个数组元素都能与一打开的文件描述符(不管是Socket描述符,还是其他 文件或命名管道或设备描述符)建立联系, 当调用select()时,由内核根据IO状态修改fd_set的内容,由此来通知执行了select()的进程哪一Socket或文件可读。
下面我们用select函数实现单进程单线程的并发服务,客户端代码同多进程版本,服务器端代码如下
- #include<stdio.h>
- #include<stdlib.h>
- #include<string.h>
- #include<sys/types.h>
- #include<sys/socket.h>
- #include<netinet/in.h>
- #include<arpa/inet.h>
- #define MAX_LISTEN 5
- #define PORT 1987
- #define IP "127.0.0.1"
- int main()
- {
- int conn_fd;
- int sock_fd = socket(AF_INET,SOCK_STREAM,0);
- if (sock_fd < 0) {
- perror("create socket failed");
- exit(1);
- }
- struct sockaddr_in addr_client;
- int client_size = sizeof(struct sockaddr_in);
- struct sockaddr_in addr_serv;
- memset(&addr_serv, 0, sizeof(addr_serv));
- addr_serv.sin_family = AF_INET;
- addr_serv.sin_port = htons(PORT);
- addr_serv.sin_addr.s_addr = inet_addr(IP);
- if (bind(sock_fd,(struct sockaddr *)&addr_serv,sizeof(struct sockaddr_in)) < 0) {
- perror("bind error");
- exit(1);
- }
- if (listen(sock_fd,MAX_LISTEN) < 0) {
- perror("listen failed");
- exit(1);
- }
- int recv_num;
- int send_num;
- char recv_buf[100];
- char send_buf[100];
- //用一个数组记录描述符的状态
- int i, ready, max_fd;
- int client[FD_SETSIZE];
- for (i = 0;i < FD_SETSIZE;i ++) {
- client[i] = -1;
- }
- fd_set readset;
- max_fd = sock_fd;
- //最大可用描述符的个数,一般受操作系统内核的设置影响,我的环境下这个值是1024
- printf("max fd num %d\n",FD_SETSIZE);
- while (1) {
- //重置监听的描述符
- FD_ZERO(&readset);
- FD_SET(sock_fd,&readset);
- for (i = 0;i < FD_SETSIZE;i ++) {
- if (client[i] == 1) {
- FD_SET(i, &readset);
- }
- }
- //开始监听描述符,是异步的,不会阻塞
- ready = select(max_fd+1, &readset, NULL, NULL, NULL);
- //可用描述符如果是创建连接描述符,则创建一个新的连接
- if (FD_ISSET(sock_fd, &readset)) {
- conn_fd = accept(sock_fd, (struct sockaddr *)&addr_client, &client_size);
- if (conn_fd < 0) {
- perror("accept failed");
- exit(1);
- }
- FD_SET(conn_fd, &readset);
- FD_CLR(sock_fd, &readset);
- if (conn_fd > max_fd) {
- max_fd = conn_fd;
- }
- client[conn_fd] = 1;
- }
- //检查所有的描述符,查看可读的是哪个,针对它进行IO读写
- for (i = 0; i < FD_SETSIZE; i ++) {
- if (FD_ISSET(i, &readset)) {
- recv_num = recv(i, recv_buf, sizeof(recv_buf), 0);
- if (recv_num <= 0) {
- FD_CLR(i, &readset);
- client[i] = -1;
- }
- recv_buf[recv_num] = '\0';
- memset(send_buf,0,sizeof(send_buf));
- sprintf(send_buf, "server proc got %d bytes\n", recv_num);
- send_num = send(i, send_buf, strlen(send_buf), 0);
- if (send_num <= 0) {
- FD_CLR(i, &readset);
- client[i] = -1;
- }
- }
- }
- }
- close(sock_fd);
- return 0;
- }
1,每次select之前我都重新设置了一遍需要监听的描述符,要是能把这个提取到while循环外就好了(不过放出去程序就会错误,未查清原因)
2,我没有引入可写的fd_set,其实应该将文件描述符的可写状态也作为监听变量传给select函数,不过上面的示例代码可用即可。