shunf4/squawk
2
0
Fork 0
forked from blue/squawk
Code Issues Pull requests Projects Releases Wiki Activity

New object for cached database, also ClientInfo class

Browse source
This commit is contained in:
Blue 2022-08-20 00:28:59 +03:00
parent d162494ec8
commit 2ae75a4b91
Signed by untrusted user: blue
GPG key ID: 9B203B252A63EE38
10 changed files with 287 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

52
core/storage/storage.cpp
View file

@ -21,7 +21,12 @@
#include "storage.h"
Core::Storage::Storage(const QString& p_name):
#include <shared/clientinfo.h>
template class Core::Storage<Shared::ClientInfo>;
template <class T>
Core::Storage<T>::Storage(const QString& p_name):
name(p_name),
opened(false),
environment(),
@ -29,12 +34,14 @@ Core::Storage::Storage(const QString& p_name):
{
}
Core::Storage::~Storage()
template <class T>
Core::Storage<T>::~Storage()
{
close();
}
void Core::Storage::open()
template <class T>
void Core::Storage<T>::open()
{
if (!opened) {
mdb_env_create(&environment);
@ -61,7 +68,8 @@ void Core::Storage::open()
}
}
void Core::Storage::close()
template <class T>
void Core::Storage<T>::close()
{
if (opened) {
mdb_dbi_close(environment, base);
@ -70,19 +78,22 @@ void Core::Storage::close()
}
}
void Core::Storage::addRecord(const QString& key, const QString& value)
template <class T>
void Core::Storage<T>::addRecord(const QString& key, const T& value)
{
if (!opened) {
throw Archive::Closed("addRecord", name.toStdString());
}
QByteArray ba;
QDataStream ds(&ba, QIODevice::WriteOnly);
const std::string& id = key.toStdString();
const std::string& val = value.toStdString();
ds << value;
MDB_val lmdbKey, lmdbData;
lmdbKey.mv_size = id.size();
lmdbKey.mv_data = (char*)id.c_str();
lmdbData.mv_size = val.size();
lmdbData.mv_data = (char*)val.c_str();
lmdbData.mv_size = ba.size();
lmdbData.mv_data = (uint8_t*)ba.data();
MDB_txn *txn;
mdb_txn_begin(environment, NULL, 0, &txn);
int rc;
@ -99,19 +110,23 @@ void Core::Storage::addRecord(const QString& key, const QString& value)
}
}
void Core::Storage::changeRecord(const QString& key, const QString& value)
template <class T>
void Core::Storage<T>::changeRecord(const QString& key, const T& value)
{
if (!opened) {
throw Archive::Closed("changeRecord", name.toStdString());
}
QByteArray ba;
QDataStream ds(&ba, QIODevice::WriteOnly);
const std::string& id = key.toStdString();
const std::string& val = value.toStdString();
ds << value;
MDB_val lmdbKey, lmdbData;
lmdbKey.mv_size = id.size();
lmdbKey.mv_data = (char*)id.c_str();
lmdbData.mv_size = val.size();
lmdbData.mv_data = (char*)val.c_str();
lmdbData.mv_size = ba.size();
lmdbData.mv_data = (uint8_t*)ba.data();
MDB_txn *txn;
mdb_txn_begin(environment, NULL, 0, &txn);
int rc;
@ -126,7 +141,8 @@ void Core::Storage::changeRecord(const QString& key, const QString& value)
}
}
QString Core::Storage::getRecord(const QString& key) const
template <class T>
T Core::Storage<T>::getRecord(const QString& key) const
{
if (!opened) {
throw Archive::Closed("addElement", name.toStdString());
@ -149,14 +165,18 @@ QString Core::Storage::getRecord(const QString& key) const
throw Archive::Unknown(name.toStdString(), mdb_strerror(rc));
}
} else {
std::string sId((char*)lmdbData.mv_data, lmdbData.mv_size);
QString value(sId.c_str());
QByteArray ba((char*)lmdbData.mv_data, lmdbData.mv_size);
QDataStream ds(&ba, QIODevice::ReadOnly);
T value;
ds >> value;
mdb_txn_abort(txn);
return value;
}
}
void Core::Storage::removeRecord(const QString& key)
template <class T>
void Core::Storage<T>::removeRecord(const QString& key)
{
if (!opened) {
throw Archive::Closed("addElement", name.toStdString());