shift lmdb initialization to provider constructor which is called only once

This commit is contained in:
tomasz.ziolkowski
2022-04-29 06:49:00 +02:00
parent 00483e4009
commit 82326ffe2b
2 changed files with 12 additions and 32 deletions

View File

@@ -501,26 +501,8 @@ end_txn:
} }
MDBEnvProvider::MDBEnvProvider() : MDBEnvProvider::MDBEnvProvider() : m_env(NULL) {
m_env(NULL), initialized(false) {
pthread_mutex_init(&m_lock, NULL);
}
MDB_env* MDBEnvProvider::GetEnv() {
init();
return m_env;
}
MDB_dbi* MDBEnvProvider::GetDBI() {
init();
return &m_dbi;
}
void MDBEnvProvider::init() {
pthread_mutex_lock(&m_lock);
if (!initialized) {
MDB_txn *txn; MDB_txn *txn;
initialized = true;
mdb_env_create(&m_env); mdb_env_create(&m_env);
mdb_env_open(m_env, "./modsec-shared-collections", mdb_env_open(m_env, "./modsec-shared-collections",
MDB_WRITEMAP | MDB_NOSUBDIR, 0664); MDB_WRITEMAP | MDB_NOSUBDIR, 0664);
@@ -528,17 +510,19 @@ void MDBEnvProvider::init() {
mdb_dbi_open(txn, NULL, MDB_CREATE | MDB_DUPSORT, &m_dbi); mdb_dbi_open(txn, NULL, MDB_CREATE | MDB_DUPSORT, &m_dbi);
mdb_txn_commit(txn); mdb_txn_commit(txn);
} }
pthread_mutex_unlock(&m_lock);
MDB_env* MDBEnvProvider::GetEnv() {
return m_env;
} }
void MDBEnvProvider::close() { MDB_dbi* MDBEnvProvider::GetDBI() {
pthread_mutex_lock(&m_lock); return &m_dbi;
if (initialized) { }
MDBEnvProvider::~MDBEnvProvider() {
mdb_dbi_close(m_env, m_dbi); mdb_dbi_close(m_env, m_dbi);
mdb_env_close(m_env); mdb_env_close(m_env);
} }
pthread_mutex_unlock(&m_lock);
}
#endif #endif

View File

@@ -83,16 +83,12 @@ class MDBEnvProvider {
} }
MDB_env* GetEnv(); MDB_env* GetEnv();
MDB_dbi* GetDBI(); MDB_dbi* GetDBI();
~MDBEnvProvider();
private: private:
MDB_env *m_env; MDB_env *m_env;
MDB_dbi m_dbi; MDB_dbi m_dbi;
pthread_mutex_t m_lock;
MDBEnvProvider(); MDBEnvProvider();
bool initialized;
void init();
void close();
}; };
class LMDB : class LMDB :