接着上文的[集群聊天服务器]----(八)群组类、群组操作接口以及业务模块之创建群组,加入群组以及群组聊天开发,项目真正操作,还需要客户端进行相关操作的,接下来我们剖析客户端的实现。
main函数
客户端主要对用户输入的消息进行序列化,一定要与之前服务端设置的键值相对应
int main(int argc, char **argv)
{if (argc < 3){cerr << "command invalid! example: ./ChatClient 127.0.0.1 6000" << endl;exit(-1);}// 解析通过命令行参数传递的ip和portchar *ip = argv[1];uint16_t port = atoi(argv[2]);// 创建client端的socketint clientfd = socket(AF_INET, SOCK_STREAM, 0);if (-1 == clientfd){cerr << "socket create error" << endl;exit(-1);}// 填写client需要连接的server信息ip+port 绑定ip和端口号sockaddr_in server;memset(&server, 0, sizeof(sockaddr_in));server.sin_family = AF_INET;server.sin_port = htons(port);server.sin_addr.s_addr = inet_addr(ip);// client和server进行连接if (-1 == connect(clientfd, (sockaddr *)&server, sizeof(sockaddr_in))){cerr << "connect server error" << endl;close(clientfd);exit(-1);}// 初始化读写线程通信用的信号量sem_init(&rwsem, 0, 0);// 连接服务器成功,启动接收子线程std::thread readTask(readTaskHandler, clientfd); // pthread_createreadTask.detach();// main线程用于接收用户输入,负责发送数据for (;;){// 显示首页面菜单 登录、注册、退出cout << "========================" << endl;cout << "1. login" << endl;cout << "2. register" << endl;cout << "3. quit" << endl;cout << "========================" << endl;cout << "choice:";int choice = 0;cin >> choice;cin.get(); // 读掉缓冲区残留的回车 如果不回收到回车,后面在输入就会读取回车 不读取输入的switch (choice){case 1: // login业务{int id = 0;char pwd[50] = {0};cout << "userid:";cin >> id;cin.get(); // 读掉缓冲区残留的回车cout << "userpassword:";cin.getline(pwd, 50);json js;js["msgid"] = LOGIN_MSG;js["id"] = id;js["password"] = pwd;string request = js.dump();g_isLoginSuccess = false;int len = send(clientfd, request.c_str(), strlen(request.c_str()) + 1, 0);if (len == -1){cerr << "send login msg error:" << request << endl;}sem_wait(&rwsem); // 等待信号量,由子线程处理完登录的响应消息后,通知这if (g_isLoginSuccess){isMainMenuRunning = true;// 进入聊天主菜单页面mainMenu(clientfd);}}break;case 2: // register业务{char name[50] = {0};char pwd[50] = {0};cout << "username:";cin.getline(name, 50); // 遇见回车结束cout << "userpassword:";cin.getline(pwd, 50);json js;js["msgid"] = REG_MSG;js["name"] = name;js["password"] = pwd;string request = js.dump();int len = send(clientfd, request.c_str(), strlen(request.c_str()) + 1, 0);if (len == -1){cerr << "send reg msg error:" << request << endl;}sem_wait(&rwsem); // 等待信号量,由子线程处理完注册的响应消息后,通知这}break;case 3: // quit业务close(clientfd);sem_destroy(&rwsem);exit(0);default:cerr << "invalid input!" << endl;break;}}return 0;
}
- 解析通过命令行参数获取传递的ip和port
- 调用
socket
函数,创建client端的socket - 填写client需要连接的server信息ip+port 绑定ip和端口号
- 调用
connect()
函数将client和server进行连接 - 初始化读写线程通信用的信号量
- 连接服务器成功,启动接收子线程,main线程用于接收用户输入,负责发送数据,子线程用作接收线程
- 显示首页面菜单 登录、注册、退出操作,注意读掉缓冲区残留的回车 如果不回收到回车,后面在输入就会读取回车 不读取输入的
- 根据用户的选择登录LOGIN_MSG、注册REG_MSG、退出进行相关操作
登录响应
void doLoginResponse(json &responsejs)
{if (0 != responsejs["errno"].get<int>()) // 登录失败{cerr << responsejs["errmsg"] << endl;g_isLoginSuccess = false;}else{// 记录当前用户的id和nameg_currentUser.setId(responsejs["id"].get<int>());g_currentUser.setName(responsejs["name"]);// 记录当前用户的好友列表信息if (responsejs.contains("friends")){g_currentUserFriendList.clear();vector<string> vec = responsejs["friends"];for (string &str : vec){json js = json::parse(str);User user;user.setId(js["id"].get<int>());user.setName(js["name"]);user.setState(js["state"]);g_currentUserFriendList.push_back(user);}}// 记录当前用户的群组列表信息if (responsejs.contains("groups")){// 初始化g_currentUserGroupList.clear();vector<string> vec1 = responsejs["groups"];for (string &groupstr : vec1){json grpjs = json::parse(groupstr);Group group;group.setId(grpjs["id"].get<int>());group.setName(grpjs["groupname"]);group.setDesc(grpjs["groupdesc"]);vector<string> vec2 = grpjs["users"];for (string &userstr : vec2){GroupUser user;json js = json::parse(userstr);user.setId(js["id"].get<int>());user.setName(js["name"]);user.setState(js["state"]);user.setRole(js["role"]);group.getUsers().push_back(user);}g_currentUserGroupList.push_back(group);}}// 显示登录用户的基本信息showCurrentUserData();// 显示当前用户的离线消息 个人聊天或者群组消息if (responsejs.contains("offlinemsg")){vector<string> vec = responsejs["offlinemsg"];for (string &str : vec){json js = json::parse(str);int msgtype = js["msgid"].get<int>();if (ONE_CHAT_MSG == msgtype){cout << js["time"].get<string>() << "[" << js["id"] << "]" << js["name"].get<string>()<< " said: " << js["msg"].get<string>() << endl;}else if (GROUP_CHAT_MSG == msgtype){cout << "群消息[" << js["groupid"] << "]: " << js["time"].get<string>() << "["<< js["id"] << "]" << js["name"].get<string>()<< " said: " << js["msg"].get<string>() << endl;}}}g_isLoginSuccess = true;}
}
- 登录成功以后记录当前用户的id和name
- 记录当前用户的好友列表信息
- 记录当前用户的群组列表信息
- 显示登录用户的基本信息
- 显示当前用户的离线消息 个人聊天或者群组消息
注册相应
void doRegResponse(json &responsejs)
{if (0 != responsejs["errno"].get<int>()){cerr << "name is already exist, register error!" << endl;}else{cerr << "name register success, userid is" << responsejs["id"]<< ", do not forget it!" << endl;}
}
子线程做接收线程
void readTaskHandler(int clientfd)
{for (;;){char buffer[1024] = {0};int len = recv(clientfd, buffer, 1024, 0); // 阻塞了if (-1 == len || 0 == len){close(clientfd);exit(-1);}// 接受chatserver转发的数据,反序列化生成json数据对象json js = json::parse(buffer);int msgtype = js["msgid"].get<int>();if (ONE_CHAT_MSG == msgtype){cout << js["time"].get<string>() << "[" << js["id"] << "]" << js["name"].get<string>()<< " said: " << js["msg"].get<string>() << endl;continue;}if (GROUP_CHAT_MSG == msgtype){cout << "群消息[" << js["groupid"] << "]: " << js["time"].get<string>() << "["<< js["id"] << "]" << js["name"].get<string>()<< " said: " << js["msg"].get<string>() << endl;continue;}if (LOGIN_MSG_ACK == msgtype) // 登录响应消息{doLoginResponse(js); // 处理登录响应的业务逻辑sem_post(&rwsem); // 通知主线程 登录结果处理完成continue;}if (REG_MSG_ACK == msgtype){doRegResponse(js);sem_post(&rwsem); // 通知主线程 注册结果continue;}}
}
- 根据msgid获取对应的消息类别,并作出相关的回应
- ** 通知主线程 登录以及注册结果**
显示当前登录成功用户的基本信息
void showCurrentUserData()
{cout << "======================login user======================" << endl;cout << "current login user => id:" << g_currentUser.getId() << " name:" << g_currentUser.getName() << endl;cout << "----------------------friend list---------------------" << endl;if (!g_currentUserFriendList.empty()){for (User &user : g_currentUserFriendList){cout << user.getId() << " " << user.getName() << " " << user.getState() << endl;}}cout << "----------------------group list----------------------" << endl;if (!g_currentUserGroupList.empty()){for (Group &group : g_currentUserGroupList){cout << group.getId() << " " << group.getName() << " " << group.getDesc() << endl;for (GroupUser &user : group.getUsers()){cout << user.getId() << " " << user.getName() << " " << user.getState()<< " " << user.getRole() << endl;}}}cout << "======================================================" << endl;
}
获取系统时间
string getCurrentTime()
{auto tt = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());struct tm *ptm = localtime(&tt);char date[60] = {0};sprintf(date, "%d-%02d-%02d %02d:%02d:%02d",(int)ptm->tm_year + 1900, (int)ptm->tm_mon + 1, (int)ptm->tm_mday,(int)ptm->tm_hour, (int)ptm->tm_min, (int)ptm->tm_sec);return std::string(date);
}
系统支持的客户端命令列表commandMap
unordered_map<string, string> commandMap = {{"help", "显示所有支持的命令,格式help"},{"chat", "一对一聊天,格式chat:friendid:message"},{"addfriend", "添加好友,格式addfriend:friendid"},{"creategroup", "创建群组,格式creategroup:groupname:groupdesc"},{"addgroup", "加入群组,格式addgroup:groupid"},{"groupchat", "群聊,格式groupchat:groupid:message"},{"loginout", "注销,格式loginout"}};
注册系统支持的客户端命令处理
unordered_map<string, function<void(int, string)>> commandHandlerMap = {{"help", help},{"chat", chat},{"addfriend", addfriend},{"creategroup", creategroup},{"addgroup", addgroup},{"groupchat", groupchat},{"loginout", loginout}};
help,chat,addfriend,creategroup,addgroup,groupchat,loginout都对应着相应的处理函数
主聊天页面程序
void mainMenu(int clientfd)
{help();char buffer[1024] = {0};while (isMainMenuRunning){cin.getline(buffer, 1024);string commandbuf(buffer);string command;// 找到了返回起始下标int idx = commandbuf.find(":"); // 除了help和loginout没有:if (-1 == idx){command = commandbuf;}else{command = commandbuf.substr(0, idx);}auto it = commandHandlerMap.find(command);if (it == commandHandlerMap.end()){cerr << "invaild input command!" << endl;continue;}// 调用相应命令的事件处理回调 mainMenu对修改封闭 添加新功能不需要修改该函数it->second(clientfd, commandbuf.substr(idx + 1, commandbuf.size() - idx));}
}
- 通过
find
以及substr
函数对命令command
进行了分离,命令存在的话,调用相应命令的事件处理回调
help
命令
void help(int, string)
{cout << "show command list >>> " << endl;for (auto &p : commandMap){cout << p.first << " : " << p.second << endl;}cout << endl;
}
显示系统支持的客户端命令列表
addfriend
命令
void addfriend(int clientfd, string str)
{int friendid = atoi(str.c_str());json js;js["msgid"] = ADD_FRIEND_MSG;js["id"] = g_currentUser.getId();js["friendid"] = friendid;string buffer = js.dump();int len = send(clientfd, buffer.c_str(), strlen(buffer.c_str()) + 1, 0);if (-1 == len){cerr << "send addfriend msg error -> " << buffer << endl;}
}
根据客户输入的命令,调用send()
函数,发向服务端
chat
命令
void chat(int clientfd, string str)
{int idx = str.find(":"); // friendid:messageif (-1 == idx){cerr << "chat command invalid!" << endl;return;}int friendid = atoi(str.substr(0, idx).c_str());string message = str.substr(idx + 1, str.size() - idx);json js;js["msgid"] = ONE_CHAT_MSG;js["id"] = g_currentUser.getId();js["name"] = g_currentUser.getName();js["toid"] = friendid;js["msg"] = message;js["time"] = getCurrentTime();string buffer = js.dump();int len = send(clientfd, buffer.c_str(), strlen(buffer.c_str()) + 1, 0);if (-1 == len){cerr << "send chat msg error -> " << buffer << endl;}
}
- 获取好友id,以及要发送的消息
creategroup
命令
void creategroup(int clientfd, string str)
{int idx = str.find(":");if (-1 == idx){cerr << "creategroup command invalid!" << endl;return;}string groupname = str.substr(0, idx);string groupdesc = str.substr(idx + 1, str.size() - idx);json js;js["msgid"] = CREATE_GROUP_MSG;js["id"] = g_currentUser.getId();js["groupname"] = groupname;js["groupdesc"] = groupdesc;string buffer = js.dump();int len = send(clientfd, buffer.c_str(), strlen(buffer.c_str()) + 1, 0);if (-1 == len){cerr << "send creategroup msg error -> " << buffer << endl;}
}
addgroup
命令
void addgroup(int clientfd, string str)
{int groupid = atoi(str.c_str());json js;js["msgid"] = ADD_GROUP_MSG;js["id"] = g_currentUser.getId();js["groupid"] = groupid;string buffer = js.dump();int len = send(clientfd, buffer.c_str(), strlen(buffer.c_str()) + 1, 0);if (-1 == len){cerr << "send addgroup msg error -> " << buffer << endl;}
}
groupchat
命令
void groupchat(int clientfd, string str)
{int idx = str.find(":"); // friendid:messageif (-1 == idx){cerr << "groupchat command invalid!" << endl;return;}int groupid = atoi(str.substr(0, idx).c_str()); // atoi处理的char*string message = str.substr(idx + 1, str.size() - idx); // 第二个参数是长度json js;js["msgid"] = GROUP_CHAT_MSG;js["id"] = g_currentUser.getId();js["name"] = g_currentUser.getName();js["groupid"] = groupid;js["msg"] = message;js["time"] = getCurrentTime();string buffer = js.dump();int len = send(clientfd, buffer.c_str(), strlen(buffer.c_str()) + 1, 0);if (-1 == len){cerr << "send groupchat msg error -> " << buffer << endl;}
}
loginout
命令
void loginout(int clientfd, string str)
{json js;js["msgid"] = LOGINOUT_MSG;js["id"] = g_currentUser.getId();string buffer = js.dump();int len = send(clientfd, buffer.c_str(), strlen(buffer.c_str()) + 1, 0);if (-1 == len){cerr << "send loginout msg error -> " << buffer << endl;}else{isMainMenuRunning = false;}
}