Initial commit

Lts/Services/SituationReplay/handler_situation_replay.cpp

@@ -0,0 +1,174 @@
+/**
+  * @RDE_HEADER@
+  * Handler for all request journal objects
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Hanna E. Rathje, SLSE, 2015-01-13
+  */
+
+#include "handler_situation_replay.h"
+#include "object_situation_replay.h"
+#include "db_reader.h"
+#include <SituationReplay/srp_db.h>
+
+#include <ngldb/nsqlquery.h>
+#include <QCoreApplication>
+#include <NGLLogger/NGLLoggerBootstrap.h>
+
+namespace SituationReplay
+{
+
+HandlerSituationReplay::HandlerSituationReplay(const std::shared_ptr<DdsHandler> &ddsHandler,
+                                               const std::shared_ptr<DatabaseInterface> &db,
+                                               const std::shared_ptr<DbReader> &dbReader)
+  : Interface::Fts::ChangeReplayStateRequestHandler(ddsHandler->fts())
+  , m_objSrp(nullptr)
+  , m_ddsHandler(ddsHandler)
+  , m_db(db)
+  , m_dbReader(dbReader)
+  , m_exerciseId(0)
+  , m_partitionId(0)
+  , m_logger(getLogger("HandlerSituationReplay"))
+{}
+
+HandlerSituationReplay::~HandlerSituationReplay()
+{
+  delete m_objSrp;
+}
+
+bool HandlerSituationReplay::init(int exerciseId, int partitionId)
+{
+  m_exerciseId  = exerciseId;
+  m_partitionId = partitionId;
+
+  const QString tableName = m_db->schemaName().isEmpty() ? QStringLiteral("exercise") : QStringLiteral("exm.exercise");
+  const QString stmt(QStringLiteral("SELECT start_time FROM %1 WHERE id = %2").arg(tableName).arg(exerciseId));
+  NSqlQuery q(m_db->database());
+  const qint64 exerciseStartTime = (q.execAndFetch(stmt) ? q.value(0).toLongLong() : 0);
+
+  DEBUG_LOG(m_logger) << "stmt: " << stmt << ", res " << exerciseStartTime;
+
+  Database::SrpSituationReplay newData;
+  newData.srpStateId        = SrpState::Stopped;
+  newData.replayPartitionId = m_partitionId;
+  newData.beginTime         = exerciseStartTime;
+  newData.endTime           = exerciseStartTime;
+  newData.pace              = 0.0;
+
+  return createObject(newData);
+}
+
+bool HandlerSituationReplay::createObject(const Database::SrpSituationReplay &newData)
+{
+  delete m_objSrp;
+  m_objSrp = new ObjectSituationReplay(m_ddsHandler, m_db, m_dbReader, m_exerciseId);
+  Core::FunctionReplyObject ret = m_objSrp->createObject(newData);
+  if (ret.error())
+  {
+    delete m_objSrp;
+    m_objSrp = nullptr;
+    ERROR_LOG(m_logger) << ret();
+  }
+  return m_objSrp != nullptr;
+}
+
+Core::FunctionReplyObject HandlerSituationReplay::startReplayRequest(const SituationReplay::ChangeReplayStateRequest &request)
+{
+  DEBUG_LOG(m_logger) << "onStartReplayRequest: " << request;
+  // Start only own replay
+  if (request.replayPartitionId() != m_partitionId)
+    return Core::FunctionReplyObject(request.replayPartitionId(),tr_lts("dds", "Wrong replay partition"));
+  if (m_objSrp && m_objSrp->srpState() != SrpState::Stopped)
+    return Core::FunctionReplyObject(request.replayPartitionId(), tr_lts("dds", "Replay is not yet stopped"));
+
+  Database::SrpSituationReplay newData;
+  newData.srpStateId        = SrpState::Started;
+  newData.replayPartitionId = m_partitionId;
+  newData.beginTime         = request.beginTime();
+  newData.endTime           = request.endTime();
+  newData.pace              = request.pace().is_set() ? request.pace().get() : 1;
+
+  // round to sec
+  newData.beginTime = (newData.beginTime / 1000) * 1000;
+  if (newData.endTime)
+    newData.endTime = ((newData.endTime / 1000) + 1) * 1000;
+
+  if (!m_objSrp && !createObject(newData))
+    return Core::FunctionReplyObject(request.replayPartitionId(),tr_lts("dds", "Internal error"));
+
+  m_objSrp->setSrpData(newData);
+  return Core::FunctionReplyObject(m_objSrp->getId());
+}
+
+Core::FunctionReplyObject HandlerSituationReplay::stopReplayRequest(const SituationReplay::ChangeReplayStateRequest &request)
+{
+  DEBUG_LOG(m_logger) << "onStopReplayRequest: " << request;
+
+  if (m_objSrp == nullptr)
+    return Core::FunctionReplyObject(request.id(), tr_lts("dds", "SituationReplay for replayPartitionId %1 not found")).arg(request.replayPartitionId());
+
+  if (m_objSrp && (m_objSrp->getId() != request.id() || m_objSrp->getReplayPartitionId() != request.replayPartitionId()))
+    return Core::FunctionReplyObject(request.id(),tr_lts("dds", "Given request id %1 not equal object id %2")).arg(request.id()).arg(m_objSrp->getId());
+
+  Core::FunctionReplyObject ret = m_objSrp->switchToState(SrpState::Stopped);
+  if (!ret.error())
+    ret = m_objSrp->removeObject();
+
+  // remove all old data from DDS-Bus & clear maps
+  m_dbReader->clearData();
+
+  return ret;
+}
+
+Core::FunctionReplyObject HandlerSituationReplay::changeReplayStateRequest(const SituationReplay::ChangeReplayStateRequest &request)
+{
+  if (request.state() != SrpState_def::Started)
+  {
+    if (m_objSrp == nullptr)
+      return Core::FunctionReplyObject(request.id(), tr_lts("dds", "SituationReplay for replayPartitionId %1 not found")).arg(request.replayPartitionId());
+
+    if (m_objSrp && (m_objSrp->getId() != request.id() || m_objSrp->getReplayPartitionId() != request.replayPartitionId()))
+      return Core::FunctionReplyObject(request.id(),tr_lts("dds", "Given request id %1 not equal object id %2")).arg(request.id()).arg(m_objSrp->getId());
+  }
+
+  if (request.pace().is_set())
+    m_objSrp->setPace(request.pace().get());
+
+  switch (request.state().underlying())
+  {
+  case SrpState_def::Unknown:
+    if (request.pace().is_set())
+      return Core::FunctionReplyObject(request.id());
+    return Core::FunctionReplyObject(request.id(), tr_lts("dds", "Can not handle command 'Unknown' in changeReplayStateRequest"));
+  case SrpState_def::Started:
+    return startReplayRequest(request);
+  case SrpState_def::Resumed:
+  {
+    DEBUG_LOG(m_logger) << "Resume request: " << request.beginTime() << ", cur playback time: " << m_objSrp->currentReplayTime();
+    if (request.beginTime() != 0 && request.beginTime() != m_objSrp->currentReplayTime())
+    {
+      stopReplayRequest(request);
+      return startReplayRequest(request);
+    }
+    Q_FALLTHROUGH();
+  case SrpState_def::Paused:
+    return m_objSrp->switchToState(request.state());
+  }
+  case SrpState_def::Stopped:
+    return stopReplayRequest(request);
+  }
+  return Core::FunctionReplyObject(request.id(), tr_lts("dds", "Unhandled request state %1")).arg(request.state().underlying());
+}
+
+void HandlerSituationReplay::onChangeReplayStateRequest(const rti::core::SampleIdentity &cookie,
+                                                        const ChangeReplayStateRequest &request)
+{
+  DEBUG_LOG(m_logger) << "onChangeReplayStateRequest: " << request;
+  auto ret = changeReplayStateRequest(request);
+  if (ret.error())
+    WARNING_LOG(m_logger) << ret();
+  m_ddsHandler->fts()->changeReplayStateReply(cookie, ret);
+}
+
+} // Namespace SituationReplay