forked from blue/lmdbal
project name change
This commit is contained in:
parent
19229f6c26
commit
7377f13534
29 changed files with 316 additions and 310 deletions
121
src/storage.cpp
Normal file
121
src/storage.cpp
Normal file
|
@ -0,0 +1,121 @@
|
|||
// Squawk messenger.
|
||||
// Copyright (C) 2019 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);}
|
Loading…
Add table
Add a link
Reference in a new issue