当前位置：首页 > news >正文

Chap18-AddFriend

news 2026/3/27 6:07:50

Chap18-AddFriend

这一节我们完成好友申请和通知的功能（发送好友申请，收到好友申请通知）。

为了清晰的分辨我们直接分为前端和后端的修改，而非穿插进行。

前端

添加好友请求

正如上节展示的那样，我们将每个查询到的用户数据都使用一个自定的FriendItem展示出来，在每个FriendItem插入到ListWidget的时候，我们都connect了这个添加按钮的信号和槽函数。当点击按钮，触发on_send_data信号，发送的内容如下

connect(_applyFriend,&QPushButton::clicked,this,[this](bool){QJsonObject obj;obj["fromUid"] = static_cast<int>(UserManager::GetInstance()->GetUid());obj["fromName"] = UserManager::GetInstance()->GetName();obj["fromEmail"] = UserManager::GetInstance()->GetEmail();obj["fromDesc"] = UserManager::GetInstance()->GetDesc();obj["fromSex"] = UserManager::GetInstance()->GetSex();obj["fromIcon"] = UserManager::GetInstance()->GetIcon();obj["toUid"] = this->_uid; // 对方的uidqDebug() << "fromUid" << obj["fromUid"] << "\t" << "toUid" << this->_uid;QJsonDocument doc;doc.setObject(obj);QByteArray data = doc.toJson(QJsonDocument::Compact);emit TcpManager::GetInstance()->on_send_data(RequestType::ID_ADD_FRIEND_REQ,data);this->_applyFriend->setEnabled(false);showToolTip(_applyFriend,"已发送好友请求");});

这时候我们就可以等待对端回包，当服务器将包回复过来之后，我们在TcpManager处理：

/*** @brief 用户添加请求回包处理*/_handlers[RequestType::ID_ADD_FRIEND_RSP] = [this](RequestType requestType,int len,QByteArray data){QJsonDocument jsonDoc = QJsonDocument::fromJson(data);if (jsonDoc.isNull()){qDebug() << "Error occured about Json";return;}QJsonObject jsonObj = jsonDoc.object();if (!jsonObj.contains("error")){int err = static_cast<int>(ErrorCodes::ERROR_JSON);qDebug() << "AddFriend Failed,Error Is Json Parse Error " <<err;return;}int err = jsonObj["error"].toInt();if (err != static_cast<int>(ErrorCodes::SUCCESS)){qDebug() << "AddFriend Failed,Error Is " << err;return;}UserInfo info;info.id = jsonObj["fromUid"].toInt();info.email = jsonObj["fromEmail"].toString();info.name = jsonObj["fromName"].toString();info.status = jsonObj["fromStatus"].toString();// TODO:qDebug() << "申请添加好友成功";emit on_add_friend(info);};

在这里我们触发了on_add_friend信号，我们接下来的思路是实现一个侧边滑动的专门用于查看好友通知系统通知的通知栏，我们将在通知栏中处理这个信号加入到通知栏的ListWidget中展示。

用户添加好友请求

当服务器发送给用户好友请求的时候，我们仍然需要在TcpManager里面进行处理:

/*** @brief 用户请求添加好友通知处理*/_handlers[RequestType::ID_NOTIFY_ADD_FRIEND_REQ] = [this](RequestType requestType,int len,QByteArray data){QJsonDocument jsonDoc = QJsonDocument::fromJson(data);if (jsonDoc.isNull()){return;}QJsonObject jsonObj = jsonDoc.object();if (!jsonObj.contains("error")){int err = static_cast<int>(ErrorCodes::ERROR_JSON);return;}int err = jsonObj["error"].toInt();if (err != static_cast<int>(ErrorCodes::SUCCESS)){return;}int from_uid = jsonObj["from_uid"].toInt();int from_sex = jsonObj["sex"].toInt();QString from_name = jsonObj["from_name"].toString();QString from_icon = jsonObj["from_icon"].toString();QString from_desc = jsonObj["from_desc"].toString();auto user_info = std::make_shared<UserInfo>();user_info->id = from_uid;user_info->sex = from_sex;user_info->name = from_name;user_info->avatar = from_icon;user_info->desc = from_desc;qDebug() << "收到好友请求";emit on_auth_friend(user_info);};

当服务器传回其他用户的好友通知时候，我们发送on_auth_friend信号，同样要在通知栏处理。

后端（ChatServer）

我们主要在LogicSystem对客户端发来的请求进行处理,对于好友请求申请我们如下处理：

 /** * @brief 好友申请请求*/_function_callbacks[MsgId::ID_ADD_FRIEND_REQ] = [this](std::shared_ptr<Session> session, uint16_t msg_id, const std::string& msg) {json j = json::parse(msg);j["error"] = ErrorCodes::SUCCESS;Defer defer([this, &j, session]() {// 回复请求方的信息session->Send(j.dump(), static_cast<int>(MsgId::ID_ADD_FRIEND_RSP));});auto toUid = j["toUid"].get<int>();auto fromUid = j["fromUid"].get<int>();auto fromName = j["fromName"].get<std::string>();auto fromSex = j["fromSex"].get<int>();auto fromDesc = j["fromDesc"].get<std::string>();// auto fromIcon = j["fromIcon"].get<std::string>();auto fromIcon = j.value("fromIcon", "");std::string uid_str = std::to_string(toUid);bool b_apply = MysqlManager::GetInstance()->AddFriendApply(std::to_string(fromUid), uid_str);if (!b_apply) {return;}auto to_key = USERIP_PREFIX + uid_str;std::string to_ip_value;bool b_ip = RedisManager::GetInstance()->Get(to_key, to_ip_value);if (!b_ip) {return;}// 是对方发送请求信息auto& cfg = ConfigManager::GetInstance();auto self_name = cfg["SelfServer"]["name"];if (to_ip_value == self_name) {auto session2 = UserManager::GetInstance()->GetSession(toUid);if (session2) {SPDLOG_INFO("FROM UID:{},to:{}", fromUid, toUid);SPDLOG_INFO("FROM SESSION:{},to:{}", session->GetSessionId(), session2->GetSessionId());json jj;jj["error"] = ErrorCodes::SUCCESS;jj["fromUid"] = fromUid;jj["fromName"] = fromName;session2->Send(jj.dump(), static_cast<int>(MsgId::ID_NOTIFY_ADD_FRIEND_REQ));}return;}// std::string base_key = USER_BASE_INFO_PREFIX + toUid;// auto apply_info = std::make_shared<UserInfo>();// bool b_info = GetBaseInfo(base_key, std::stoi(toUid),apply_info);AddFriendRequest req;req.set_fromuid(fromUid);req.set_touid(toUid);req.set_name(fromName);req.set_desc(fromDesc);req.set_sex(fromSex);req.set_icon(fromIcon);ChatGrpcClient::GetInstance()->NotifyAddFriend(to_ip_value, req);};

首先将申请好友信息加入mysql防止信息丢失（以后会拓展为离线不发送，上线再发送，不会丢失）。然后查询被申请用户的服务器名称，如果没有查询到，说明没有在线，我们直接return.如果查询到之后，我们需要检查被申请用户是否连接当前服务器，如果连接了当前服务器，我们直接根据uid查找session然后发送信息。如果连接的是其他服务器，我们就通过grpc服务器将信息发送给其他服务器转发给被申请人。

我们先看mysql(Dao)的AddFriendApply函数：

bool MysqlDao::AddFriendApply(const std::string& fromUid, const std::string& toUid)
{auto conn = _pool->GetConnection();if (!conn) {SPDLOG_ERROR("Failed to get connection from pool");return false;}Defer defer([this, &conn]() {_pool->ReturnConnection(std::move(conn));});try {mysqlpp::Query query = conn->query();query << "Insert into friend_apply (from_uid,to_uid) values(%0,%1) "<< "on duplicate key update from_uid = from_uid,to_uid=to_uid";query.parse();mysqlpp::SimpleResult res = query.execute(std::stoi(fromUid), std::stoi(toUid));int rowCount = res.rows();return rowCount >= 0;} catch (const mysqlpp::Exception& e) {SPDLOG_ERROR("MySQL++ exception: {}", e.what());return false;}return true;
}

接下来是grpc请求NotifyAddFriend(客户端)

AddFriendResponse ChatGrpcClient::NotifyAddFriend(std::string server_ip, const AddFriendRequest& req)
{SPDLOG_INFO("发送好友请求to:{}", req.touid());SPDLOG_INFO("目标服务名称:{}", server_ip);AddFriendResponse rsp;Defer defer([&rsp, &req]() {rsp.set_error(static_cast<int>(ErrorCodes::SUCCESS));rsp.set_fromuid(req.fromuid());rsp.set_touid(req.touid());});auto it = _pool.find(server_ip);if (it == _pool.end()) {return rsp;}auto& pool = it->second;SPDLOG_INFO("服务端ip,{}:{}", _pool[server_ip]->_host, _pool[server_ip]->_port);grpc::ClientContext context;auto stub = pool->GetConnection();Defer defer2([&pool, &stub]() {pool->ReturnConnection(std::move(stub));});Status status = stub->NotifyAddFriend(&context, req, &rsp);if (!status.ok()) {rsp.set_error(static_cast<int>(ErrorCodes::RPCFAILED));return rsp;}return rsp;
}

然后是grpc请求NotifyAddFriend(服务端):

Status ChatGrpcServer::NotifyAddFriend(grpc::ServerContext* context, const AddFriendRequest* request, AddFriendResponse* response)
{SPDLOG_INFO("Add Friend Request From {}", request->fromuid());// 首先在本服务器查询auto to_uid = request->touid();auto session = UserManager::GetInstance()->GetSession(to_uid);Defer defer([request, response]() {response->set_error(static_cast<int>(ErrorCodes::SUCCESS));response->set_fromuid(request->fromuid());response->set_touid(request->touid());});// 不在内存中if (session == nullptr) {return Status::OK;}// 在内存中z直接发送通知json j;j["error"] = ErrorCodes::SUCCESS;j["from_uid"] = request->fromuid();j["name"] = request->name();j["icon"] = request->icon();j["sex"] = request->sex();j["desc"] = request->desc();session->Send(j.dump(), static_cast<int>(MsgId::ID_NOTIFY_ADD_FRIEND_REQ));return Status::OK;
}