// 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 #include #include #include 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(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 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 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(sender()); if (!reply) return; QMutexLocker l(&m_mutex); auto it = std::find_if(m_requests.begin(), m_requests.end(), [reply](QPair 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"