聊天项目(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. slice介绍和使用

    喜欢(521) 浏览(1880)
  2. 解密定时器的实现细节

    喜欢(566) 浏览(2125)
  3. C++ 类的继承封装和多态

    喜欢(588) 浏览(2960)
  4. Linux环境搭建和编码

    喜欢(594) 浏览(6419)
  5. windows环境搭建和vscode配置

    喜欢(587) 浏览(1801)

最新评论

  1. C++ 虚函数表原理和类成员内存分布 WangQi888888:class Test{ int m; int b; }中b成员是int,为什么在内存中只占了1个字节。不应该是4个字节吗?是不是int应该改为char。这样的话就会符合图上说明的情况
  2. 类和对象 陈宇航:支持!!!!
  3. 解决博客回复区被脚本注入的问题 secondtonone1:走到现在我忽然明白一个道理,无论工作也好生活也罢,最重要的是开心,即使一份安稳的工作不能给我带来事业上的积累也要合理的舍弃,所以我还是想去做喜欢的方向。
  4. slice介绍和使用 恋恋风辰:切片作为引用类型极大的提高了数据传递的效率和性能,但也要注意切片的浅拷贝隐患,算是一把双刃剑,这世间的常态就是在两极之间寻求一种稳定。
  5. 泛型算法的定制操作 secondtonone1:lambda和bind是C11新增的利器,善于利用这两个机制可以极大地提升编程安全性和效率。

个人公众号

个人微信