1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright (C) 2017 Klaralvdalens Datakonsult AB (KDAB).
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
#include "qdownloadhelperservice_p.h"
#include "qdownloadnetworkworker_p.h"
#include <QNetworkAccessManager>
#include <QNetworkRequest>
#include <QNetworkReply>
#include <QDataStream>
QT_BEGIN_NAMESPACE
namespace Qt3DCore {
QDownloadNetworkWorker::QDownloadNetworkWorker(QObject *parent)
: QObject(parent)
, m_networkManager(nullptr)
{
connect(this, &QDownloadNetworkWorker::submitRequest,
this, &QDownloadNetworkWorker::onRequestSubmited);
connect(this, &QDownloadNetworkWorker::cancelRequest,
this, &QDownloadNetworkWorker::onRequestCancelled);
connect(this, &QDownloadNetworkWorker::cancelAllRequests,
this, &QDownloadNetworkWorker::onAllRequestsCancelled);
}
void QDownloadNetworkWorker::onRequestSubmited(const QDownloadRequestPtr &request)
{
QMutexLocker l(&m_mutex);
if (!m_networkManager) {
m_networkManager = new QNetworkAccessManager(this);
connect(m_networkManager, &QNetworkAccessManager::finished,
this, &QDownloadNetworkWorker::onRequestFinished);
}
auto reply = m_networkManager->get(QNetworkRequest(request->url()));
m_requests << QPair<QDownloadRequestPtr, QNetworkReply *>(request, reply);
connect(reply, &QNetworkReply::downloadProgress, this, &QDownloadNetworkWorker::onDownloadProgressed);
}
void QDownloadNetworkWorker::onRequestCancelled(const QDownloadRequestPtr &request)
{
QMutexLocker l(&m_mutex);
auto it = std::find_if(m_requests.begin(), m_requests.end(),
[request](QPair<QDownloadRequestPtr, QNetworkReply *> e) {
return e.first == request;
});
if (it == m_requests.end())
return;
(*it).first->m_cancelled = true;
(*it).second->abort();
}
void QDownloadNetworkWorker::onAllRequestsCancelled()
{
QMutexLocker l(&m_mutex);
for (auto &e: std::as_const(m_requests)) {
e.first->m_cancelled = true;
e.second->abort();
}
m_requests.clear();
}
void QDownloadNetworkWorker::onRequestFinished(QNetworkReply *reply)
{
QMutexLocker l(&m_mutex);
auto it = std::find_if(m_requests.begin(), m_requests.end(),
[reply](QPair<QDownloadRequestPtr, QNetworkReply *> e) {
return e.second == reply;
});
if (it == m_requests.end())
return;
auto request = (*it).first;
if (reply->error() == QNetworkReply::NoError) {
request->m_succeeded = true;
}
request->onDownloaded();
emit requestDownloaded(request);
m_requests.erase(it);
}
void QDownloadNetworkWorker::onDownloadProgressed(qint64 bytesReceived, qint64 bytesTotal)
{
Q_UNUSED(bytesReceived);
Q_UNUSED(bytesTotal);
// TODO forward progress details somewhere
QNetworkReply *reply = qobject_cast<QNetworkReply *>(sender());
if (!reply)
return;
QMutexLocker l(&m_mutex);
auto it = std::find_if(m_requests.begin(), m_requests.end(),
[reply](QPair<QDownloadRequestPtr, QNetworkReply *> e) {
return e.second == reply;
});
if (it == m_requests.end())
return;
auto request = (*it).first;
QDataStream stream(&request->m_data, QIODevice::Append);
QByteArray data = reply->readAll();
stream.writeRawData(data.data(), data.size());
}
} // namespace Qt3DCore
QT_END_NAMESPACE
#include "moc_qdownloadnetworkworker_p.cpp"
|