dbmanager.cpp

#include "dbmanager.h"
#include <string>
#include <vector>
#include <map>
#include <rocksdb/db.h>
#include <boost/range/adaptors.hpp>
#include <boost/range/algorithm.hpp>
#include <folly/Singleton.h>
namespace cxxdoor {

namespace {
struct DbManagerTag {
};
}

static folly::Singleton<DbManager, DbManagerTag> the_instance;

DbManager::DbManager(const std::string &db_name) : _db_name(db_name) {
  LOG(INFO) << "Initializing DbManager";
  //rocksdb::DB *db;
  rocksdb::Options options;
  options.create_if_missing = true;
  options.create_missing_column_families = true;
  //options.
  std::vector<std::string> cfNames;
  std::vector<rocksdb::ColumnFamilyDescriptor> cfDescr;
  std::vector<rocksdb::ColumnFamilyHandle *> cfHandles;
  auto s = rocksdb::DB::ListColumnFamilies(options, _db_name, &cfNames);
  for (auto &cfName: cfNames) {
    cfDescr.push_back(rocksdb::ColumnFamilyDescriptor(cfName, rocksdb::ColumnFamilyOptions()));
  }
  rocksdb::DB::Open(options, _db_name, cfDescr, &cfHandles, &_db);
  if (!s.ok()) {
    LOG(WARNING) << s.ToString();
  }
  LOG(INFO) << "Listing column families";
  for (auto cfHandle: cfHandles) {
    LOG(INFO) << cfHandle->GetName();
    auto name = cfHandle->GetName();
    _columnFamilies[name] = cfHandle;
  }


}

std::shared_ptr<DbManager> DbManager::getInstance() {
  return the_instance.try_get();
}

DbManager::~DbManager() {
  LOG(INFO) << "Destroying DB";
  for (const auto &cf: _columnFamilies) {
    try {

      auto h = cf.second;
      LOG(INFO) << "Destroying ColumnFamilyHandle: " << h->GetName();
      auto s = _db->DestroyColumnFamilyHandle(h);
      // h = nullptr;
      if (!s.ok()) {
        LOG(WARNING) << s.ToString();
      }
    } catch (std::exception &e) {
      LOG(ERROR) << e.what();
    }
  }
  delete _db;
}
}