百万服务器并发代码实现

#include <stdio.h> #include <time.h> #include <string.h> #include <stdlib.h> // #include <winsock2.h> // #include <mswsock.h> // #include <windows.h> // #include <sys/types.h> // #include <unistd.h> // #include <fcntl.h> #include <netinet/tcp.h> #include <arpa/inet.h> #include <pthread.h> #include <errno.h> #include <fcntl.h> #include <unistd.h> #include <sys/epoll.h> #define BUFFER_LENGTH 1024 #define EPOLL_SIZE 1024 #define MAX_PORT 100 int islistenfd(int fd,int *fds){ int i=0; for (i=0;i<MAX_PORT;i++){ if(fd==*(fds+i)) return fd; } return 0; } int main(int argc,char *argv[]){ if (argc < 2) { printf("Param Error \n"); return -1; } int port=atoi(argv[1]);//开始的端口 int sockfds[MAX_PORT]={0}; //lisiten fd int epfd=epoll_create(1); int i=0; for(i=0;i<MAX_PORT;i++){ int sockfd=socket(AF_INET,SOCK_STREAM,0); struct sockaddr_in addr; memset(&addr,0,sizeof(struct sockaddr_in)); addr.sin_family=AF_INET; addr.sin_port=htons(port+i); //8888 ---> 8889 ----> 8890 ----> .... addr.sin_addr.s_addr=INADDR_ANY; if (bind(sockfd,(struct sockaddr*)&addr,sizeof(struct sockaddr_in))<0){ perror("bind"); return -2; } if(listen(sockfd,5)<0){ perror("listen"); return -3; } printf("tcp server listen on port : %d\n", port + i); struct epoll_event ev; ev.events=EPOLLIN; ev.data.fd = sockfd; epoll_ctl(epfd,EPOLL_CTL_ADD,sockfd,&ev); sockfds[i]=sockfd; } struct epoll_event events[EPOLL_SIZE] = {0}; //创建一个结构体数组 events 用于存储 epoll_wait() 返回的事件列表。 while (1){ int nready=epoll_wait(epfd,events,EPOLL_SIZE,5); // if (nready == -1) continue; //表示5秒内，没有事件，继续监听 int i=0; for (i=0;i<nready;i++){ int sockfd=islistenfd(events[i].data.fd ,sockfds); if(sockfd){ struct sockaddr_in client_addr; memset(&client_addr,0,sizeof(struct sockaddr_in)); socklen_t client_len =sizeof(client_addr); int clientfd=accept(sockfd,(struct sockaddr *)&client_addr,&client_len); fcntl(clientfd,F_SETFL,O_NONBLOCK); int reuse=1; setsockopt(clientfd, SOL_SOCKET, SO_REUSEADDR, (char *)&reuse, sizeof(reuse)); struct epoll_event ev; ev.events=EPOLLIN | EPOLLET; //EPOLLET 则表示将 I/O 事件设置为边缘触发模式。 ev.data.fd=clientfd; epoll_ctl(epfd,EPOLL_CTL_ADD,clientfd,&ev); } else{ //当某个客户端套接字上出现可读事件时（即该文件描述符在 events 中对应的元素有 EPOLLIN 标志），则调用 recv() 函数从该套接字中读取数据 int clientfd=events[i].data.fd; char buffer[BUFFER_LENGTH]={0}; int len=recv(clientfd,buffer,BUFFER_LENGTH,0); if (len < 0){//出现了异常情况或者非阻塞状态下没有更多数据可读 //关闭该套接字并将其从 epoll 实例中删除 close(clientfd); struct epoll_event ev; ev.events=EPOLLIN; ev.data.fd=clientfd; epoll_ctl(epfd,EPOLL_CTL_DEL,clientfd,&ev); //从 epoll 实例中删除 clientfd 对应的文件描述符，并且停止监听该套接字上的事件。 } else if(len == 0) {//对方已经断开连接 //关闭该套接字并将其从 epoll 实例中删除 close(clientfd); struct epoll_event ev; ev.events=EPOLLIN; ev.data.fd=clientfd; epoll_ctl(epfd,EPOLL_CTL_DEL,clientfd,&ev); } else{ printf("Recv: %s, %d byte(s), clientfd: %d\n", buffer, len, clientfd); } } } } return 0; }