Ralph Wessel
119 lines
4.2 KiB
C++
119 lines
4.2 KiB
C++
#include "Connector/Database/Model/Card/ModelCardDatabase.h"
|
|
|
|
#include "Active/Database/Storage/Storage.h"
|
|
#include "Active/Serialise/JSON/JSONTransport.h"
|
|
#include "Speckle/Database/Storage/DocumentStore/DocumentStoreEngine.h"
|
|
|
|
#include <array>
|
|
|
|
using namespace active::container;
|
|
using namespace active::database;
|
|
using namespace active::serialise;
|
|
using namespace active::serialise::json;
|
|
using namespace active::setting;
|
|
using namespace connector::database;
|
|
using namespace speckle::database;
|
|
using namespace speckle::utility;
|
|
|
|
namespace connector::database {
|
|
|
|
///ModelCard database engine declaration
|
|
class ModelCardDatabase::Engine : public DocumentStoreEngine<ModelCard, ModelCard, JSONTransport, active::utility::String> {
|
|
using base = DocumentStoreEngine<ModelCard, ModelCard, JSONTransport, active::utility::String>;
|
|
using base::base;
|
|
};
|
|
|
|
///ModelCard database storage declaration
|
|
class ModelCardDatabase::Store : public active::database::Storage<connector::database::ModelCard, active::serialise::json::JSONTransport,
|
|
active::utility::String, active::utility::String, active::utility::String, active::utility::String> {
|
|
using base = active::database::Storage<connector::database::ModelCard, active::serialise::json::JSONTransport,
|
|
active::utility::String, active::utility::String, active::utility::String, active::utility::String>;
|
|
using base::base;
|
|
};
|
|
|
|
}
|
|
|
|
namespace {
|
|
|
|
///The document storage identifier for model cards
|
|
const char* modelCardDBaseName = "connector::database::ModelModelCardDatabase";
|
|
///The document storage identifier for model cards
|
|
const char* modelCardTableName = "models";
|
|
|
|
}
|
|
|
|
/*--------------------------------------------------------------------
|
|
Constructor
|
|
--------------------------------------------------------------------*/
|
|
ModelCardDatabase::ModelCardDatabase() {
|
|
auto engine = std::make_shared<Engine>(modelCardDBaseName,
|
|
//Schema
|
|
DBaseSchema{active::utility::String{modelCardDBaseName},
|
|
//Tables
|
|
{
|
|
//Model card table
|
|
{
|
|
modelCardTableName, 0, 0, {} //The table specifics aren't relevant in this context
|
|
}
|
|
}
|
|
}
|
|
);
|
|
m_store = std::make_shared<Store>(m_engine);
|
|
} //ModelCardDatabase::ModelCardDatabase
|
|
|
|
|
|
/*--------------------------------------------------------------------
|
|
Destructor
|
|
--------------------------------------------------------------------*/
|
|
ModelCardDatabase::~ModelCardDatabase() {}
|
|
|
|
|
|
/*--------------------------------------------------------------------
|
|
Get all model cards
|
|
|
|
return: All the cards
|
|
--------------------------------------------------------------------*/
|
|
Vector<ModelCard> ModelCardDatabase::getCards() const {
|
|
return m_store->getObjects();
|
|
} //ModelCardDatabase::getCards
|
|
|
|
|
|
/*--------------------------------------------------------------------
|
|
Write a card to storage
|
|
|
|
card: The card to write
|
|
--------------------------------------------------------------------*/
|
|
void ModelCardDatabase::write(const ModelCard& card) const {
|
|
m_store->write(card);
|
|
} //ModelCardDatabase::write
|
|
|
|
|
|
/*--------------------------------------------------------------------
|
|
Erase a card
|
|
|
|
cardID: The ID of the card to erase
|
|
--------------------------------------------------------------------*/
|
|
void ModelCardDatabase::erase(const speckle::utility::String& cardID) const {
|
|
m_store->erase(cardID);
|
|
} //ModelCardDatabase::erase
|
|
|
|
|
|
/*--------------------------------------------------------------------
|
|
Get a serialisation wrapper for the database
|
|
|
|
return: A database wrapper
|
|
--------------------------------------------------------------------*/
|
|
std::unique_ptr<Cargo> ModelCardDatabase::wrapper() const {
|
|
return m_store->wrapper();
|
|
} //ModelCardDatabase::wrapper
|
|
|
|
|
|
/*--------------------------------------------------------------------
|
|
Get the database subscription (the content is document-based, and must react to document operations)
|
|
|
|
return: The database subscription (add weakly to publisher)
|
|
--------------------------------------------------------------------*/
|
|
std::shared_ptr<active::event::Subscriber> ModelCardDatabase::getSubscription() {
|
|
return std::dynamic_pointer_cast<active::event::Subscriber>(m_engine);
|
|
} //ModelCardDatabase::getSubscription
|