在 VC++ 客户端中统计下载的网速

源码资源网

要在 VC++ 客户端中统计下载的网速，可以在下载过程中记录已下载的字节数和所用的时间。可以使用 `std::chrono` 库来测量时间，并在每次读取数据后计算当前的下载速度。

以下是修改后的代码示例，添加了下载速度的统计功能：

1. 
   
2. ```cpp
   
3. #include <windows.h>
   
4. #include <winhttp.h>
   
5. #include <iostream>
   
6. #include <fstream>
   
7. #include <chrono>
   
8. #include <thread>
   
9. 
   
10. #pragma comment(lib, "winhttp.lib")
    
11. 
    
12. void DownloadFile(const std::wstring& server, const std::wstring& filePath, const std::wstring& savePath, int speedLimit) {
    
13.     HINTERNET hSession = WinHttpOpen(L"A WinHTTP Example Program/1.0",
    
14.                                      WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
    
15.                                      WINHTTP_NO_PROXY_NAME,
    
16.                                      WINHTTP_NO_PROXY_BYPASS, 0);
    
17. 
    
18.     if (!hSession) {
    
19.         std::cerr << "WinHttpOpen failed." << std::endl;
    
20.         return;
    
21.     }
    
22. 
    
23.     HINTERNET hConnect = WinHttpConnect(hSession, server.c_str(), INTERNET_DEFAULT_HTTP_PORT, 0);
    
24.     if (!hConnect) {
    
25.         std::cerr << "WinHttpConnect failed." << std::endl;
    
26.         WinHttpCloseHandle(hSession);
    
27.         return;
    
28.     }
    
29. 
    
30.     HINTERNET hRequest = WinHttpOpenRequest(hConnect, L"GET", filePath.c_str(),
    
31.                                             NULL, WINHTTP_NO_REFERER,
    
32.                                             WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
    
33.    
    
34.     if (!hRequest) {
    
35.         std::cerr << "WinHttpOpenRequest failed." << std::endl;
    
36.         WinHttpCloseHandle(hConnect);
    
37.         WinHttpCloseHandle(hSession);
    
38.         return;
    
39.     }
    
40. 
    
41.     // 发送请求
    
42.     if (!WinHttpSendRequest(hRequest, WINHTTP_NO_ADDITIONAL_HEADERS, 0,
    
43.                             WINHTTP_NO_REQUEST_DATA, 0, 0, 0)) {
    
44.         std::cerr << "WinHttpSendRequest failed." << std::endl;
    
45.         WinHttpCloseHandle(hRequest);
    
46.         WinHttpCloseHandle(hConnect);
    
47.         WinHttpCloseHandle(hSession);
    
48.         return;
    
49.     }
    
50. 
    
51.     // 等待响应
    
52.     if (!WinHttpReceiveResponse(hRequest, NULL)) {
    
53.         std::cerr << "WinHttpReceiveResponse failed." << std::endl;
    
54.         WinHttpCloseHandle(hRequest);
    
55.         WinHttpCloseHandle(hConnect);
    
56.         WinHttpCloseHandle(hSession);
    
57.         return;
    
58.     }
    
59. 
    
60.     // 打开文件以写入
    
61.     std::ofstream outFile(savePath, std::ios::binary);
    
62.     if (!outFile.is_open()) {
    
63.         std::cerr << "Failed to open file for writing." << std::endl;
    
64.         WinHttpCloseHandle(hRequest);
    
65.         WinHttpCloseHandle(hConnect);
    
66.         WinHttpCloseHandle(hSession);
    
67.         return;
    
68.     }
    
69. 
    
70.     // 读取数据并写入文件
    
71.     DWORD bytesRead;
    
72.     char buffer[4096]; // 4KB 缓冲区
    
73.     long long totalBytesRead = 0; // 总共读取的字节数
    
74. 
    
75.     auto startTime = std::chrono::high_resolution_clock::now(); // 开始时间
    
76. 
    
77.     while (true) {
    
78.         if (!WinHttpReadData(hRequest, buffer, sizeof(buffer), &bytesRead) || bytesRead == 0) {
    
79.             break; // 读取完成或出错
    
80.         }
    
81. 
    
82.         outFile.write(buffer, bytesRead);
    
83.         totalBytesRead += bytesRead;
    
84. 
    
85.         // 限制速度
    
86.         std::this_thread::sleep_for(std::chrono::milliseconds(1000 * bytesRead / speedLimit)); // 限速
    
87. 
    
88.         // 计算并输出下载速度
    
89.         auto currentTime = std::chrono::high_resolution_clock::now();
    
90.         std::chrono::duration<double> elapsed = currentTime - startTime;
    
91. 
    
92.         if (elapsed.count() > 0) {
    
93.             double speed = (totalBytesRead / 1024.0) / elapsed.count(); // KB/s
    
94.             std::cout << "当前下载速度: " << speed << " KB/s" << std::endl;
    
95.         }
    
96.     }
    
97. 
    
98.     // 关闭文件
    
99.     outFile.close();
    
100. 
     
101.     // 清理
     
102.     WinHttpCloseHandle(hRequest);
     
103.     WinHttpCloseHandle(hConnect);
     
104.     WinHttpCloseHandle(hSession);
     
105. 
     
106.     std::cout << "文件下载成功: " << savePath << std::endl;
     
107. }
     
108. 
     
109. int main() {
     
110.     std::wstring server = L"localhost"; // 替换为你的服务器地址
     
111.     std::wstring filePath = L"/download"; // 替换为下载路径
     
112.     std::wstring savePath = L"C:\\path\\to\\save\\file.zip"; // 替换为保存路径
     
113.     int speedLimit = 50 * 1024; // 限制速度为 50 KB/s
     
114. 
     
115.     DownloadFile(server, filePath, savePath, speedLimit);
     
116. 
     
117.     return 0;
     
118. }
     
119. ```

复制代码

### 代码说明

1. **总字节数**：使用 `long long totalBytesRead` 变量来跟踪已读取的总字节数。
2. **开始时间**：在开始下载时记录开始时间。
3. **计算速度**：在每次读取数据后，计算经过的时间，并根据已读取的字节数计算下载速度（单位为 KB/s）。
4. **输出速度**：在控制台输出当前下载速度。

### 注意事项

- 你可以根据需要调整限速值。
- 计算速度的逻辑可以根据需要进行优化，例如每隔一定时间输出一次速度，而不是每次读取数据后都输出。

通过这种方式，你可以在下载过程中实时监控下载速度。