聊天项目(17) 实现验证服务和客户端数据管理

完善proto

在proto文件里新增登陆验证服务

  1. message LoginReq{
  2. int32 uid = 1;
  3. string token= 2;
  4. }
  5. message LoginRsp {
  6. int32 error = 1;
  7. int32 uid = 2;
  8. string token = 3;
  9. }
  10. service StatusService {
  11. rpc GetChatServer (GetChatServerReq) returns (GetChatServerRsp) {}
  12. rpc Login(LoginReq) returns(LoginRsp);
  13. }

接下来是调用grpc命令生成新的pb文件覆盖原有的,并且也拷贝给StatusServer一份

我们完善登陆逻辑,先去StatusServer验证token是否合理,如果合理再从内存中寻找用户信息,如果没找到则从数据库加载一份。

  1. void LogicSystem::LoginHandler(shared_ptr<CSession> session, const short &msg_id, const string &msg_data) {
  2. Json::Reader reader;
  3. Json::Value root;
  4. reader.parse(msg_data, root);
  5. auto uid = root["uid"].asInt();
  6. std::cout << "user login uid is " << uid << " user token is "
  7. << root["token"].asString() << endl;
  8. //从状态服务器获取token匹配是否准确
  9. auto rsp = StatusGrpcClient::GetInstance()->Login(uid, root["token"].asString());
  10. Json::Value rtvalue;
  11. Defer defer([this, &rtvalue, session]() {
  12. std::string return_str = rtvalue.toStyledString();
  13. session->Send(return_str, MSG_CHAT_LOGIN_RSP);
  14. });
  15. rtvalue["error"] = rsp.error();
  16. if (rsp.error() != ErrorCodes::Success) {
  17. return;
  18. }
  19. //内存中查询用户信息
  20. auto find_iter = _users.find(uid);
  21. std::shared_ptr<UserInfo> user_info = nullptr;
  22. if (find_iter == _users.end()) {
  23. //查询数据库
  24. user_info = MysqlMgr::GetInstance()->GetUser(uid);
  25. if (user_info == nullptr) {
  26. rtvalue["error"] = ErrorCodes::UidInvalid;
  27. return;
  28. }
  29. _users[uid] = user_info;
  30. }
  31. else {
  32. user_info = find_iter->second;
  33. }
  34. rtvalue["uid"] = uid;
  35. rtvalue["token"] = rsp.token();
  36. rtvalue["name"] = user_info->name;
  37. }

StatusServer验证token

在StatusServer验证token之前,我们需要在StatusServer中的GetServer的服务里将token写入内存

  1. Status StatusServiceImpl::GetChatServer(ServerContext* context, const GetChatServerReq* request, GetChatServerRsp* reply)
  2. {
  3. std::string prefix("llfc status server has received : ");
  4. const auto& server = getChatServer();
  5. reply->set_host(server.host);
  6. reply->set_port(server.port);
  7. reply->set_error(ErrorCodes::Success);
  8. reply->set_token(generate_unique_string());
  9. insertToken(request->uid(), reply->token());
  10. return Status::OK;
  11. }

接下来我们实现登陆验证服务

  1. Status StatusServiceImpl::Login(ServerContext* context, const LoginReq* request, LoginRsp* reply)
  2. {
  3. auto uid = request->uid();
  4. auto token = request->token();
  5. std::lock_guard<std::mutex> guard(_token_mtx);
  6. auto iter = _tokens.find(uid);
  7. if (iter == _tokens.end()) {
  8. reply->set_error(ErrorCodes::UidInvalid);
  9. return Status::OK;
  10. }
  11. if (iter->second != token) {
  12. reply->set_error(ErrorCodes::TokenInvalid);
  13. return Status::OK;
  14. }
  15. reply->set_error(ErrorCodes::Success);
  16. reply->set_uid(uid);
  17. reply->set_token(token);
  18. return Status::OK;
  19. }

这样当GateServer访问StatusServer的Login服务做验证后,就可以将数据返回给QT前端了。

客户端处理登陆回包

QT 的客户端TcpMgr收到请求后要进行对应的逻辑处理。所以我们在TcpMgr的构造函数中调用initHandlers注册消息

  1. void TcpMgr::initHandlers()
  2. {
  3. //auto self = shared_from_this();
  4. _handlers.insert(ID_CHAT_LOGIN_RSP, [this](ReqId id, int len, QByteArray data){
  5. qDebug()<< "handle id is "<< id << " data is " << data;
  6. // 将QByteArray转换为QJsonDocument
  7. QJsonDocument jsonDoc = QJsonDocument::fromJson(data);
  8. // 检查转换是否成功
  9. if(jsonDoc.isNull()){
  10. qDebug() << "Failed to create QJsonDocument.";
  11. return;
  12. }
  13. QJsonObject jsonObj = jsonDoc.object();
  14. if(!jsonObj.contains("error")){
  15. int err = ErrorCodes::ERR_JSON;
  16. qDebug() << "Login Failed, err is Json Parse Err" << err ;
  17. emit sig_login_failed(err);
  18. return;
  19. }
  20. int err = jsonObj["error"].toInt();
  21. if(err != ErrorCodes::SUCCESS){
  22. qDebug() << "Login Failed, err is " << err ;
  23. emit sig_login_failed(err);
  24. return;
  25. }
  26. UserMgr::GetInstance()->SetUid(jsonObj["uid"].toInt());
  27. UserMgr::GetInstance()->SetName(jsonObj["name"].toString());
  28. UserMgr::GetInstance()->SetToken(jsonObj["token"].toString());
  29. emit sig_swich_chatdlg();
  30. });
  31. }

并且增加处理请求

  1. void TcpMgr::handleMsg(ReqId id, int len, QByteArray data)
  2. {
  3. auto find_iter = _handlers.find(id);
  4. if(find_iter == _handlers.end()){
  5. qDebug()<< "not found id ["<< id << "] to handle";
  6. return ;
  7. }
  8. find_iter.value()(id,len,data);
  9. }

用户管理

为管理用户数据,需要创建一个UserMgr类,统一管理用户数据,我们这么声明

  1. #ifndef USERMGR_H
  2. #define USERMGR_H
  3. #include <QObject>
  4. #include <memory>
  5. #include <singleton.h>
  6. class UserMgr:public QObject,public Singleton<UserMgr>,
  7. public std::enable_shared_from_this<UserMgr>
  8. {
  9. Q_OBJECT
  10. public:
  11. friend class Singleton<UserMgr>;
  12. ~ UserMgr();
  13. void SetName(QString name);
  14. void SetUid(int uid);
  15. void SetToken(QString token);
  16. private:
  17. UserMgr();
  18. QString _name;
  19. QString _token;
  20. int _uid;
  21. };
  22. #endif // USERMGR_H

简单实现几个功能

  1. #include "usermgr.h"
  2. UserMgr::~UserMgr()
  3. {
  4. }
  5. void UserMgr::SetName(QString name)
  6. {
  7. _name = name;
  8. }
  9. void UserMgr::SetUid(int uid)
  10. {
  11. _uid = uid;
  12. }
  13. void UserMgr::SetToken(QString token)
  14. {
  15. _token = token;
  16. }
  17. UserMgr::UserMgr()
  18. {
  19. }

详细和复杂的管理后续不断往这里补充就行了。

登陆界面

登陆界面响应TcpMgr返回的登陆请求,在其构造函数中添加

  1. //连接tcp管理者发出的登陆失败信号
  2. connect(TcpMgr::GetInstance().get(), &TcpMgr::sig_login_failed, this, &LoginDialog::slot_login_failed);

并实现槽函数

  1. void LoginDialog::slot_login_failed(int err)
  2. {
  3. QString result = QString("登录失败, err is %1")
  4. .arg(err);
  5. showTip(result,false);
  6. enableBtn(true);
  7. }

到此完成了登陆的请求和响应,接下来要实现响应登陆成功后跳转到聊天界面。下一篇先实现聊天布局。

热门评论

热门文章

  1. windows环境搭建和vscode配置

    喜欢(587) 浏览(2806)
  2. Linux环境搭建和编码

    喜欢(594) 浏览(12145)
  3. 解密定时器的实现细节

    喜欢(566) 浏览(3466)
  4. slice介绍和使用

    喜欢(521) 浏览(2478)
  5. C++ 类的继承封装和多态

    喜欢(588) 浏览(4951)

最新评论

  1. C++ 并发三剑客future, promise和async Yunfei:大佬您好,如果这个线程池中加入的异步任务的形参如果有右值引用,这个commit中的返回类型推导和bind绑定就会出现问题,请问实际工程中,是不是不会用到这种任务,如果用到了,应该怎么解决?
  2. Qt MVC结构之QItemDelegate介绍 胡歌-此生不换:gpt, google
  3. 聊天项目(9) redis服务搭建 pro_lin:redis线程池的析构函数,除了pop出队列,还要free掉redis连接把
  4. 答疑汇总(thread,async源码分析) Yagus:如果引用计数为0,则会执行 future 的析构进而等待任务执行完成,那么看到的输出将是 这边应该不对吧,std::future析构只在这三种情况都满足的时候才回block: 1.共享状态是std::async 创造的(类型是_Task_async_state) 2.共享状态没有ready 3.这个future是共享状态的最后一个引用 这边共享状态类型是“_Package_state”,引用计数即使为0也不应该block啊

个人公众号

个人微信