AmdaProvider.cpp
284 lines
| 11.1 KiB
| text/x-c
|
CppLexer
Alexandre Leroux
|
r348 | #include "AmdaProvider.h" | ||
|
Alexandre Leroux
|
r380 | #include "AmdaDefs.h" | ||
|
Alexandre Leroux
|
r351 | #include "AmdaResultParser.h" | ||
|
Alexandre Leroux
|
r450 | #include <Common/DateUtils.h> | ||
|
Alexandre Leroux
|
r349 | #include <Data/DataProviderParameters.h> | ||
| r358 | #include <Network/NetworkController.h> | |||
| #include <SqpApplication.h> | ||||
| #include <Variable/Variable.h> | ||||
|
Alexandre Leroux
|
r349 | |||
| #include <QNetworkAccessManager> | ||||
| #include <QNetworkReply> | ||||
|
Alexandre Leroux
|
r350 | #include <QTemporaryFile> | ||
| r358 | #include <QThread> | |||
|
Alexandre Leroux
|
r348 | |||
| Q_LOGGING_CATEGORY(LOG_AmdaProvider, "AmdaProvider") | ||||
|
Alexandre Leroux
|
r349 | namespace { | ||
| /// URL format for a request on AMDA server. The parameters are as follows: | ||||
| /// - %1: start date | ||||
| /// - %2: end date | ||||
| /// - %3: parameter id | ||||
|
Alexandre Leroux
|
r723 | /// Old url: http://amda.irap.omp.eu/php/rest/ | ||
|
Alexandre Leroux
|
r349 | const auto AMDA_URL_FORMAT = QStringLiteral( | ||
|
Alexandre Leroux
|
r723 | "http://amdatest.irap.omp.eu/php/rest/" | ||
|
Alexandre Leroux
|
r437 | "getParameter.php?startTime=%1&stopTime=%2¶meterID=%3&outputFormat=ASCII&" | ||
|
Alexandre Leroux
|
r349 | "timeFormat=ISO8601&gzip=0"); | ||
| /// Dates format passed in the URL (e.g 2013-09-23T09:00) | ||||
| r406 | const auto AMDA_TIME_FORMAT = QStringLiteral("yyyy-MM-ddThh:mm:ss"); | |||
|
Alexandre Leroux
|
r349 | |||
| /// Formats a time to a date that can be passed in URL | ||||
| r471 | QString dateFormat(double sqpRange) noexcept | |||
|
Alexandre Leroux
|
r349 | { | ||
| r471 | auto dateTime = DateUtils::dateTime(sqpRange); | |||
|
Alexandre Leroux
|
r349 | return dateTime.toString(AMDA_TIME_FORMAT); | ||
| } | ||||
|
Alexandre Leroux
|
r533 | AmdaResultParser::ValueType valueType(const QString &valueType) | ||
| { | ||||
| if (valueType == QStringLiteral("scalar")) { | ||||
| return AmdaResultParser::ValueType::SCALAR; | ||||
| } | ||||
| else if (valueType == QStringLiteral("vector")) { | ||||
| return AmdaResultParser::ValueType::VECTOR; | ||||
| } | ||||
| else { | ||||
| return AmdaResultParser::ValueType::UNKNOWN; | ||||
| } | ||||
| } | ||||
|
Alexandre Leroux
|
r349 | } // namespace | ||
|
Alexandre Leroux
|
r376 | AmdaProvider::AmdaProvider() | ||
|
Alexandre Leroux
|
r348 | { | ||
|
Alexandre Leroux
|
r424 | qCDebug(LOG_AmdaProvider()) << tr("AmdaProvider::AmdaProvider") << QThread::currentThread(); | ||
| r358 | if (auto app = sqpApp) { | |||
| auto &networkController = app->networkController(); | ||||
| r695 | connect(this, SIGNAL(requestConstructed(std::shared_ptr<QNetworkRequest>, QUuid, | |||
|
Alexandre Leroux
|
r383 | std::function<void(QNetworkReply *, QUuid)>)), | ||
| &networkController, | ||||
| r695 | SLOT(onProcessRequested(std::shared_ptr<QNetworkRequest>, QUuid, | |||
|
Alexandre Leroux
|
r383 | std::function<void(QNetworkReply *, QUuid)>))); | ||
| r391 | ||||
| r693 | connect(&sqpApp->networkController(), | |||
| r695 | SIGNAL(replyDownloadProgress(QUuid, std::shared_ptr<QNetworkRequest>, double)), | |||
| this, | ||||
| SLOT(onReplyDownloadProgress(QUuid, std::shared_ptr<QNetworkRequest>, double))); | ||||
| r358 | } | |||
|
Alexandre Leroux
|
r348 | } | ||
|
Alexandre Leroux
|
r656 | std::shared_ptr<IDataProvider> AmdaProvider::clone() const | ||
| { | ||||
| // No copy is made in the clone | ||||
| return std::make_shared<AmdaProvider>(); | ||||
| } | ||||
| r510 | void AmdaProvider::requestDataLoading(QUuid acqIdentifier, const DataProviderParameters ¶meters) | |||
|
Alexandre Leroux
|
r348 | { | ||
| // NOTE: Try to use multithread if possible | ||||
|
Alexandre Leroux
|
r375 | const auto times = parameters.m_Times; | ||
|
Alexandre Leroux
|
r380 | const auto data = parameters.m_Data; | ||
|
Alexandre Leroux
|
r375 | for (const auto &dateTime : qAsConst(times)) { | ||
| r704 | qCDebug(LOG_AmdaProvider()) << tr("TORM AmdaProvider::requestDataLoading ") << acqIdentifier | |||
| << dateTime; | ||||
| r511 | this->retrieveData(acqIdentifier, dateTime, data); | |||
| r517 | ||||
| r693 | ||||
| r587 | // TORM when AMDA will support quick asynchrone request | |||
| QThread::msleep(1000); | ||||
|
Alexandre Leroux
|
r348 | } | ||
| } | ||||
| r510 | void AmdaProvider::requestDataAborting(QUuid acqIdentifier) | |||
| r388 | { | |||
| if (auto app = sqpApp) { | ||||
| auto &networkController = app->networkController(); | ||||
| r510 | networkController.onReplyCanceled(acqIdentifier); | |||
| r388 | } | |||
| } | ||||
| r693 | void AmdaProvider::onReplyDownloadProgress(QUuid acqIdentifier, | |||
| r695 | std::shared_ptr<QNetworkRequest> networkRequest, | |||
| double progress) | ||||
| r693 | { | |||
| r703 | qCDebug(LOG_AmdaProvider()) << tr("onReplyDownloadProgress") << acqIdentifier | |||
| << networkRequest.get() << progress; | ||||
| r693 | auto acqIdToRequestProgressMapIt = m_AcqIdToRequestProgressMap.find(acqIdentifier); | |||
| if (acqIdToRequestProgressMapIt != m_AcqIdToRequestProgressMap.end()) { | ||||
| r703 | // Update the progression for the current request | |||
| r695 | auto requestPtr = networkRequest; | |||
| auto findRequest = [requestPtr](const auto &entry) { return requestPtr == entry.first; }; | ||||
| r693 | ||||
| auto &requestProgressMap = acqIdToRequestProgressMapIt->second; | ||||
| auto requestProgressMapEnd = requestProgressMap.end(); | ||||
| auto requestProgressMapIt | ||||
| = std::find_if(requestProgressMap.begin(), requestProgressMapEnd, findRequest); | ||||
| if (requestProgressMapIt != requestProgressMapEnd) { | ||||
| requestProgressMapIt->second = progress; | ||||
| } | ||||
| else { | ||||
| r701 | // This case can happened when a progression is send after the request has been | |||
| // finished. | ||||
| // Generaly the case when aborting a request | ||||
| r704 | qCDebug(LOG_AmdaProvider()) << tr("Can't retrieve Request in progress") << acqIdentifier | |||
| << networkRequest.get() << progress; | ||||
| r693 | } | |||
| r703 | // Compute the current final progress and notify it | |||
| r693 | double finalProgress = 0.0; | |||
| auto fraq = requestProgressMap.size(); | ||||
| for (auto requestProgress : requestProgressMap) { | ||||
| finalProgress += requestProgress.second; | ||||
| r703 | qCDebug(LOG_AmdaProvider()) << tr("Current final progress without fraq:") | |||
| r695 | << finalProgress << requestProgress.second; | |||
| r693 | } | |||
| if (fraq > 0) { | ||||
| finalProgress = finalProgress / fraq; | ||||
| } | ||||
| r703 | qCDebug(LOG_AmdaProvider()) << tr("Current final progress: ") << fraq << finalProgress; | |||
| r693 | emit dataProvidedProgress(acqIdentifier, finalProgress); | |||
| } | ||||
| else { | ||||
| r701 | // This case can happened when a progression is send after the request has been finished. | |||
| // Generaly the case when aborting a request | ||||
| emit dataProvidedProgress(acqIdentifier, 100.0); | ||||
| r693 | } | |||
| } | ||||
| r471 | void AmdaProvider::retrieveData(QUuid token, const SqpRange &dateTime, const QVariantHash &data) | |||
|
Alexandre Leroux
|
r349 | { | ||
|
Alexandre Leroux
|
r380 | // Retrieves product ID from data: if the value is invalid, no request is made | ||
| auto productId = data.value(AMDA_XML_ID_KEY).toString(); | ||||
| if (productId.isNull()) { | ||||
| qCCritical(LOG_AmdaProvider()) << tr("Can't retrieve data: unknown product id"); | ||||
| return; | ||||
| } | ||||
|
Alexandre Leroux
|
r533 | // Retrieves the data type that determines whether the expected format for the result file is | ||
| // scalar, vector... | ||||
| auto productValueType = valueType(data.value(AMDA_DATA_TYPE_KEY).toString()); | ||||
|
Alexandre Leroux
|
r349 | // /////////// // | ||
| // Creates URL // | ||||
| // /////////// // | ||||
|
Alexandre Leroux
|
r375 | auto startDate = dateFormat(dateTime.m_TStart); | ||
| auto endDate = dateFormat(dateTime.m_TEnd); | ||||
|
Alexandre Leroux
|
r349 | |||
| auto url = QUrl{QString{AMDA_URL_FORMAT}.arg(startDate, endDate, productId)}; | ||||
| r703 | qCInfo(LOG_AmdaProvider()) << tr("TORM AmdaProvider::retrieveData url:") << url; | |||
| r358 | auto tempFile = std::make_shared<QTemporaryFile>(); | |||
|
Alexandre Leroux
|
r349 | |||
| r358 | // LAMBDA | |||
|
Alexandre Leroux
|
r533 | auto httpDownloadFinished = [this, dateTime, tempFile, | ||
| productValueType](QNetworkReply *reply, QUuid dataId) noexcept { | ||||
| // Don't do anything if the reply was abort | ||||
| r703 | if (reply->error() == QNetworkReply::NoError) { | |||
|
Alexandre Leroux
|
r533 | |||
| if (tempFile) { | ||||
| auto replyReadAll = reply->readAll(); | ||||
| if (!replyReadAll.isEmpty()) { | ||||
| tempFile->write(replyReadAll); | ||||
| } | ||||
| tempFile->close(); | ||||
| // Parse results file | ||||
| if (auto dataSeries | ||||
| = AmdaResultParser::readTxt(tempFile->fileName(), productValueType)) { | ||||
| emit dataProvided(dataId, dataSeries, dateTime); | ||||
| } | ||||
| else { | ||||
| /// @todo ALX : debug | ||||
| r704 | emit dataProvidedFailed(dataId); | |||
|
Alexandre Leroux
|
r533 | } | ||
| } | ||||
| r701 | qCDebug(LOG_AmdaProvider()) << tr("acquisition requests erase because of finishing") | |||
| << dataId; | ||||
| r693 | m_AcqIdToRequestProgressMap.erase(dataId); | |||
|
Alexandre Leroux
|
r533 | } | ||
| r703 | else { | |||
| qCCritical(LOG_AmdaProvider()) << tr("httpDownloadFinished ERROR"); | ||||
| r704 | emit dataProvidedFailed(dataId); | |||
| r703 | } | |||
|
Alexandre Leroux
|
r533 | |||
| }; | ||||
| r397 | auto httpFinishedLambda | |||
| = [this, httpDownloadFinished, tempFile](QNetworkReply *reply, QUuid dataId) noexcept { | ||||
|
Alexandre Leroux
|
r350 | |||
| r397 | // Don't do anything if the reply was abort | |||
| r703 | if (reply->error() == QNetworkReply::NoError) { | |||
|
Alexandre Leroux
|
r723 | auto downloadFileUrl = QUrl{QString{reply->readAll()}.trimmed()}; | ||
| r388 | ||||
| r703 | qCInfo(LOG_AmdaProvider()) | |||
| r517 | << tr("TORM AmdaProvider::retrieveData downloadFileUrl:") << downloadFileUrl; | |||
| r397 | // Executes request for downloading file // | |||
| r358 | ||||
| r397 | // Creates destination file | |||
| if (tempFile->open()) { | ||||
| r695 | // Executes request and store the request for progression | |||
| r693 | auto request = std::make_shared<QNetworkRequest>(downloadFileUrl); | |||
| updateRequestProgress(dataId, request, 0.0); | ||||
| r695 | emit requestConstructed(request, dataId, httpDownloadFinished); | |||
| r397 | } | |||
| r704 | else { | |||
| emit dataProvidedFailed(dataId); | ||||
| } | ||||
| r397 | } | |||
| r693 | else { | |||
| r701 | qCDebug(LOG_AmdaProvider()) | |||
| << tr("acquisition requests erase because of aborting") << dataId; | ||||
| r703 | qCCritical(LOG_AmdaProvider()) << tr("httpFinishedLambda ERROR"); | |||
| r693 | m_AcqIdToRequestProgressMap.erase(dataId); | |||
| r704 | emit dataProvidedFailed(dataId); | |||
| r693 | } | |||
| r397 | }; | |||
| r358 | ||||
| // //////////////// // | ||||
| // Executes request // | ||||
| // //////////////// // | ||||
| r693 | ||||
| auto request = std::make_shared<QNetworkRequest>(url); | ||||
| r695 | qCDebug(LOG_AmdaProvider()) << tr("First Request creation") << request.get(); | |||
| r693 | updateRequestProgress(token, request, 0.0); | |||
| r695 | emit requestConstructed(request, token, httpFinishedLambda); | |||
| r693 | } | |||
| void AmdaProvider::updateRequestProgress(QUuid acqIdentifier, | ||||
| std::shared_ptr<QNetworkRequest> request, double progress) | ||||
| { | ||||
| auto acqIdToRequestProgressMapIt = m_AcqIdToRequestProgressMap.find(acqIdentifier); | ||||
| if (acqIdToRequestProgressMapIt != m_AcqIdToRequestProgressMap.end()) { | ||||
| auto &requestProgressMap = acqIdToRequestProgressMapIt->second; | ||||
| auto requestProgressMapIt = requestProgressMap.find(request); | ||||
| if (requestProgressMapIt != requestProgressMap.end()) { | ||||
| requestProgressMapIt->second = progress; | ||||
| r695 | qCDebug(LOG_AmdaProvider()) << tr("updateRequestProgress new progress for request") | |||
| << acqIdentifier << request.get() << progress; | ||||
| r693 | } | |||
| else { | ||||
| r695 | qCDebug(LOG_AmdaProvider()) << tr("updateRequestProgress new request") << acqIdentifier | |||
| << request.get() << progress; | ||||
| r693 | acqIdToRequestProgressMapIt->second.insert(std::make_pair(request, progress)); | |||
| } | ||||
| } | ||||
| else { | ||||
| r695 | qCDebug(LOG_AmdaProvider()) << tr("updateRequestProgress new acqIdentifier") | |||
| << acqIdentifier << request.get() << progress; | ||||
| r693 | auto requestProgressMap = std::map<std::shared_ptr<QNetworkRequest>, double>{}; | |||
| requestProgressMap.insert(std::make_pair(request, progress)); | ||||
| m_AcqIdToRequestProgressMap.insert( | ||||
| std::make_pair(acqIdentifier, std::move(requestProgressMap))); | ||||
| } | ||||
|
Alexandre Leroux
|
r348 | } | ||
