137 lines
4.1 KiB
C++
Executable File
137 lines
4.1 KiB
C++
Executable File
/**
|
|
* DbReader for RecordingManagement
|
|
* $Id$
|
|
* $HeadURL$
|
|
* $Author$
|
|
*/
|
|
#include "db_reader.h"
|
|
|
|
#include "logger.h"
|
|
#include <DDS/DDSConnector/dds_connector.h>
|
|
|
|
#include <LtsPluginManager/lts_plugin_loader.h>
|
|
#include <ngldb/nsqlquery.h>
|
|
|
|
namespace SituationReplay
|
|
{
|
|
|
|
DbReader::DbReader(const std::shared_ptr<DDSConnector::DdsConnector> &ddsConnector,
|
|
int exerciseId,
|
|
bool bCanHandleSchemas)
|
|
: m_exerciseId(exerciseId)
|
|
, m_query(nullptr)
|
|
, m_ddsConnector(ddsConnector)
|
|
, m_bCanHandleSchemas(bCanHandleSchemas)
|
|
, m_logger(getLogger("DbReader"))
|
|
{}
|
|
|
|
DbReader::~DbReader()
|
|
{
|
|
delete m_query;
|
|
LtsPluginManager::LtsPluginLoader::getInstance()->unloadAllPlugins();
|
|
m_plugins.clear();
|
|
}
|
|
|
|
bool DbReader::loadPlugin(const LtsPluginManager::PluginInstance &pi)
|
|
{
|
|
RecordReplay::RecordReplayPlugin *recPlugin = qobject_cast<RecordReplay::RecordReplayPlugin*>(pi.plugin);
|
|
if (!recPlugin)
|
|
{
|
|
ERROR_LOG(m_logger) << "Cannot load RecordReplayPlugin " << pi.fileName;
|
|
return false;
|
|
}
|
|
|
|
if (!recPlugin->initDts(m_ddsConnector, m_exerciseId, m_bCanHandleSchemas, *m_query))
|
|
{
|
|
ERROR_LOG(m_logger) << "Initialisation for RecordReplayPlugin " << pi.fileName << " failed ";
|
|
return false;
|
|
}
|
|
|
|
INFO_LOG(m_logger) << "Loaded record replay plugin " << recPlugin->pluginName();
|
|
m_plugins.push_back(recPlugin);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool DbReader::init(const QSqlDatabase &db)
|
|
{
|
|
delete m_query;
|
|
m_query = new NSqlQuery(db);
|
|
|
|
auto cb = [this](const LtsPluginManager::PluginInstance &pi) { return loadPlugin(pi); };
|
|
LtsPluginManager::LtsPluginLoader::getInstance()->loadPlugins(LtsPluginManager::PluginType::SPECIAL,
|
|
cb,
|
|
QStringList(),
|
|
QStringLiteral("recordreplay"));
|
|
return true;
|
|
}
|
|
|
|
bool DbReader::setReplayStartTime(qint64 startTime)
|
|
{
|
|
INFO_LOG(m_logger) << "IN setReplayStartTime " << startTime;
|
|
if (!m_query)
|
|
{
|
|
WARNING_LOG(m_logger) << "Internal error - no query!";
|
|
return false;
|
|
}
|
|
for (RecordReplay::RecordReplayPlugin *recPlugin : m_plugins)
|
|
{
|
|
INFO_LOG(m_logger) << "IN setReplayStartTime, current Plugin: " << recPlugin->pluginName()
|
|
<< " typename=" << typeid(*recPlugin).name();
|
|
if (!recPlugin->setReplayStartTime(*m_query, startTime))
|
|
{
|
|
WARNING_LOG(m_logger) << "setReplayStartTime for plugin " << recPlugin->svcShortName() << " failed!";
|
|
return false;
|
|
}
|
|
}
|
|
INFO_LOG(m_logger) << "IN setReplayStartTime " << startTime << " DONE";
|
|
return true;
|
|
}
|
|
|
|
bool DbReader::readData(qint64 fromTime, qint64 toTime)
|
|
{
|
|
INFO_LOG(m_logger) << "IN readData " << fromTime << " to " << toTime;
|
|
if (!m_query)
|
|
{
|
|
WARNING_LOG(m_logger) << "Internal error - no query!";
|
|
return false;
|
|
}
|
|
for (RecordReplay::RecordReplayPlugin *recPlugin : m_plugins)
|
|
{
|
|
if (!recPlugin->ddsTopicsFromDatabase(*m_query, fromTime, toTime))
|
|
{
|
|
WARNING_LOG(m_logger) << "ddsTopicsFromDatabase for plugin " << recPlugin->svcShortName() << " failed!";
|
|
return false;
|
|
}
|
|
}
|
|
INFO_LOG(m_logger) << "IN readData " << fromTime << " to " << toTime << " DONE";
|
|
return true;
|
|
}
|
|
|
|
void DbReader::publish(qint64 desiredTime)
|
|
{
|
|
INFO_LOG(m_logger) << "IN publish " << desiredTime;
|
|
for (RecordReplay::RecordReplayPlugin *recPlugin : m_plugins)
|
|
if (!recPlugin->publishData(desiredTime))
|
|
WARNING_LOG(m_logger) << "publishData for plugin " << recPlugin->svcShortName() << " failed!";
|
|
INFO_LOG(m_logger) << "IN publish " << desiredTime << " DONE";
|
|
}
|
|
|
|
void DbReader::waitForAsyncOperations(qint64 desiredTime)
|
|
{
|
|
INFO_LOG(m_logger) << "IN waitForAsyncOperations " << desiredTime;
|
|
for (RecordReplay::RecordReplayPlugin *recPlugin : m_plugins)
|
|
if (!recPlugin->waitForAsyncOperations(desiredTime))
|
|
WARNING_LOG(m_logger) << "waitForAsyncOperations for plugin " << recPlugin->svcShortName() << " failed!";
|
|
}
|
|
|
|
|
|
void DbReader::clearData()
|
|
{
|
|
INFO_LOG(m_logger) << "IN clearData";
|
|
for (RecordReplay::RecordReplayPlugin *recPlugin : m_plugins)
|
|
recPlugin->clearData();
|
|
}
|
|
|
|
} // RecordingManagement
|