Initial commit

Lts/Services/SituationReplay/db_reader.cpp

@@ -0,0 +1,136 @@
+/**
+  * 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