恋恋风辰的官方博客

异步回调实现续传持久化
Posted on 2025-11-13 10:13:56 by 恋恋风辰 | In C++ | 原文链接

续传信息持久化

增加redis接口

  1. bool RedisMgr::SetFileInfo(const std::string& md5, std::shared_ptr<FileInfo> file_info)
  2. {
  3.     Json::Reader reader;
  4.     Json::Value root;
  5.     root["file_path_str"] = file_info->_file_path_str;
  6.     root["name"] = file_info->_name;
  7.     root["seq"] = file_info->_seq;
  8.     root["total_size"] = file_info->_total_size;
  9.     root["trans_size"] = file_info->_trans_size;
  10.     auto file_info_str = root.toStyledString();
  11.     auto redis_key = "file_upload_" + md5;
  12.     bool success = SetExp(redis_key, file_info_str, 3600);
  13.     return success;
  14. }

新增超时设置

  1. bool RedisMgr::SetExp(const std::string& key, const std::string& value, int expire_seconds) {
  2.     //执行redis命令行
  3.     auto connect = _con_pool->getConnection();
  4.     if (connect == nullptr) {
  5.         return false;
  6.     }
  7.     auto reply = (redisReply*)redisCommand(connect, "SETEX %s %d %s", key.c_str(), 
  8.               expire_seconds,
  9.         value.c_str());
  11.     if (NULL == reply) {
  12.         std::cout << "Execute command [ SETEX " << key << " " << expire_seconds
  13.             << " " << value << " ] failure ! " << std::endl;
  14.         _con_pool->returnConnection(connect);
  15.         return false;
  16.     }
  18.     if (!(reply->type == REDIS_REPLY_STATUS &&
  19.         (strcmp(reply->str, "OK") == 0 || strcmp(reply->str, "ok") == 0))) {
  20.         std::cout << "Execute command [ SETEX " << key << " " << expire_seconds
  21.             << " " << value << " ] failure ! " << std::endl;
  22.         freeReplyObject(reply);
  23.         _con_pool->returnConnection(connect);
  24.         return false;
  25.     }
  27.     freeReplyObject(reply);
  28.     std::cout << "Execute command [ SETEX " << key << " " << expire_seconds
  29.         << " " << value << " ] success ! " << std::endl;
  30.     _con_pool->returnConnection(connect);
  31.     return true;
  32. }

每次收到上传信息后，更新上传进度到redis中

  1. _fun_callbacks[ID_UPLOAD_HEAD_ICON_REQ] = [this](shared_ptr<CSession> session, const short& msg_id,
  2.         const string& msg_data) {
  3.             Json::Reader reader;
  4.             Json::Value root;
  5.             reader.parse(msg_data, root);
  6.             auto md5 = root["md5"].asString();
  7.             auto seq = root["seq"].asInt();
  8.             auto name = root["name"].asString();
  9.             auto total_size = root["total_size"].asInt();
  10.             auto trans_size = root["trans_size"].asInt();
  11.             auto last = root["last"].asInt();
  12.             auto file_data = root["data"].asString();
  13.             auto uid = root["uid"].asInt();
  14.             auto token = root["token"].asString();
  15.             auto last_seq = root["last_seq"].asInt();
  16.             //转化为字符串
  17.             auto uid_str = std::to_string(uid);
  19.             auto file_path = ConfigMgr::Inst().GetFileOutPath();
  20.             auto file_path_str = (file_path / uid_str / name).string();
  21.             Json::Value  rtvalue;
  22.             auto callback = [=](const Json::Value& result) {
  24.                 // 在异步任务完成后调用
  25.                 Json::Value rtvalue = result;
  26.                 rtvalue["error"] = ErrorCodes::Success;
  27.                 rtvalue["total_size"] = total_size;
  28.                 rtvalue["seq"] = seq;
  29.                 rtvalue["name"] = name;
  30.                 rtvalue["trans_size"] = trans_size;
  31.                 rtvalue["last"] = last;
  32.                 rtvalue["md5"] = md5;
  33.                 rtvalue["uid"] = uid;
  34.                 rtvalue["last_seq"] = last_seq;
  35.                 std::string return_str = rtvalue.toStyledString();
  36.                 session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  37.             };
  39.             //第一个包校验一下token是否合理
  40.             if (seq == 1) {
  41.                 //从redis获取用户token是否正确
  42.                 std::string uid_str = std::to_string(uid);
  43.                 std::string token_key = USERTOKENPREFIX + uid_str;
  44.                 std::string token_value = "";
  45.                 bool success = RedisMgr::GetInstance()->Get(token_key, token_value);
  46.                 if (!success) {
  47.                     rtvalue["error"] = ErrorCodes::UidInvalid;
  48.                     std::string return_str = rtvalue.toStyledString();
  49.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  50.                     return;
  51.                 }
  53.                 if (token_value != token) {
  54.                     rtvalue["error"] = ErrorCodes::TokenInvalid;
  55.                     std::string return_str = rtvalue.toStyledString();
  56.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  57.                     return;
  58.                 }
  59.             }
  61.             // 使用 std::hash 对字符串进行哈希
  62.             std::hash<std::string> hash_fn;
  63.             size_t hash_value = hash_fn(name); // 生成哈希值
  64.             int index = hash_value % FILE_WORKER_COUNT;
  65.             std::cout << "Hash value: " << hash_value << std::endl;
  67.             //第一个包
  68.             if (seq == 1) {
  69.                 //构造数据存储
  70.                 auto file_info = std::make_shared<FileInfo>();
  71.                 file_info->_file_path_str = file_path_str;
  72.                 file_info->_name = name;
  73.                 file_info->_seq = seq;
  74.                 file_info->_total_size = total_size;
  75.                 file_info->_trans_size = trans_size;
  76.                 //LogicSystem::GetInstance()->AddMD5File(md5, file_info);
  77.                 //改为用redis存储
  78.                 bool success = RedisMgr::GetInstance()->SetFileInfo(name, file_info);
  79.                 if (!success) {
  80.                     rtvalue["error"] = ErrorCodes::FileSaveRedisFailed;
  81.                     std::string return_str = rtvalue.toStyledString();
  82.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  83.                     return;
  84.                 }
  85.             }
  86.             else {
  87.                 //auto file_info = LogicSystem::GetInstance()->GetFileInfo(md5);
  88.                 //改为从redis中加载
  89.                 auto file_info = RedisMgr::GetInstance()->GetFileInfo(name);
  90.                 if (file_info == nullptr) {
  91.                     rtvalue["error"] = ErrorCodes::FileNotExists;
  92.                     std::string return_str = rtvalue.toStyledString();
  93.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  94.                     return;
  95.                 }
  96.                 file_info->_seq = seq;
  97.                 file_info->_trans_size = trans_size;
  98.                 bool success = RedisMgr::GetInstance()->SetFileInfo(name, file_info);
  99.                 if (!success) {
  100.                     rtvalue["error"] = ErrorCodes::FileSaveRedisFailed;
  101.                     std::string return_str = rtvalue.toStyledString();
  102.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  103.                     return;
  104.                 }
  105.             }
  108.             FileSystem::GetInstance()->PostMsgToQue(
  109.                 std::make_shared<FileTask>(session, uid, file_path_str, name, seq, total_size,
  110.                     trans_size, last, file_data, callback),
  111.                 index
  112.             );
  114.     };

资源url更新

因为上头像传资源后，要将资源的路径存储到mysql数据库中，所以我们新增MysqlMgr,这个直接从ChatServer拷贝一份即可。

但是要注意，添加如下函数

  1. bool MysqlMgr::UpdateUserIcon(int uid, const std::string& icon) {
  2.     return _dao.UpdateHeadInfo(uid, icon);
  3. }

Dao层面实现更新头像逻辑

  1. bool MysqlDao::UpdateHeadInfo(int uid, const std::string& icon)
  2. {
  3.     auto con = pool_->getConnection();
  4.     if (!con) {
  5.         return false;
  6.     }
  7.     Defer defer([this, &con]() {
  8.         pool_->returnConnection(std::move(con));
  9.         });
  11.     auto& conn = con->_con;
  12.     try {
  13.         std::string update_sql =
  14.             "UPDATE user SET icon = ? WHERE uid = ?;";
  16.         std::unique_ptr<sql::PreparedStatement> pstmt(conn->prepareStatement(update_sql));
  17.         pstmt->setString(1, icon);
  18.         pstmt->setInt64(2, uid);
  20.         int affected_rows = pstmt->executeUpdate();
  22.         // 检查是否有行被更新（可选）
  23.         if (affected_rows == 0) {
  24.             std::cerr << "No user found with uid: " << uid << std::endl;
  25.             return false;
  26.         }
  28.         return true;
  29.     }
  30.     catch (sql::SQLException& e) {
  31.         std::cerr << "SQLException in UpdateHeadInfo: " << e.what() << std::endl;
  32.         return false;
  33.     }
  34.     return false;
  35. }

封装异步回调

之前我们处理文件上传是异步方式，将要保存的文件投递给消息队列，交给独立线程后台保存。我们没有等待处理完成就直接将消息回传给客户端，这么做不是很好，所以改为异步方式，简单的方式就是通过回调函数处理，或者包装一个future等待。这里考虑保留异步结构，所以还是用回调处理

  1. struct FileTask {
  2.     FileTask(std::shared_ptr<CSession> session, int uid, std::string path, std::string name,
  3.         int seq, int total_size, int trans_size, int last, 
  4.         std::string file_data,
  5.         std::function<void(const Json::Value&)> callback) :_session(session), _uid(uid),
  6.         _seq(seq), _path(path), _name(name), _total_size(total_size),
  7.         _trans_size(trans_size), _last(last), _file_data(file_data), _callback(callback)
  8.     {}
  9.     ~FileTask(){}
  10.     std::shared_ptr<CSession> _session;
  11.     int _uid;
  12.     int _seq ;
  13.     std::string _path;
  14.     std::string _name ;
  15.     int _total_size ;
  16.     int _trans_size ;
  17.     int _last ;
  18.     std::string _file_data;
  19.     std::function<void(const Json::Value&)>  _callback;  //添加回调函数
  20. };

改进后的处理

  1.     _fun_callbacks[ID_UPLOAD_HEAD_ICON_REQ] = [this](shared_ptr<CSession> session, const short& msg_id,
  2.         const string& msg_data) {
  3.             Json::Reader reader;
  4.             Json::Value root;
  5.             reader.parse(msg_data, root);
  6.             auto md5 = root["md5"].asString();
  7.             auto seq = root["seq"].asInt();
  8.             auto name = root["name"].asString();
  9.             auto total_size = root["total_size"].asInt();
  10.             auto trans_size = root["trans_size"].asInt();
  11.             auto last = root["last"].asInt();
  12.             auto file_data = root["data"].asString();
  13.             auto uid = root["uid"].asInt();
  14.             auto token = root["token"].asString();
  15.             auto last_seq = root["last_seq"].asInt();
  16.             //转化为字符串
  17.             auto uid_str = std::to_string(uid);
  19.             auto file_path = ConfigMgr::Inst().GetFileOutPath();
  20.             auto file_path_str = (file_path / uid_str / name).string();
  21.             Json::Value  rtvalue;
  22.             auto callback = [=](const Json::Value& result) {
  24.                 // 在异步任务完成后调用
  25.                 Json::Value rtvalue = result;
  26.                 rtvalue["error"] = ErrorCodes::Success;
  27.                 rtvalue["total_size"] = total_size;
  28.                 rtvalue["seq"] = seq;
  29.                 rtvalue["name"] = name;
  30.                 rtvalue["trans_size"] = trans_size;
  31.                 rtvalue["last"] = last;
  32.                 rtvalue["md5"] = md5;
  33.                 rtvalue["uid"] = uid;
  34.                 rtvalue["last_seq"] = last_seq;
  35.                 std::string return_str = rtvalue.toStyledString();
  36.                 session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  37.             };
  39.             //第一个包校验一下token是否合理
  40.             if (seq == 1) {
  41.                 //从redis获取用户token是否正确
  42.                 std::string uid_str = std::to_string(uid);
  43.                 std::string token_key = USERTOKENPREFIX + uid_str;
  44.                 std::string token_value = "";
  45.                 bool success = RedisMgr::GetInstance()->Get(token_key, token_value);
  46.                 if (!success) {
  47.                     rtvalue["error"] = ErrorCodes::UidInvalid;
  48.                     std::string return_str = rtvalue.toStyledString();
  49.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  50.                     return;
  51.                 }
  53.                 if (token_value != token) {
  54.                     rtvalue["error"] = ErrorCodes::TokenInvalid;
  55.                     std::string return_str = rtvalue.toStyledString();
  56.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  57.                     return;
  58.                 }
  59.             }
  61.             // 使用 std::hash 对字符串进行哈希
  62.             std::hash<std::string> hash_fn;
  63.             size_t hash_value = hash_fn(name); // 生成哈希值
  64.             int index = hash_value % FILE_WORKER_COUNT;
  65.             std::cout << "Hash value: " << hash_value << std::endl;
  67.             //第一个包
  68.             if (seq == 1) {
  69.                 //构造数据存储
  70.                 auto file_info = std::make_shared<FileInfo>();
  71.                 file_info->_file_path_str = file_path_str;
  72.                 file_info->_name = name;
  73.                 file_info->_seq = seq;
  74.                 file_info->_total_size = total_size;
  75.                 file_info->_trans_size = trans_size;
  76.                 //LogicSystem::GetInstance()->AddMD5File(md5, file_info);
  77.                 //改为用redis存储
  78.                 bool success = RedisMgr::GetInstance()->SetFileInfo(name, file_info);
  79.                 if (!success) {
  80.                     rtvalue["error"] = ErrorCodes::FileSaveRedisFailed;
  81.                     std::string return_str = rtvalue.toStyledString();
  82.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  83.                     return;
  84.                 }
  85.             }
  86.             else {
  87.                 //auto file_info = LogicSystem::GetInstance()->GetFileInfo(md5);
  88.                 //改为从redis中加载
  89.                 auto file_info = RedisMgr::GetInstance()->GetFileInfo(name);
  90.                 if (file_info == nullptr) {
  91.                     rtvalue["error"] = ErrorCodes::FileNotExists;
  92.                     std::string return_str = rtvalue.toStyledString();
  93.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  94.                     return;
  95.                 }
  96.                 file_info->_seq = seq;
  97.                 file_info->_trans_size = trans_size;
  98.                 bool success = RedisMgr::GetInstance()->SetFileInfo(name, file_info);
  99.                 if (!success) {
  100.                     rtvalue["error"] = ErrorCodes::FileSaveRedisFailed;
  101.                     std::string return_str = rtvalue.toStyledString();
  102.                     session->Send(return_str, ID_UPLOAD_HEAD_ICON_RSP);
  103.                     return;
  104.                 }
  105.             }
  108.             FileSystem::GetInstance()->PostMsgToQue(
  109.                 std::make_shared<FileTask>(session, uid, file_path_str, name, seq, total_size,
  110.                     trans_size, last, file_data, callback),
  111.                 index
  112.             );
  114.     };

callback是我们封装的回调函数，投递给FileTask, 将来在后台线程处理FileTask时回调。

  1. void FileWorker::task_callback(std::shared_ptr<FileTask> task)
  2. {
  3.     // 解码
  4.     std::string decoded = base64_decode(task->_file_data);
  6.     auto file_path_str = task->_path;
  7.     auto last = task->_last;
  8.     //std::cout << "file_path_str is " << file_path_str << std::endl;
  10.     boost::filesystem::path file_path(file_path_str);
  11.     boost::filesystem::path dir_path = file_path.parent_path();
  12.     // 获取完整文件名（包含扩展名）
  13.     std::string filename = file_path.filename().string();
  14.     Json::Value result;
  15.     // Check if directory exists, if not, create it
  16.     if (!boost::filesystem::exists(dir_path)) {
  17.         if (!boost::filesystem::create_directories(dir_path)) {
  18.             std::cerr << "Failed to create directory: " << dir_path.string() << std::endl;
  19.             result["error"] = ErrorCodes::FileNotExists;
  20.             task->_callback(result);
  21.             return;
  22.         }
  23.     }
  26.     std::ofstream outfile;
  27.     //第一个包
  28.     if (task->_seq == 1) {
  29.         // 打开文件，如果存在则清空，不存在则创建
  30.         outfile.open(file_path_str, std::ios::binary | std::ios::trunc);
  31.     }
  32.     else {
  33.         // 保存为文件
  34.         outfile.open(file_path_str, std::ios::binary | std::ios::app);
  35.     }
  38.     if (!outfile) {
  39.         std::cerr << "无法打开文件进行写入。" << std::endl;
  40.         result["error"] = ErrorCodes::FileWritePermissionFailed;
  41.         task->_callback(result);
  42.         return ;
  43.     }
  45.     outfile.write(decoded.data(), decoded.size());
  46.     if (!outfile) {
  47.         std::cerr << "写入文件失败。" << std::endl;
  48.         result["error"] = ErrorCodes::FileWritePermissionFailed;
  49.         task->_callback(result);
  50.         return ;
  51.     }
  53.     outfile.close();
  54.     if (last) {
  55.         std::cout << "文件已成功保存为: " << task->_name << std::endl;
  56.         //更新头像
  57.         MysqlMgr::GetInstance()->UpdateUserIcon(task->_uid, filename);
  58.         //获取用户信息
  59.         auto user_info = MysqlMgr::GetInstance()->GetUser(task->_uid);
  60.         if (user_info == nullptr) {
  61.             return ;
  62.         }
  64.         //将数据库内容写入redis缓存
  65.         Json::Value redis_root;
  66.         redis_root["uid"] = task->_uid;
  67.         redis_root["pwd"] = user_info->pwd;
  68.         redis_root["name"] = user_info->name;
  69.         redis_root["email"] = user_info->email;
  70.         redis_root["nick"] = user_info->nick;
  71.         redis_root["desc"] = user_info->desc;
  72.         redis_root["sex"] = user_info->sex;
  73.         redis_root["icon"] = user_info->icon;
  74.         std::string base_key = USER_BASE_INFO + std::to_string(task->_uid);
  75.         RedisMgr::GetInstance()->Set(base_key, redis_root.toStyledString());
  76.     }
  78.     if (task->_callback) {
  79.         task->_callback(result);
  80.     }
  82. }

客户端上传逻辑修改

  1. //上传头像
  2. void UserInfoPage::slot_up_load()
  3. {
  4.     // 1. 让对话框也能选 *.webp
  5.     QString filename = QFileDialog::getOpenFileName(
  6.         this,
  7.         tr("选择图片"),
  8.         QString(),
  9.         tr("图片文件 (*.png *.jpg *.jpeg *.bmp *.webp)")
  10.     );
  11.     if (filename.isEmpty())
  12.         return;
  14.     // 2. 直接用 QPixmap::load() 加载，无需手动区分格式
  15.     QPixmap inputImage;
  16.     if (!inputImage.load(filename)) {
  17.         QMessageBox::critical(
  18.             this,
  19.             tr("错误"),
  20.             tr("加载图片失败！请确认已部署 WebP 插件。"),
  21.             QMessageBox::Ok
  22.         );
  23.         return;
  24.     }
  26.     QPixmap image = ImageCropperDialog::getCroppedImage(filename, 600, 400, CropperShape::CIRCLE);
  27.     if (image.isNull())
  28.         return;
  30.     QPixmap scaledPixmap = image.scaled( ui->head_lb->size(), Qt::KeepAspectRatio, Qt::SmoothTransformation); // 将图片缩放到label的大小
  31.     ui->head_lb->setPixmap(scaledPixmap); // 将缩放后的图片设置到QLabel上
  32.     ui->head_lb->setScaledContents(true); // 设置QLabel自动缩放图片内容以适应大小
  34.     QString storageDir = QStandardPaths::writableLocation(
  35.                              QStandardPaths::AppDataLocation);
  36.     // 2. 在其下再建一个 avatars 子目录
  37.     QDir dir(storageDir);
  38.     if (!dir.exists("avatars")) {
  39.         if (!dir.mkpath("avatars")) {
  40.             qWarning() << "无法创建 avatars 目录：" << dir.filePath("avatars");
  41.             QMessageBox::warning(
  42.                 this,
  43.                 tr("错误"),
  44.                 tr("无法创建存储目录，请检查权限或磁盘空间。")
  45.             );
  46.             return;
  47.         }
  48.     }
  49.     // 3. 拼接最终的文件名 head.png
  50.     QString file_name = generateUniqueIconName();
  51.     QString filePath = dir.filePath("avatars" +
  52.                           QString(QDir::separator()) + file_name);
  54.     // 4. 保存 scaledPixmap 为 PNG（无损、最高质量）
  55.     if (!scaledPixmap.save(filePath, "PNG")) {
  56.         QMessageBox::warning(
  57.             this,
  58.             tr("保存失败"),
  59.             tr("头像保存失败，请检查权限或磁盘空间。")
  60.         );
  61.     } else {
  62.         qDebug() << "头像已保存到：" << filePath;
  63.         // 以后读取直接用同一路径：storageDir/avatars/head.png
  64.     }
  66.     //实现头像上传
  67.     QFile file(filePath);
  68.     if(!file.open(QIODevice::ReadOnly)){
  69.         qWarning() << "Could not open file:" << file.errorString();
  70.         return;
  71.     }
  73.     //保存当前文件位置指针
  74.     qint64 originalPos = file.pos();
  76.     QCryptographicHash hash(QCryptographicHash::Md5);
  77.     if (!hash.addData(&file)) {
  78.             qWarning() << "Failed to read data from file:" << filePath;
  79.             return ;
  80.     }
  82.     // 5. 转化为16进制字符串
  83.     QString file_md5 = hash.result().toHex(); // 返回十六进制字符串
  85.     //读取文件内容并发送
  86.     QByteArray buffer;
  87.     int seq = 0;
  89.     //创建QFileInfo 对象
  90.     auto fileInfo = std::make_shared<QFileInfo>(filePath);
  91.     //获取文件名
  92.     QString fileName = fileInfo->fileName();
  93.     //文件名
  94.     qDebug() << "文件名是: " << fileName;
  96.     //获取文件大小
  97.     int total_size = fileInfo->size();
  98.     //最后一个发送序列
  99.     int last_seq = 0;
  100.     //获取最后一个发送序列
  101.     if(total_size % MAX_FILE_LEN){
  102.         last_seq = (total_size / MAX_FILE_LEN) +1;
  103.     }else{
  104.         last_seq = total_size / MAX_FILE_LEN;
  105.     }
  107.     // 恢复文件指针到原来的位置
  108.     file.seek(originalPos);
  110.     //每次读取MAX_FILE_LEN字节并发送
  111.     buffer = file.read(MAX_FILE_LEN);
  113.     QJsonObject jsonObj;
  114.     //将文件内容转化为Base64 编码(可选)
  115.     QString base64Data = buffer.toBase64();
  116.     ++seq;
  117.     jsonObj["md5"] = file_md5;
  118.     jsonObj["name"] = file_name;
  119.     jsonObj["seq"] = seq;
  120.     jsonObj["trans_size"] = buffer.size() + (seq - 1) * MAX_FILE_LEN;
  121.     jsonObj["total_size"] = total_size;
  122.     jsonObj["token"] = UserMgr::GetInstance()->GetToken();
  123.     jsonObj["uid"] = UserMgr::GetInstance()->GetUid();
  125.     if (buffer.size() + (seq - 1) * MAX_FILE_LEN == total_size) {
  126.         jsonObj["last"] = 1;
  127.     } else {
  128.         jsonObj["last"] = 0;
  129.     }
  131.     jsonObj["data"] = base64Data;
  132.     jsonObj["last_seq"] = last_seq;
  133.     QJsonDocument doc(jsonObj);
  134.     auto send_data = doc.toJson();
  135.     //将md5信息和文件信息关联存储
  136.     UserMgr::GetInstance()->AddNameFile(file_name, fileInfo);
  137.     //发送消息
  138.     FileTcpMgr::GetInstance()->SendData(ID_UPLOAD_HEAD_ICON_REQ, send_data);
  139.     file.close();
  140. }

客户端加载头像

服务器将上传的头像信息保存为url更新到mysql中，接下来客户端登录需要加载新的头像

在ChatDialog的构造函数中将头像加载逻辑修改为

  1.     //模拟加载自己头像
  2.     QString head_icon = UserMgr::GetInstance()->GetIcon();
  3.     //使用正则表达式检查是否使用默认头像
  4.     QRegularExpression regex("^:/res/head_(\\d+)\\.jpg$");
  5.     QRegularExpressionMatch match = regex.match(head_icon);
  6.     if (match.hasMatch()) {
  7.         // 如果是默认头像（:/res/head_X.jpg 格式）
  8.         QPixmap pixmap(head_icon); // 加载默认头像图片
  9.         QPixmap scaledPixmap = pixmap.scaled(ui->side_head_lb->size(), Qt::KeepAspectRatio, Qt::SmoothTransformation);
  10.         ui->side_head_lb->setPixmap(scaledPixmap); // 将缩放后的图片设置到QLabel上
  11.         ui->side_head_lb->setScaledContents(true); // 设置QLabel自动缩放图片内容以适应大小
  12.     }
  13.     else {
  14.         // 如果是用户上传的头像，获取存储目录
  15.         QString storageDir = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
  16.         QDir avatarsDir(storageDir + "/avatars");
  18.         // 确保目录存在
  19.         if (avatarsDir.exists()) {
  20.             QString avatarPath = avatarsDir.filePath(QFileInfo(head_icon).fileName()); // 获取上传头像的完整路径
  21.             QPixmap pixmap(avatarPath); // 加载上传的头像图片
  22.             if (!pixmap.isNull()) {
  23.                 QPixmap scaledPixmap = pixmap.scaled(ui->side_head_lb->size(), Qt::KeepAspectRatio, Qt::SmoothTransformation);
  24.                 ui->side_head_lb->setPixmap(scaledPixmap);
  25.                 ui->side_head_lb->setScaledContents(true);
  26.             }
  27.             else {
  28.                 qWarning() << "无法加载上传的头像：" << avatarPath;
  29.             }
  30.         }
  31.         else {
  32.             qWarning() << "头像存储目录不存在：" << avatarsDir.path();
  33.         }
  34.     }

聊天页面也需要修改头像加载逻辑

  1. void ChatPage::AppendChatMsg(std::shared_ptr<ChatDataBase> msg)
  2. {
  3.     auto self_info = UserMgr::GetInstance()->GetUserInfo();
  4.     ChatRole role;
  5.     if (msg->GetSendUid() == self_info->_uid) {
  6.         role = ChatRole::Self;
  7.         ChatItemBase* pChatItem = new ChatItemBase(role);
  9.         pChatItem->setUserName(self_info->_name);
  11.         // 使用正则表达式检查是否是默认头像
  12.         QRegularExpression regex("^:/res/head_(\\d+)\\.jpg$");
  13.         QRegularExpressionMatch match = regex.match(self_info->_icon);
  14.         if (match.hasMatch()) {
  15.             pChatItem->setUserIcon(QPixmap(self_info->_icon));
  16.         }
  17.         else {
  18.             // 如果是用户上传的头像，获取存储目录
  19.             QString storageDir = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
  20.             QDir avatarsDir(storageDir + "/avatars");
  22.             // 确保目录存在
  23.             if (avatarsDir.exists()) {
  24.                 QString avatarPath = avatarsDir.filePath(QFileInfo(self_info->_icon).fileName()); // 获取上传头像的完整路径
  25.                 QPixmap pixmap(avatarPath); // 加载上传的头像图片
  26.                 if (!pixmap.isNull()) {
  27.                     pChatItem->setUserIcon(pixmap);
  28.                 }
  29.                 else {
  30.                     qWarning() << "无法加载上传的头像：" << avatarPath;
  31.                 }
  32.             }
  33.             else {
  34.                 qWarning() << "头像存储目录不存在：" << avatarsDir.path();
  35.             }
  36.         }
  38.         QWidget* pBubble = nullptr;
  39.         if (msg->GetMsgType() == ChatMsgType::TEXT) {
  40.             pBubble = new TextBubble(role, msg->GetMsgContent());
  41.         }
  43.         pChatItem->setWidget(pBubble);
  44.         auto status = msg->GetStatus();
  45.         pChatItem->setStatus(status);
  46.         ui->chat_data_list->appendChatItem(pChatItem);
  47.         if (status == 0) {
  48.             _unrsp_item_map[msg->GetUniqueId()] = pChatItem;
  49.         }
  50.     }
  51.     else {
  52.         role = ChatRole::Other;
  53.         ChatItemBase* pChatItem = new ChatItemBase(role);
  54.         auto friend_info = UserMgr::GetInstance()->GetFriendById(msg->GetSendUid());
  55.         if (friend_info == nullptr) {
  56.             return;
  57.         }
  58.         pChatItem->setUserName(friend_info->_name);
  60.         // 使用正则表达式检查是否是默认头像
  61.         QRegularExpression regex("^:/res/head_(\\d+)\\.jpg$");
  62.         QRegularExpressionMatch match = regex.match(friend_info->_icon);
  63.         if (match.hasMatch()) {
  64.             pChatItem->setUserIcon(QPixmap(friend_info->_icon));
  65.         }
  66.         else {
  67.             // 如果是用户上传的头像，获取存储目录
  68.             QString storageDir = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
  69.             QDir avatarsDir(storageDir + "/avatars");
  71.             // 确保目录存在
  72.             if (avatarsDir.exists()) {
  73.                 QString avatarPath = avatarsDir.filePath(QFileInfo(friend_info->_icon).fileName()); // 获取上传头像的完整路径
  74.                 QPixmap pixmap(avatarPath); // 加载上传的头像图片
  75.                 if (!pixmap.isNull()) {
  76.                     pChatItem->setUserIcon(pixmap);
  77.                 }
  78.                 else {
  79.                     qWarning() << "无法加载上传的头像：" << avatarPath;
  80.                 }
  81.             }
  82.             else {
  83.                 qWarning() << "头像存储目录不存在：" << avatarsDir.path();
  84.             }
  85.         }
  87.         QWidget* pBubble = nullptr;
  88.         if (msg->GetMsgType() == ChatMsgType::TEXT) {
  89.             pBubble = new TextBubble(role, msg->GetMsgContent());
  90.         }
  91.         pChatItem->setWidget(pBubble);
  92.         auto status = msg->GetStatus();
  93.         pChatItem->setStatus(status);
  94.         ui->chat_data_list->appendChatItem(pChatItem);
  95.         if (status == 0) {
  96.             _unrsp_item_map[msg->GetUniqueId()] = pChatItem;
  97.         }
  98.     }
  101. }

聊天列表中加载头像逻辑修改一下

  1. void ChatUserWid::SetChatData(std::shared_ptr<ChatThreadData> chat_data) {
  2.     _chat_data = chat_data;
  3.     auto other_id = _chat_data->GetOtherId();
  4.     auto other_info = UserMgr::GetInstance()->GetFriendById(other_id);
  5.     // 加载图片
  7.     QString head_icon = UserMgr::GetInstance()->GetIcon();
  9.     // 使用正则表达式检查是否是默认头像
  10.     QRegularExpression regex("^:/res/head_(\\d+)\\.jpg$");
  11.     QRegularExpressionMatch match = regex.match(other_info->_icon);
  13.     if (match.hasMatch()) {
  14.         // 如果是默认头像（:/res/head_X.jpg 格式）
  15.         QPixmap pixmap(other_info->_icon); // 加载默认头像图片
  16.         QPixmap scaledPixmap = pixmap.scaled(ui->icon_lb->size(), Qt::KeepAspectRatio, Qt::SmoothTransformation);
  17.         ui->icon_lb->setPixmap(scaledPixmap); // 将缩放后的图片设置到QLabel上
  18.         ui->icon_lb->setScaledContents(true); // 设置QLabel自动缩放图片内容以适应大小
  19.     }
  20.     else {
  21.         // 如果是用户上传的头像，获取存储目录
  22.         QString storageDir = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
  23.         QDir avatarsDir(storageDir + "/avatars");
  25.         // 确保目录存在
  26.         if (avatarsDir.exists()) {
  27.             QString avatarPath = avatarsDir.filePath(QFileInfo(other_info->_icon).fileName()); // 获取上传头像的完整路径
  28.             QPixmap pixmap(avatarPath); // 加载上传的头像图片
  29.             if (!pixmap.isNull()) {
  30.                 QPixmap scaledPixmap = pixmap.scaled(ui->icon_lb->size(), Qt::KeepAspectRatio, Qt::SmoothTransformation);
  31.                 ui->icon_lb->setPixmap(scaledPixmap);
  32.                 ui->icon_lb->setScaledContents(true);
  33.             }
  34.             else {
  35.                 qWarning() << "无法加载上传的头像：" << avatarPath;
  36.             }
  37.         }
  38.         else {
  39.             qWarning() << "头像存储目录不存在：" << avatarsDir.path();
  40.         }
  41.     }
  43.     ui->user_name_lb->setText(other_info->_name);
  45.     ui->user_chat_lb->setText(chat_data->GetLastMsg());
  46. }

用户信息加载头像

  1. UserInfoPage::UserInfoPage(QWidget *parent) :
  2.     QWidget(parent),
  3.     ui(new Ui::UserInfoPage)
  4. {
  5.     ui->setupUi(this);
  6.     auto icon = UserMgr::GetInstance()->GetIcon();
  7.     qDebug() << "icon is " << icon ;
  9.     //使用正则表达式检查是否使用默认头像
  10.     QRegularExpression regex("^:/res/head_(\\d+)\\.jpg$");
  11.     QRegularExpressionMatch match = regex.match(icon);
  12.     if (match.hasMatch()) {
  13.         QPixmap pixmap(icon);
  14.         QPixmap scaledPixmap = pixmap.scaled(ui->head_lb->size(), 
  15.             Qt::KeepAspectRatio, Qt::SmoothTransformation); // 将图片缩放到label的大小
  16.         ui->head_lb->setPixmap(scaledPixmap); // 将缩放后的图片设置到QLabel上
  17.         ui->head_lb->setScaledContents(true); // 设置QLabel自动缩放图片内容以适应大小
  18.     }
  19.     else {
  20.         // 如果是用户上传的头像，获取存储目录
  21.         QString storageDir = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation);
  22.         QDir avatarsDir(storageDir + "/avatars");
  24.         // 确保目录存在
  25.         if (avatarsDir.exists()) {
  26.             QString avatarPath = avatarsDir.filePath(QFileInfo(icon).fileName()); // 获取上传头像的完整路径
  27.             QPixmap pixmap(avatarPath); // 加载上传的头像图片
  28.             if (!pixmap.isNull()) {
  29.                 QPixmap scaledPixmap = pixmap.scaled(ui->head_lb->size(),
  30.                     Qt::KeepAspectRatio, Qt::SmoothTransformation); // 将图片缩放到label的大小
  31.                 ui->head_lb->setPixmap(scaledPixmap); // 将缩放后的图片设置到QLabel上
  32.                 ui->head_lb->setScaledContents(true); // 设置QLabel自动缩放图片内容以适应大小
  33.             }
  34.             else {
  35.                 qWarning() << "无法加载上传的头像：" << avatarPath;
  36.             }
  37.         }
  38.         else {
  39.             qWarning() << "头像存储目录不存在：" << avatarsDir.path();
  40.         }
  41.     }
  45.     //获取nick
  46.     auto nick = UserMgr::GetInstance()->GetNick();
  47.     //获取name
  48.     auto name = UserMgr::GetInstance()->GetName();
  49.     //描述
  50.     auto desc = UserMgr::GetInstance()->GetDesc();
  51.     ui->nick_ed->setText(nick);
  52.     ui->name_ed->setText(name);
  53.     ui->desc_ed->setText(desc);
  54.     //连接上
  55.     connect(ui->up_btn, &QPushButton::clicked, this, &UserInfoPage::slot_up_load);
  56. }

测试效果

image-20250923094811744

热门评论
点击查看更多评论 没有更多评论了,点击收起 展开评论

热门文章

  1. 使用hexo搭建个人博客
    喜欢(533) 浏览(13711)

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

  3. MarkDown在线编辑器
    喜欢(514) 浏览(15557)

  4. 聊天项目(28) 分布式服务通知好友申请
    喜欢(507) 浏览(7021)

  5. vscode搭建windows C++开发环境
    喜欢(596) 浏览(96518)

最新评论

  1. 解决博客回复区被脚本注入的问题 secondtonone1:走到现在我忽然明白一个道理，无论工作也好生活也罢，最重要的是开心，即使一份安稳的工作不能给我带来事业上的积累也要合理的舍弃，所以我还是想去做喜欢的方向。
  2. 处理网络粘包问题 zyouth: //消息的长度小于头部规定的长度，说明数据未收全，则先将部分消息放到接收节点里 if (bytes_transferred < data_len) { memcpy(_recv_msg_node->_data + _recv_msg_node->_cur_len, _data + copy_len, bytes_transferred); _recv_msg_node->_cur_len += bytes_transferred; ::memset(_data, 0, MAX_LENGTH); _socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH), std::bind(&CSession::HandleRead, this, std::placeholders::_1, std::placeholders::_2, shared_self)); //头部处理完成 _b_head_parse = true; return; } 把_b_head_parse = true;放在_socket.async_read_some前面是不是更好
  3. C++ 线程池原理和实现 mzx2023:两种方法解决，一种是改排序算法，就是当线程耗尽的时候，使用普通递归，另一种是当在线程池commit的时候，判断线程是否耗尽，耗尽的话就直接当前线程执行task
  4. 利用指针和容器实现文本查询 越今朝:应该添加一个过滤功能以解决部分单词无法被查询的问题: eg: "I am a teacher."中的teacher无法被查询，因为在示例代码中teacher.被解释为一个单词从而忽略了teacher本身。
  5. 无锁并发队列 TenThousandOne:_head  和 _tail  替换为原子变量。那里pop的逻辑，val = _data[h] 可以移到循环外面吗

其他链接

  1. GitHub
  2. 个人博客
  3. 博客园
  4. CSDN

个人公众号

个人微信