diff options
Diffstat (limited to 'src/rgw/driver/dbstore/dbstore_mgr.cc')
-rw-r--r-- | src/rgw/driver/dbstore/dbstore_mgr.cc | 140 |
1 files changed, 140 insertions, 0 deletions
diff --git a/src/rgw/driver/dbstore/dbstore_mgr.cc b/src/rgw/driver/dbstore/dbstore_mgr.cc new file mode 100644 index 000000000..6835f526b --- /dev/null +++ b/src/rgw/driver/dbstore/dbstore_mgr.cc @@ -0,0 +1,140 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab + +#include "dbstore_mgr.h" +#include "common/dbstore_log.h" + +#include <filesystem> + +static constexpr auto dout_subsys = ceph_subsys_rgw; + +using namespace std; + + +/* Given a tenant, find and return the DBStore handle. + * If not found and 'create' set to true, create one + * and return + */ +DB *DBStoreManager::getDB (string tenant, bool create) +{ + map<string, DB*>::iterator iter; + DB *dbs = nullptr; + pair<map<string, DB*>::iterator,bool> ret; + + if (tenant.empty()) + return default_db; + + if (DBStoreHandles.empty()) + goto not_found; + + iter = DBStoreHandles.find(tenant); + + if (iter != DBStoreHandles.end()) + return iter->second; + +not_found: + if (!create) + return nullptr; + + dbs = createDB(tenant); + + return dbs; +} + +/* Create DBStore instance */ +DB *DBStoreManager::createDB(std::string tenant) { + DB *dbs = nullptr; + pair<map<string, DB*>::iterator,bool> ret; + const auto& db_path = g_conf().get_val<std::string>("dbstore_db_dir"); + const auto& db_name = g_conf().get_val<std::string>("dbstore_db_name_prefix") + "-" + tenant; + + auto db_full_path = std::filesystem::path(db_path) / db_name; + ldout(cct, 0) << "DB initialization full db_path("<<db_full_path<<")" << dendl; + + /* Create the handle */ +#ifdef SQLITE_ENABLED + dbs = new SQLiteDB(db_full_path.string(), cct); +#else + dbs = new DB(db_full_path.string(), cct); +#endif + + /* API is DB::Initialize(string logfile, int loglevel); + * If none provided, by default write in to dbstore.log file + * created in current working directory with loglevel L_EVENT. + * XXX: need to align these logs to ceph location + */ + if (dbs->Initialize("", -1) < 0) { + ldout(cct, 0) << "DB initialization failed for tenant("<<tenant<<")" << dendl; + + delete dbs; + return nullptr; + } + + /* XXX: Do we need lock to protect this map? + */ + ret = DBStoreHandles.insert(pair<string, DB*>(tenant, dbs)); + + /* + * Its safe to check for already existing entry (just + * incase other thread raced and created the entry) + */ + if (ret.second == false) { + /* Entry already created by another thread */ + delete dbs; + + dbs = ret.first->second; + } + + return dbs; +} + +void DBStoreManager::deleteDB(string tenant) { + map<string, DB*>::iterator iter; + DB *dbs = nullptr; + + if (tenant.empty() || DBStoreHandles.empty()) + return; + + /* XXX: Check if we need to perform this operation under a lock */ + iter = DBStoreHandles.find(tenant); + + if (iter == DBStoreHandles.end()) + return; + + dbs = iter->second; + + DBStoreHandles.erase(iter); + dbs->Destroy(dbs->get_def_dpp()); + delete dbs; + + return; +} + +void DBStoreManager::deleteDB(DB *dbs) { + if (!dbs) + return; + + (void)deleteDB(dbs->getDBname()); +} + + +void DBStoreManager::destroyAllHandles(){ + map<string, DB*>::iterator iter; + DB *dbs = nullptr; + + if (DBStoreHandles.empty()) + return; + + for (iter = DBStoreHandles.begin(); iter != DBStoreHandles.end(); + ++iter) { + dbs = iter->second; + dbs->Destroy(dbs->get_def_dpp()); + delete dbs; + } + + DBStoreHandles.clear(); + + return; +} + + |