radio/lib/wDatabase/database.cpp

231 lines
5.7 KiB
C++
Raw Normal View History

2018-08-04 21:46:25 +00:00
#include "database.h"
#include <sys/stat.h>
#include <sys/types.h>
#include <wType/bytearray.h>
#include <algorithm>
Database::Database(const W::String& dbName, QObject* parent):
M::ICatalogue(W::Address({dbName}), parent),
name(dbName),
opened(false),
environment(lmdb::env::create()),
dbi(0),
elements()
{
}
Database::~Database()
{
}
void Database::open()
{
if (!opened) {
checkDirAndOpenEnvironment();
index();
opened = true;
}
}
void Database::addIndex(const W::String& fieldName, W::Object::objectType fieldType)
{
ICatalogue::addIndex(fieldName, fieldType);
if (opened) {
lmdb::txn rtxn = lmdb::txn::begin(environment, nullptr, MDB_RDONLY);
lmdb::cursor cursor = lmdb::cursor::open(rtxn, dbi);
lmdb::val key;
lmdb::val value;
while (cursor.get(key, value, MDB_NEXT)) {
uint64_t iKey = *((uint64_t*) key.data());
W::ByteArray ba(value.size());
ba.fill(value.data(), value.size());
W::Vocabulary* wVal = static_cast<W::Vocabulary*>(W::Object::fromByteArray(ba));
IndexMap::const_iterator itr = indexes.find(fieldName);
itr->second->add(wVal->at(itr->first), iKey);
delete wVal;
}
cursor.close();
rtxn.abort();
}
}
uint64_t Database::addElement(const W::Vocabulary& record)
{
if (!opened) {
throw 6; //TODO
}
uint64_t id = ICatalogue::addElement(record);
elements.insert(id);
int size = record.size();
W::ByteArray ba(size + 1);
ba.push8(record.getType());
record.serialize(ba);
lmdb::val key((uint8_t*) &id, 8);
lmdb::val value(ba.getData(), ba.size());
lmdb::txn wTrans = lmdb::txn::begin(environment);
dbi.put(wTrans, key, value);
wTrans.commit();
return id;
}
void Database::checkDirAndOpenEnvironment()
{
int state1 = mkdir("./db", S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
if (state1 != 0 && errno != EEXIST) {
emit serviceMessage("Failed to create a root database folder");
throw 1;
}
W::String path("./db/");
path += name;
int state2 = mkdir(path.toString().c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
if (state2 != 0 && errno != EEXIST) {
emit serviceMessage(QString("Failed to create ") + name.toString().c_str() + " database folder");
throw 1;
}
environment.set_mapsize(1UL * 1024UL * 1024UL * 1024UL);
environment.set_max_dbs(10);
environment.open(path.toString().c_str(), 0, 0664);
lmdb::txn wTrans = lmdb::txn::begin(environment);
dbi = lmdb::dbi::open(wTrans, "main", MDB_CREATE | MDB_INTEGERKEY);
wTrans.commit();
}
void Database::index()
{
lmdb::txn rtxn = lmdb::txn::begin(environment, nullptr, MDB_RDONLY);
lmdb::cursor cursor = lmdb::cursor::open(rtxn, dbi);
lmdb::val key;
lmdb::val value;
while (cursor.get(key, value, MDB_NEXT)) {
uint64_t iKey = *((uint64_t*) key.data());
W::ByteArray ba(value.size());
ba.fill(value.data(), value.size());
W::Vocabulary* wVal = static_cast<W::Vocabulary*>(W::Object::fromByteArray(ba));
ICatalogue::addElement(*wVal);
elements.insert(iKey);
delete wVal;
}
cursor.close();
rtxn.abort();
emit countChange(elements.size());
}
W::Vocabulary* Database::getElement(uint64_t id)
{
lmdb::txn rtxn = lmdb::txn::begin(environment, nullptr, MDB_RDONLY);
lmdb::val key((uint8_t*) &id, 8);
lmdb::val value;
if (dbi.get(rtxn, key, value)) {
W::ByteArray ba(value.size());
ba.fill(value.data(), value.size());
W::Vocabulary* wVal = static_cast<W::Vocabulary*>(W::Object::fromByteArray(ba));
rtxn.abort();
return wVal;
} else {
rtxn.abort();
throw 3;
}
}
std::set<uint64_t> Database::getAll() const
{
return elements;
}
void Database::removeElement(uint64_t id)
{
if (!opened) {
throw 6; //TODO
}
ICatalogue::removeElement(id);
lmdb::txn transaction = lmdb::txn::begin(environment);
lmdb::val key((uint8_t*) &id, 8);
dbi.del(transaction, key);
transaction.commit();
elements.erase(id);
}
void Database::clear()
{
if (!opened) {
throw 6; //TODO
}
M::ICatalogue::clear();
lmdb::txn transaction = lmdb::txn::begin(environment);
dbi.drop(transaction);
transaction.commit();
}
void Database::addModel(M::Model* model)
{
connect(model, SIGNAL(subscribersCountChange(uint64_t)), this, SLOT(onChildSubscribersCountChange(uint64_t)));
M::ICatalogue::addModel(model);
}
void Database::removeModel(M::Model* model)
{
disconnect(model, SIGNAL(subscribersCountChange(uint64_t)), this, SLOT(onChildSubscribersCountChange(uint64_t)));
M::ICatalogue::removeModel(model);
}
void Database::onChildSubscribersCountChange(uint64_t count)
{
if (count == 0) {
M::Model* model = static_cast<M::Model*>(sender());
removeModel(model);
emit serviceMessage(QString("Unregistered model ") + model->getAddress().toString().c_str() + " because there no subscribers left");
model->deleteLater();
}
}
void Database::modifyElement(uint64_t id, const W::Vocabulary& newValue)
{
if (!opened) {
throw 6; //TODO
}
int size = newValue.size();
W::ByteArray ba(size + 1);
ba.push8(newValue.getType());
newValue.serialize(ba);
lmdb::val key((uint8_t*) &id, 8);
lmdb::val value(ba.getData(), ba.size());
lmdb::txn wTrans = lmdb::txn::begin(environment);
dbi.put(wTrans, key, value);
wTrans.commit();
}
uint64_t Database::size() const
{
return elements.size();
}