epoll高并发聊天室(C++,linux,socket,pthread)
为了具体地理解epoll的使用，本博文是用epoll实现了一个高并发聊天室服务器。
客户端使用pthread的多线程来实时读取所收到的信息，并且最初连接后输入的数据代表了该客户的名字;
服务器支持客户端查找当前所在线的客户name和id，支持客户端对指定的客户端私信，服务器采用边缘触发的模式;（其中需要注意的是若客户端想要私信另一个客户端，则通过输入@targetname@message的形式来私信,客户端输入cl来获取当前在线的client列表）
参考的代码：https://blog.csdn.net/sinat_35297665/article/details/79476256
参考了此博主对于epoll的详细使用。
chatroom.h
#include
#include server.cpp
#include "chatroom.h"
int main(int argc, char *argv[])
{
//服务器IP + port
struct sockaddr_in serverAddr;
serverAddr.sin_family = PF_INET;
serverAddr.sin_port = htons(SERVER_PORT);
serverAddr.sin_addr.s_addr = inet_addr(SERVER_IP);
//创建监听socket
int listener = socket(PF_INET, SOCK_STREAM, 0);
if(listener < 0) { perror("listener"); exit(-1);}
printf("listen socket created \n");
//绑定地址
const int on=1;
setsockopt(listener,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on));
if( bind(listener, (struct sockaddr *)&serverAddr, sizeof(serverAddr)) < 0) {
perror("bind error");
exit(-1);
}
//监听
int ret = listen(listener, 5);
if(ret < 0) { perror("listen error"); exit(-1);}
printf("Start to listen: %s\n", SERVER_IP);
//在内核中创建事件表
int epfd = epoll_create(EPOLL_SIZE);
if(epfd < 0) { perror("epfd error"); exit(-1);}
printf("epoll created, epollfd = %d\n", epfd);
static struct epoll_event events[EPOLL_SIZE];
//往内核事件表里添加事件
addfd(epfd, listener, true);
//主循环
while(1)
{
//epoll_events_count表示就绪事件的数目
int epoll_events_count = epoll_wait(epfd, events, EPOLL_SIZE, -1);
if(epoll_events_count < 0) {
perror("epoll failure");
break;
}
//处理这epoll_events_count个就绪事件
for(int i = 0; i < epoll_events_count; ++i)
{
int sockfd = events[i].data.fd;
//新用户连接
if(sockfd == listener)
{
struct sockaddr_in client_address;
socklen_t client_addrLength = sizeof(struct sockaddr_in);
int clientfd = accept( listener, ( struct sockaddr* )&client_address, &client_addrLength );
char name[100]={};
recv(clientfd,name,sizeof(name),0);//接收客户端的名字信息
printf("\n%s connected, socketID : %d\n",
name,
clientfd);
addfd(epfd, clientfd, true);
// 服务端用list保存用户连接
clt_map[clientfd]=string(name);
printf("Now there are %d clients int the chat room\n", (int)clt_map.size());
// 服务端发送欢迎信息
char message[BUF_SIZE];
bzero(message, BUF_SIZE);
sprintf(message, SERVER_WELCOME, clientfd);
int ret = send(clientfd, message, BUF_SIZE, 0);
if(ret < 0) { perror("send error"); exit(-1); }
}
//处理用户发来的消息，并广播，使其他用户收到信息
else
{
int ret = sendBroadcastmessage(sockfd);
if(ret < 0) { perror("error");exit(-1); }
}
}
}
close(listener); //关闭socket
close(epfd); //关闭内核
return 0;
}
client.cpp
#include "../server/chatroom.h"
using namespace std;
void* receive(void* arg){
int *temp=((int*)arg);
int sock=*temp;
while(true){
char recvBuf[BUF_SIZE] = {};
int reLen = recv(sock, recvBuf, BUF_SIZE, 0);
cout<<endl<<recvBuf<<endl;
}
pthread_exit(NULL);
}
int main()
{
int sock = socket(AF_INET,SOCK_STREAM,0);
struct sockaddr_in server;
server.sin_family = AF_INET;
server.sin_port = htons(SERVER_PORT);
server.sin_addr.s_addr = inet_addr(SERVER_IP);
if(connect(sock,(struct sockaddr *)&server,sizeof(server)) < 0)
{
perror("connect");
return 2;
}
string name;
cout<<"Enter name:";
getline(cin,name);
write(sock,(char*)name.c_str(),name.length());
void* temp=&sock;
pthread_t th;
pthread_create(&th,NULL,receive,temp);
while(true){
string s;
getline(cin,s);
write(sock,(char*)s.c_str(),s.length());
}
close(sock);
return 0;
}