6068a95a0e
Summary: There was only two files in dbms directory so I moved them to database directory. Reviewers: buda Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D540
61 lines
1.5 KiB
C++
61 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <algorithm>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "gflags/gflags.h"
|
|
|
|
#include "data_structures/concurrent/concurrent_map.hpp"
|
|
#include "database/graph_db.hpp"
|
|
#include "database/graph_db_accessor.hpp"
|
|
#include "durability/recovery.hpp"
|
|
|
|
DECLARE_string(snapshot_directory);
|
|
DECLARE_bool(recover_on_startup);
|
|
|
|
namespace fs = std::experimental::filesystem;
|
|
class Dbms {
|
|
public:
|
|
Dbms() {
|
|
if (FLAGS_recover_on_startup) {
|
|
if (fs::exists(fs::path(FLAGS_snapshot_directory))) {
|
|
auto accessor = dbs.access();
|
|
for (auto &snapshot_db :
|
|
fs::directory_iterator(FLAGS_snapshot_directory)) {
|
|
// create db and set it active
|
|
active(snapshot_db.path().filename(), snapshot_db);
|
|
}
|
|
}
|
|
}
|
|
|
|
// create the default database and set is a active
|
|
active("default");
|
|
}
|
|
|
|
/**
|
|
* Returns an accessor to the active database.
|
|
*/
|
|
std::unique_ptr<GraphDbAccessor> active();
|
|
|
|
/**
|
|
* Set the database with the given name to be active.
|
|
* If there is no database with the given name,
|
|
* it's created. If snapshooting is true, snapshooter starts
|
|
* snapshooting on database creation.
|
|
*
|
|
* @return an accessor to the database with the given name.
|
|
*/
|
|
std::unique_ptr<GraphDbAccessor> active(
|
|
const std::string &name, const fs::path &snapshot_db_dir = fs::path());
|
|
|
|
// TODO: DELETE action
|
|
|
|
private:
|
|
// dbs container
|
|
ConcurrentMap<std::string, GraphDb> dbs;
|
|
|
|
// currently active database
|
|
std::atomic<GraphDb *> active_db;
|
|
};
|