forked from blue/lmdbal
124 lines
3.5 KiB
C++
124 lines
3.5 KiB
C++
/*
|
|
* LMDB Abstraction Layer.
|
|
* Copyright (C) 2023 Yury Gubich <blue@macaw.me>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "storage.h"
|
|
|
|
LMDBAL::iStorage::iStorage(const std::string& p_name, Base* parent):
|
|
dbi(),
|
|
db(parent),
|
|
name(p_name)
|
|
{}
|
|
|
|
LMDBAL::iStorage::~iStorage() {}
|
|
|
|
void LMDBAL::iStorage::drop() {
|
|
ensureOpened(dropMethodName);
|
|
|
|
MDB_txn *txn;
|
|
int rc = mdb_txn_begin(db->environment, NULL, 0, &txn);
|
|
if (rc) {
|
|
mdb_txn_abort(txn);
|
|
throw Unknown(db->name, mdb_strerror(rc), name);
|
|
}
|
|
rc = drop(txn);
|
|
if (rc) {
|
|
mdb_txn_abort(txn);
|
|
throw Unknown(db->name, mdb_strerror(rc), name);
|
|
}
|
|
|
|
mdb_txn_commit(txn);
|
|
}
|
|
|
|
int LMDBAL::iStorage::drop(MDB_txn* transaction) {
|
|
return mdb_drop(transaction, dbi, 0);
|
|
}
|
|
|
|
const std::string & LMDBAL::iStorage::dbName() const {
|
|
return db->name;}
|
|
|
|
bool LMDBAL::iStorage::isDBOpened() const {
|
|
return db->opened;}
|
|
|
|
void LMDBAL::iStorage::ensureOpened(const std::string& methodName) const {
|
|
if (!db->opened)
|
|
throw Closed(methodName, db->name, name);
|
|
}
|
|
|
|
uint32_t LMDBAL::iStorage::count() const {
|
|
ensureOpened(countMethodName);
|
|
|
|
MDB_txn *txn;
|
|
MDB_stat stat;
|
|
int rc = mdb_txn_begin(db->environment, NULL, MDB_RDONLY, &txn);
|
|
if (rc) {
|
|
mdb_txn_abort(txn);
|
|
throw Unknown(db->name, mdb_strerror(rc), name);
|
|
}
|
|
|
|
rc = mdb_stat(txn, dbi, &stat);
|
|
if (rc) {
|
|
mdb_txn_abort(txn);
|
|
throw Unknown(db->name, mdb_strerror(rc), name);
|
|
}
|
|
uint32_t amount = stat.ms_entries;
|
|
mdb_txn_abort(txn);
|
|
return amount;
|
|
}
|
|
|
|
void LMDBAL::iStorage::throwDuplicateOrUnknown(int rc, TransactionID txn, const std::string& key) const {
|
|
abortTransaction(txn);
|
|
if (rc == MDB_KEYEXIST)
|
|
throwDuplicate(key);
|
|
else
|
|
throwUnknown(rc);
|
|
}
|
|
|
|
void LMDBAL::iStorage::throwNotFoundOrUnknown(int rc, LMDBAL::TransactionID txn, const std::string& key) const {
|
|
abortTransaction(txn);
|
|
if (rc == MDB_NOTFOUND)
|
|
throwNotFound(key);
|
|
else
|
|
throwUnknown(rc);
|
|
}
|
|
|
|
void LMDBAL::iStorage::throwUnknown(int rc, LMDBAL::TransactionID txn) const {
|
|
abortTransaction(txn);
|
|
throwUnknown(rc);
|
|
}
|
|
|
|
void LMDBAL::iStorage::throwUnknown(int rc) const {
|
|
throw Unknown(db->name, mdb_strerror(rc), name);}
|
|
|
|
void LMDBAL::iStorage::throwDuplicate(const std::string& key) const {
|
|
throw Exist(key, db->name, name);}
|
|
|
|
void LMDBAL::iStorage::throwNotFound(const std::string& key) const {
|
|
throw NotFound(key, db->name, name);}
|
|
|
|
LMDBAL::TransactionID LMDBAL::iStorage::beginReadOnlyTransaction() const {
|
|
return db->beginReadOnlyTransaction(name);}
|
|
|
|
LMDBAL::TransactionID LMDBAL::iStorage::beginTransaction() const {
|
|
return db->beginTransaction(name);}
|
|
|
|
void LMDBAL::iStorage::abortTransaction(LMDBAL::TransactionID id) const {
|
|
db->abortTransaction(id);}
|
|
|
|
void LMDBAL::iStorage::commitTransaction(LMDBAL::TransactionID id) const {
|
|
db->commitTransaction(id);}
|