Initial commit

Lts/Services/AssetManagement/object_asset.cpp

@@ -0,0 +1,169 @@
+/**
+  * Asset object
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Iuliana Ojog, SLSE, 2017-09-20
+  */
+
+#include "object_asset.h"
+#include "main_handler.h"
+
+#include <LtsUtils/LtsUtilsCore/dds_vector.h>
+
+
+//------------------------------------------------------------------------------
+namespace AssetManagement
+{
+ObjectAsset::ObjectAsset(MainHandler *mainHandler)
+  : m_mainHandler(mainHandler)
+  , m_logger(getLogger("ObjectAsset"))
+{
+}
+
+//------------------------------------------------------------------------------
+ObjectAsset::~ObjectAsset()
+{
+  if (isValid())
+    m_mainHandler->dds()->dts()->removeAssetByKey(m_data.id());
+}
+
+//------------------------------------------------------------------------------
+bool ObjectAsset::isValid() const
+{
+  return m_data.id() > 0;
+}
+
+//------------------------------------------------------------------------------
+bool ObjectAsset::isUsed() const
+{
+  return m_data.isUsed();
+}
+
+//------------------------------------------------------------------------------
+Core::FunctionReplyObject ObjectAsset::createObject(const Asset &data)
+{
+  m_data = data;
+
+  if (m_data.id() <= 0)
+  {
+    m_data.isUsed(false);
+
+    Database::AsmAsset dbData(toDbData(data));
+
+    if (!m_mainHandler->db()->insert(dbData))
+      return Core::FunctionReplyObject(0, tr_lts("dds", "Error inserting AsmAsset-data into database"));
+
+    m_data.id(dbData.id);
+    DEBUG_LOG(m_logger) << "New assetId: " << m_data.id();
+  }
+
+  // Publish Data via DTS
+  onPublishData();
+
+  return Core::FunctionReplyObject(m_data.id());
+}
+
+//------------------------------------------------------------------------------
+Core::FunctionReplyObject ObjectAsset::updateObject(const Asset &data)
+{
+  if (!isValid())
+    return Core::FunctionReplyObject(data.id(), tr_lts("dds", "Object not valid/blocked"));
+
+  Database::AsmAsset dbData;
+  dbData = toDbData(data);
+
+  if (!m_mainHandler->db()->update(dbData))
+    return Core::FunctionReplyObject(data.id(), tr_lts("dds", "Error updating AsmAsset-data into database"));
+
+  m_data = data;
+  DEBUG_LOG(m_logger) << "Data changed for assetId: " << m_data.id();
+
+  // Publish Data via DTS
+  onPublishData();
+
+  return Core::FunctionReplyObject(m_data.id());
+}
+
+//------------------------------------------------------------------------------
+Core::FunctionReplyObject ObjectAsset::setUsed(bool isUsed)
+{
+  if (!isValid())
+    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Trying to update an already removed AsmAsset object"));
+
+  m_data.isUsed(isUsed);
+
+  // Publish Data via DTS
+  onPublishData();
+
+  return Core::FunctionReplyObject(m_data.id());
+}
+
+//------------------------------------------------------------------------------
+Core::FunctionReplyObject ObjectAsset::removeObject()
+{
+  if (isUsed())
+    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Asset still in use"));
+
+  if (!m_mainHandler->db()->removeAsmAsset(m_data.id()))
+    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Asset with id %1 can not be removed from DB")).arg(m_data.id());
+
+  if (!m_mainHandler->dds()->dts()->removeAssetByKey(m_data.id()))
+    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Asset with id %1 can not be removed from DDS")).arg(m_data.id());
+
+  return Core::FunctionReplyObject(m_data.id());
+}
+
+//------------------------------------------------------------------------------
+void ObjectAsset::onPublishData()
+{
+  INFO_LOG(m_logger) << "Publish AsmAsset-data to dds (" << m_data.id() << ") ...";
+
+  if (!m_mainHandler->dds()->dts()->publish(m_data))
+    ERROR_LOG(m_logger) << "Error publishing AsmAsset-data to dds";
+}
+
+// -------------------------------------------------------------------------------------------------------
+// convert methods
+// -------------------------------------------------------------------------------------------------------
+
+// AssetManagement::Asset => Database::AsmAsset
+Database::AsmAsset ObjectAsset::toDbData(const Asset &ddsData)
+{
+  Database::AsmAsset dbData;
+
+  dbData.id          = ddsData.id();
+  dbData.name        = QString::fromStdString(ddsData.name());
+  dbData.assetTypeId = ddsData.assetTypeId();
+  dbData.locationId  = ddsData.locationId();
+  dbData.subLocation = ddsData.subLocation();
+  dbData.certificate = LtsUtils::toQByteArray(ddsData.certificate());
+  dbData.properties  = QString::fromStdString(ddsData.properties());
+
+  return dbData;
+}
+
+// -------------------------------------------------------------------------------------------------------
+// Database::AsmAsset => AssetManagement::Asset
+Asset ObjectAsset::toDdsData(const Database::AsmAsset &dbData)
+{
+  Asset ddsData;
+
+  ddsData.id(dbData.id);
+  ddsData.name(dbData.name.toStdString());
+  ddsData.assetTypeId(dbData.assetTypeId);
+  ddsData.locationId(dbData.locationId);
+  ddsData.subLocation(dbData.subLocation);
+  ddsData.certificate(LtsUtils::convert<std::vector<uint8_t>>(dbData.certificate));
+  ddsData.properties(dbData.properties.toStdString());
+
+  // init data not in db
+  ddsData.isDeleted(false);
+  ddsData.isUsed(false);
+
+  return ddsData;
+}
+
+// -------------------------------------------------------------------------------------------------------
+
+} // Namespace AssetManagement