通过HTTPS从POCO StreamCopier获取文件下载进度

Getting file download progress from POCO StreamCopier through HTTPS

本文关键字:文件下载 获取 StreamCopier HTTPS POCO 通过      更新时间:2023-10-16

我一直在使用WinHTTP和WinInet开发HTTP客户端,最近想到了切换到POCO,因为它为HTTP实现提供了最好的API。

我确实让它工作了,但问题是我想通过定期查询流或通过某些事件处理来了解文件的下载进度。

开始搜索可以执行此操作的 API,然后在 C++ (Poco/Boost) 中遇到了这个包含进度信息的 Http 上传,该上传讨论了使用 CountingOutputStream 获取文件上传方案的进度。我觉得这是不完整的,并没有按照我预期的方式做,根本没有使用该 CountingStream 的实际实现。

我开始知道通过CountingInputStream可以实现,但我不知道如何使用HttpStreamFactory的公开调用返回的流来做到这一点。是否可以使用它以多个块读取流?或定期查询读取的数据量,以便我可以通知 UI?

这是我的代码:

bool HttpClientConnection::DownloadFile ( const std::string& file_url, const std::string file_location )
{
try
{
std::string complete_page_url = "";
std::ofstream file_stream;
std::unique_ptr<std::istream> pStr       = nullptr;

if (isSecureConnection)
{
complete_page_url = "https://";
}
else
{
complete_page_url = "http://";
}

{
complete_page_url = serverHostName + file_url;// assuming the file url itself will contain leading forward slash
}

// Create the URI from the URL to the file.
URI uri(complete_page_url);
//std::auto_ptr<std::istream>pStr(URIStreamOpener::defaultOpener().open(uri);
//StreamCopier::copyStream(*pStr.get(), std::cout);
if (isSecureConnection)
{
std::unique_ptr<HTTPSStreamFactory> https_stream_factory = nullptr;
if (_buseProxy)
{
https_stream_factory = std::unique_ptr<HTTPSStreamFactory>(new HTTPSStreamFactory(proxyHostName, proxyPort, getProxyUserName(),  getProxyPassword()));
}
else
{
https_stream_factory = std::unique_ptr<HTTPSStreamFactory>(new HTTPSStreamFactory());
}
if (https_stream_factory)
{
pStr  = std::unique_ptr<std::istream>(https_stream_factory->open(uri));
}
}
else
{
std::unique_ptr<HTTPStreamFactory> http_stream_factory = nullptr;
if (_buseProxy)
{
http_stream_factory = std::unique_ptr<HTTPStreamFactory>(new HTTPStreamFactory(proxyHostName, proxyPort, getProxyUserName(),  getProxyPassword()));
}
else
{
http_stream_factory   = std::unique_ptr<HTTPStreamFactory>(new HTTPStreamFactory());
}
if (http_stream_factory)
{
pStr  = std::unique_ptr<std::istream>(http_stream_factory->open(uri));
}
}
if (pStr)
{
file_stream.open(file_location, ios::out | ios::trunc | ios::binary);
StreamCopier::copyStream(*pStr.get(), file_stream);
file_stream.close();
}
return true;
}
catch (Exception& exc)
{
if (httpLogger)
{
httpLogger->log(dcLogger::LOG_INFO, "HttpClient:: Exception in DownloadFile , error code: %d", exc.code());
}
}
return false;

}

将文件输出流对象传递给 CountingOutputStream,并以计数输出流作为参数启动计时器。计时器会定期调用,直到数据传输完成,该数据传输获取写入的字节数并通知注册该事件的人。那奏效了!.

shared_ptr<CountingOutputStream> cout_stream = shared_ptr<CountingOutputStream>(new CountingOutputStream(file_stream));
if (callback && callback_interval_in_millis > 0)
{
shared_ptr<FileProgressHandler> file_progress = shared_ptr<FileProgressHandler>(new FileProgressHandler(cout_stream, file_size, callback));
if (file_progress)
{
TimerCallback<FileProgressHandler> callback(*(file_progress.get()), &FileProgressHandler::onTimer);
timer = shared_ptr<Timer>(new Timer(0, callback_interval_in_millis));
if (timer)
{
timer->start(callback);
}
}
StreamCopier::copyStream(*pStr.get(), *cout_stream);
if (timer)
{
timer->stop();
}
}
void onTimer(Timer& timer)
{
try
{
if (progress_callback)
{
int data_processed = 0;
counting_io_stream ? data_processed = counting_io_stream->chars() : __noop;
if (data_processed != total_processed_data)
{
total_processed_data = data_processed;
int percent     = (100 * total_processed_data) / file_size;
progress_callback(percent);
}
}
}
catch(const std::bad_function_call& e) 
{
}
catch(const std::bad_weak_ptr& e)
{
}
catch(const std::exception& e)
{
}
}