上传数据至服务端：MFC多线程、文件压缩、上传数据、进度条

// ReleaseProjDlg.cpp : 实现文件 // #include "pch.h" #include "ProjOnlineRelease.h" #include "ReleaseProjDlg.h" #include "afxdialogex.h" #include "Lib_Global.h" #include "SetReleaseStatusDlg.h" #include "ResourceSwitch.h" volatile UINT g_nFileCount = 0; volatile UINT g_nOkCount = 0; volatile ULONGLONG g_nBeginTime = 0; std::string g_sUploadFileUrl = "http://hc.gongge.com.cn:8091/api/file/uploadFiles"; std::string g_sAccess_token = "access_token: eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ0aW1lU3RhbXAiOjE2NDQ5MDE2MTA2MjcsImdJZCI6IjY3NDQ4MDk0OTMxOTAwNzA1OTgifQ._dzEiGA9IZTAtYsMRNgJx6SzupAEVCZ10VWgKexibDA"; std::string g_sUploadPartFileUrl = "http://hc.gongge.com.cn:8091/api/file/uploadBigFile"; std::string g_sProjId = "1526aedf-10d9-406e-acc9-bd9195c3b367"; std::string g_sProjDir; volatile bool g_bBeginUpload = false; volatile bool g_bEndUploadFile = false; volatile bool g_bEndUploadMdsFile = false; volatile bool g_bEndUploadPartFile = false; volatile bool g_bExitUpload = false; HANDLE g_hMutex; // 线程互斥 //#define THRED_NUM 5 DWORD WINAPI UploadFileFunc(LPVOID lpParam) { threadInfo* pInfo = (threadInfo*)lpParam; int nCodeNum = pInfo->nCodeNum; std::vector<vecFilesInfo>& vecInfo = pInfo->vecInfo; for (auto iterFileGroup = vecInfo.begin(); iterFileGroup != vecInfo.end(); ++iterFileGroup) { if (g_bExitUpload) { break; } std::vector<FileInfo>& vecFileInfo = *iterFileGroup; int nStatus = -1; if (!CUploadUtils::CurlPostUploadFile(g_sUploadFileUrl.c_str(),g_sAccess_token.c_str(),g_sProjId.c_str(),false,vecFileInfo,nStatus)) { CString sError; sError.Format(_T("ZLY::小文件组上传失败,dStatusCode= %d！"),(int)nStatus); OutputDebugString(sError); // 更新进度条 WaitForSingleObject(g_hMutex, INFINITE); g_nOkCount += (int)vecFileInfo.size(); ULONGLONG dCurrentTime = GetTickCount64(); double dTime = ((double)(dCurrentTime - g_nBeginTime))/1000; CString sTip; sTip.Format(_T("当前处理完成个数 %d / %d，耗时 %.3f 秒,剩余 %d 个!"),g_nOkCount,g_nFileCount,dTime,g_nFileCount-g_nOkCount); if (pInfo->ctrlTip) pInfo->ctrlTip->SetWindowText(sTip); if (pInfo->pctrlProgress) pInfo->pctrlProgress->SetPos(g_nOkCount); ReleaseMutex(g_hMutex); continue; } // 更新进度条 WaitForSingleObject(g_hMutex, INFINITE); g_nOkCount += (int)vecFileInfo.size(); ULONGLONG dCurrentTime = GetTickCount64(); double dTime = ((double)(dCurrentTime - g_nBeginTime))/1000; CString sTip; sTip.Format(_T("当前上传文件完成个数 %d / %d，耗时 %.3f 秒,剩余 %d 个!"),g_nOkCount,g_nFileCount,dTime,g_nFileCount-g_nOkCount); if (pInfo->ctrlTip) pInfo->ctrlTip->SetWindowText(sTip); if (pInfo->pctrlProgress) pInfo->pctrlProgress->SetPos(g_nOkCount); ReleaseMutex(g_hMutex); } if (g_nOkCount == g_nFileCount) { g_bEndUploadFile = true; } return 0; } DWORD WINAPI CompressUploadFunc(LPVOID lpParam) { threadInfo* pInfo = (threadInfo*)lpParam; int nCodeNum = pInfo->nCodeNum; std::vector<vecFilesInfo>& vecInfo = pInfo->vecInfo; for (auto iterFileGroup = vecInfo.begin(); iterFileGroup != vecInfo.end(); ++iterFileGroup) { if (g_bExitUpload) { break; } // 压缩文件 std::vector<FileInfo> vecFileInfo; vecFileInfo.clear(); std::vector<FileInfo>& vecFileInfoTemp = *iterFileGroup; for (auto itorTemp = vecFileInfoTemp.begin(); itorTemp != vecFileInfoTemp.end(); itorTemp++) { if (g_bExitUpload) { return -1; } wstring wsFolder = itorTemp->strFilePath; string sFolder = CUploadUtils::wstring2string(wsFolder); CString strFolder = sFolder.c_str(); CString strFolderZip = strFolder + _T(".zip"); CString sMsg; sMsg.Format(_T("ZLY:当前正在压缩处理文件夹名称= %s!"),strFolder); OutputDebugString(sMsg); // 文件压缩 CString strError; if (!CUploadUtils::ZipPackage(strFolder,strFolderZip,strError)) { CString sTip; sTip.Format(_T("当前文件压缩出现错误,文件名称= !"),strFolder); if (pInfo->ctrlTip) pInfo->ctrlTip->SetWindowText(sTip); sMsg.Format(_T("ZLY:文件压缩出现错误,文件名称= %s!"),strFolder); OutputDebugString(sMsg); continue; } FileInfo stufileTemp; string sTemp = strFolderZip; stufileTemp.strFilePath = CUploadUtils::string2wstring(sTemp); stufileTemp.strFileDir = itorTemp->strFileDir; vecFileInfo.push_back(stufileTemp); } Sleep(200); // 文件上传 int nStatus = -1; if (!CUploadUtils::CurlPostUploadFile(g_sUploadFileUrl.c_str(),g_sAccess_token.c_str(),g_sProjId.c_str(),true,vecFileInfo,nStatus)) { CString sError; sError.Format(_T("ZLY::压缩文件组上传失败,dStatusCode= %d！"),(int)nStatus); OutputDebugString(sError); // 更新进度条 WaitForSingleObject(g_hMutex, INFINITE); g_nOkCount += (int)vecFileInfoTemp.size(); ULONGLONG dCurrentTime = GetTickCount64(); double dTime = ((double)(dCurrentTime - g_nBeginTime))/1000; CString sTip; sTip.Format(_T("当前处理完成个数 %d / %d，耗时 %.3f 秒,剩余 %d 个!"),g_nOkCount,g_nFileCount,dTime,g_nFileCount-g_nOkCount); if (pInfo->ctrlTip) pInfo->ctrlTip->SetWindowText(sTip); if (pInfo->pctrlProgress) pInfo->pctrlProgress->SetPos(g_nOkCount); ReleaseMutex(g_hMutex); // 删除压缩包 for (auto itorTemp = vecFileInfo.begin(); itorTemp != vecFileInfo.end(); itorTemp++) { wstring wsFolder = itorTemp->strFilePath; string sFolder = CUploadUtils::wstring2string(wsFolder); CString strFolderZip = sFolder.c_str(); ::DeleteFile(strFolderZip); } continue; } //Sleep(1000); // 更新进度条 WaitForSingleObject(g_hMutex, INFINITE); g_nOkCount += (int)vecFileInfoTemp.size(); ULONGLONG dCurrentTime = GetTickCount64(); double dTime = ((double)(dCurrentTime - g_nBeginTime))/1000; CString sTip; sTip.Format(_T("当前上传完成个数 %d / %d，耗时 %.3f 秒,剩余 %d 个!"),g_nOkCount,g_nFileCount,dTime,g_nFileCount-g_nOkCount); if (pInfo->ctrlTip) { pInfo->ctrlTip->SetWindowText(sTip); } if (pInfo->pctrlProgress) pInfo->pctrlProgress->SetPos(g_nOkCount); ReleaseMutex(g_hMutex); // 删除压缩包 for (auto itorTemp = vecFileInfo.begin(); itorTemp != vecFileInfo.end(); itorTemp++) { wstring wsFolder = itorTemp->strFilePath; string sFolder = CUploadUtils::wstring2string(wsFolder); CString strFolderZip = sFolder.c_str(); ::DeleteFile(strFolderZip); } } if (g_nOkCount == g_nFileCount) { g_bEndUploadMdsFile = true; } return 0; } DWORD WINAPI UploadPartFileFunc(LPVOID lpParam) { threadPartInfo* pInfo = (threadPartInfo*)lpParam; for (int i = 0; i < pInfo->vecFileName.size(); ++i) { if (g_bExitUpload) { break; } string sFileName = pInfo->vecFileName[i]; wstring wsFileName; CString sTip; sTip.Format(_T("正在上传大文件: 当前完成 %d / %d个，剩余 %d 个!"),g_nOkCount,g_nFileCount,g_nFileCount-g_nOkCount); pInfo->ctrlUpTip->SetWindowText(sTip); CString strDirPath; CString strPath = sFileName.c_str(); CString strProjDir = g_sProjDir.c_str(); int nIndex = strPath.Find(strProjDir); if (nIndex != -1) { strDirPath = strPath.Mid(nIndex + strProjDir.GetLength()+1, strPath.GetLength() - nIndex - strProjDir.GetLength()-1); std::string sSubPath = strDirPath; wsFileName = CUploadUtils::string2wstring(sSubPath); } CString strFileName = sFileName.c_str(); int nPos = strFileName.ReverseFind('\\'); if (nPos > -1) { strFileName = strFileName.Mid(nPos+1); } if (strFileName.GetLength() > 30) { strFileName = strFileName.Mid(strFileName.GetLength()-30,30); } sTip.Format(_T("正在对大文件 %s进行分片..."),strFileName); pInfo->ctrlDownTip->SetWindowText(sTip); // 拆分文件并上传 std::vector<partFileInfo> vecPartFileInfo; int nShareCount