Revision control
++ // 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"
++