Add --storage-mode flag (#1282)

* Add --storage-mode flag
This commit is contained in:
Andi
2023-09-26 14:47:30 +02:00
committed by GitHub
parent efdf7baea0
commit 61ac7e1b11
12 changed files with 103 additions and 7 deletions

View File

@@ -10,8 +10,10 @@
// licenses/APL.txt.
#include "dbms/database.hpp"
#include "flags/storage_mode.hpp"
#include "storage/v2/disk/storage.hpp"
#include "storage/v2/inmemory/storage.hpp"
#include "storage/v2/storage_mode.hpp"
template struct memgraph::utils::Gatekeeper<memgraph::dbms::Database>;
@@ -20,10 +22,11 @@ namespace memgraph::dbms {
Database::Database(const storage::Config &config)
: trigger_store_(config.durability.storage_directory / "triggers"),
streams_{config.durability.storage_directory / "streams"} {
if (config.force_on_disk || utils::DirExists(config.disk.main_storage_directory)) {
if (config.storage_mode == memgraph::storage::StorageMode::ON_DISK_TRANSACTIONAL || config.force_on_disk ||
utils::DirExists(config.disk.main_storage_directory)) {
storage_ = std::make_unique<storage::DiskStorage>(config);
} else {
storage_ = std::make_unique<storage::InMemoryStorage>(config);
storage_ = std::make_unique<storage::InMemoryStorage>(config, config.storage_mode);
}
}