qt-migration-legatus/qt6-after/qt6-migrated-legatus/Lts/Services/AssetManagement/object_asset.cpp

/**
  * 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