C++實現(xiàn)Window環(huán)境聊天室功能
C++ Window環(huán)境聊天室,供大家參考,具體內(nèi)容如下
最近閑來無事,想學習一下C++的并發(fā)編程和Socket信息傳輸,故以聊天室為載體進行學習。話不多說,代碼開干。
聊天室分為多個客戶端和一個服務器。服務器負責接收各個客戶端傳來的信息,然后轉(zhuǎn)發(fā)給各個客戶端,起到“中介”的作用;而客戶端可以發(fā)送消息,并接收消息。
1.服務端
根據(jù)上述所說,一個服務器需要做的事:
① 接受新的客戶端連接請求
② 聆聽各個已連接的客戶端的新消息
③ 將收到的信息逐個傳送給各個已連接的服務器
所以,我設置服務器擁有兩個線程,一個線程負責①功能,一個線程負責②③功能(注:這里我使用的是接受到消息就馬上轉(zhuǎn)發(fā),但并沒有容器進行保存,若高并發(fā)階段可能會發(fā)生漏發(fā))。
① 創(chuàng)建聆聽客戶端連接請求線程
void CreateListenThread(int sock_serve,list<int>*users){ sockaddr_in user_addr; int user_size = sizeof(user_addr); // 循環(huán)接受 while(1){ // 每次循環(huán)暫停100ms,不熱會占用大量cpu this_thread::sleep_for(chrono::milliseconds(100)); int user_sock = accept(sock_serve,(sockaddr*)(&user_addr),&user_size); if(user_sock==-1){ //表明無連接 continue; } if(user_sock!=INVALID_SOCKET){ lock_guard<mutex>mtx_locker(mtx); (*users).push_back(user_sock); cout<<"成功加入了新client:"<<user_sock<<endl; } } return; }
② 創(chuàng)建聆聽客戶端發(fā)送消息的線程
//廣播消息 void Broadcast(list<int>*users, char *str){ for(auto iter=users->begin();iter!=users->end();iter++){ int flag = send(*iter, str, strlen(str)+1, 0); if(!flag){ cout<<*iter<<"用戶發(fā)送失敗"<<endl; } } return; } void CreateRecvThread(int sock_serve,list<int>*users){ char revc_string[maxLen]; int recvbuffs; while(1){ this_thread::sleep_for(chrono::milliseconds(300)); lock_guard<mutex>mtx_locker(mtx); for(auto iter=users->begin();iter!=users->end();){ recvbuffs = recv(*iter, revc_string,maxLen,0); // cout<<"iter="<<*iter<<" len="<<recvbuffs<<endl; if(recvbuffs==0){ cout<<"客戶端:"<<*iter<<"已退出!!"<<endl; users->erase(iter++); continue; } else if(recvbuffs<0){ iter++; } else{ cout<<string(revc_string,recvbuffs)<<endl; Broadcast(users, revc_string); // 進行廣播 memset(revc_string,0,sizeof(revc_string)); iter++; } } } return; }
在這里,我設置的是 當服務器接受到某個客戶端發(fā)送的消息后,立馬傳送。如果需要解決高并發(fā)出現(xiàn)的問題,我在這里設想一個提議,設置queue隊列,當服務器接受消息后立馬將消息寫入隊列中,另外再創(chuàng)建一個線程主要負責將隊列中的消息廣播出去(當然了,隊列必須需要互斥操作),從而使得服務器發(fā)送消息和接受消息解耦。
2.客戶端
根據(jù)上述所說,客戶端在這里需要做的事:
① “打電話”,也就是和指定服務器連接上
② 發(fā)送消息
③ 接收其他所有客戶端的消息
所以,這部分我創(chuàng)建了兩個線程,一個線程主要負責發(fā)送消息,一線程主要負責接受消息。
① 創(chuàng)建發(fā)送消息的線程
// 發(fā)送數(shù)據(jù) void CreateSendThread(int socket_client, const char *name){ char send_string[maxLen],tmp[maxLen]; while(1){ this_thread::sleep_for(chrono::milliseconds(300)); lock_guard<mutex>mtx_locker(mtx); memset(send_string,0,sizeof(send_string)); scanf("%s",tmp); strcat(send_string,name); strcat(send_string,"說:"); strcat(send_string,tmp); send(socket_client, send_string,strlen(send_string)+1,0); } return; }
通過send()函數(shù)將數(shù)據(jù)信息傳送到指定服務器
② 創(chuàng)建聆聽服務器廣播消息的線程
// 接受數(shù)據(jù) void CreateRecvThread(int sock_client){ char buff[maxLen]; while(1){ this_thread::sleep_for(chrono::milliseconds(300)); lock_guard<mutex>mtx_locker(mtx); int revc_len=recv(sock_client,buff,sizeof(buff),0); if(revc_len==0){ cout<<"服務器已關(guān)閉,服務結(jié)束!"<<endl; break; } else if(revc_len<0)continue; else{ printf("%s\n",buff); memset(buff,0,sizeof(buff)); } } }
通過revc()函數(shù)接受服務器廣播消息,當
返回值 == 0時,服務器已斷連,需要結(jié)束線程
返回值 <0 時,服務器沒有傳送消息過來
返回值 >0 時,服務器傳送數(shù)據(jù)過來(注意:當傳送空串過來時,返回值也為1)
3. 實驗結(jié)果
上圖中,左一為服務器,左二、左三為客戶端,可以見到,當客戶端發(fā)送消息時,其他客戶端都接受到消息。
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。