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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#include <cstring>
#include "util/log.h"
#include "DB.h"
using namespace std;
using namespace crepe;
DB::DB(const string & path) {
dbg_trace();
int ret;
// init database struct
libdb::DB * db;
if ((ret = libdb::db_create(&db, NULL, 0)) != 0)
throw nullptr; // TODO: exception
this->db = { db, [] (libdb::DB * db) { db->close(db, 0); } };
// load or create database file
if ((ret = this->db->open(this->db.get(), NULL, path.c_str(), NULL, libdb::DB_BTREE, DB_CREATE, 0)) != 0) {
throw nullptr;
}
// create cursor
libdb::DBC * cursor;
if ((ret = this->db->cursor(this->db.get(), NULL, &cursor, 0)) != 0) {
throw nullptr;
}
this->cursor = { cursor, [] (libdb::DBC * cursor) { cursor->close(cursor); } };
}
libdb::DBT DB::to_thing(const string & thing) const {
libdb::DBT thang;
memset(&thang, 0, sizeof(libdb::DBT));
thang.data = (void *) thing.data();
thang.size = thing.size();
return thang;
}
string DB::get(const string & key) {
libdb::DBT db_key = this->to_thing(key);
libdb::DBT db_val;
memset(&db_val, 0, sizeof(libdb::DBT));
int ret = this->cursor->get(this->cursor.get(), &db_key, &db_val, DB_FIRST);
if (ret != 0) throw nullptr; // TODO: proper exception
return { static_cast<char *>(db_val.data), db_val.size };
}
void DB::set(const string & key, const string & value) {
libdb::DBT db_key = this->to_thing(key);
libdb::DBT db_val = this->to_thing(value);
int ret = this->db->put(this->db.get(), NULL, &db_key, &db_val, 0);
if (ret != 0) throw nullptr; // TODO: proper exception
}
bool DB::has(const std::string & key) noexcept {
try {
this->get(key);
} catch (...) {
return false;
}
return true;
}
|