-
Notifications
You must be signed in to change notification settings - Fork 68
/
kvdb.cpp
53 lines (47 loc) · 1.36 KB
/
kvdb.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
/*
* FILE: kvdb.cpp
* Created: Jun 05, 2018
* @copyright 2016-19 Clarity Genomics BVBA
*/
#include "kvdb.hpp"
#include "common.hpp"
#include <iostream>
#include <filesystem>
KeyValueDatabase::KeyValueDatabase(std::string const &kvdbPath)
{
auto exists = std::filesystem::exists(kvdbPath);
auto is_empty = std::filesystem::is_empty(kvdbPath);
if (exists && !is_empty)
{
std::cout << STAMP << "Path '" << kvdbPath << "' exists with the following content:" << std::endl;
for (auto& subpath : std::filesystem::directory_iterator(kvdbPath))
std::cout << subpath.path().filename() << std::endl;
}
// init and open key-value database for read matches
options.IncreaseParallelism();
#if defined(_WIN32)
options.compression = rocksdb::kXpressCompression;
#else
options.compression = rocksdb::kZlibCompression;
#endif
options.create_if_missing = true;
rocksdb::Status s = rocksdb::DB::Open(options, kvdbPath, &kvdb);
assert(s.ok());
}
/*
* Remove database files from the given location
*/
int KeyValueDatabase::clear(std::string dbpath)
{
return 0;
} // ~KeyValueDatabase::clear
void KeyValueDatabase::put(std::string key, std::string val)
{
rocksdb::Status s = kvdb->Put(rocksdb::WriteOptions(), key, val);
}
std::string KeyValueDatabase::get(std::string key)
{
std::string val;
rocksdb::Status s = kvdb->Get(rocksdb::ReadOptions(), key, &val);
return val;
}