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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#include "database.h"
#include <iostream>
#include <QDir>
#include <QString>
#include <QTime>
#include <qdebug.h>
Database::Database(const QString &path)
{
int rc;
QDir dir;
dir.mkdir(path);
//create file
rc = mdb_env_create(&env);
rc = mdb_env_open(env, path.toStdString().data(), 0, 0664);
const int dbSize = 10485760*100; //10MB * 100
mdb_env_set_mapsize(env, dbSize);
if (rc) {
std::cerr << "mdb_env_open: " << rc << mdb_strerror(rc) << std::endl;
}
}
Database::~Database()
{
mdb_close(env, dbi);
mdb_env_close(env);
}
MDB_txn *Database::startTransaction()
{
int rc;
MDB_txn *transaction;
rc = mdb_txn_begin(env, NULL, 0, &transaction);
rc = mdb_open(transaction, NULL, 0, &dbi);
return transaction;
}
void Database::endTransaction(MDB_txn *transaction)
{
int rc;
rc = mdb_txn_commit(transaction);
if (rc) {
std::cerr << "mdb_txn_commit: " << rc << mdb_strerror(rc) << std::endl;
}
}
void Database::write(const std::string &sKey, const std::string &sValue, MDB_txn *transaction)
{
int rc;
MDB_val key, data;
key.mv_size = sKey.size();
key.mv_data = (void*)sKey.data();
data.mv_size = sValue.size();
data.mv_data = (void*)sValue.data();
rc = mdb_put(transaction, dbi, &key, &data, 0);
if (rc) {
std::cerr << "mdb_put: " << rc << mdb_strerror(rc) << std::endl;
}
}
void Database::read(const std::string &sKey, const std::function<void(const std::string)> &resultHandler)
{
int rc;
MDB_txn *txn;
MDB_val key;
MDB_val data;
MDB_cursor *cursor;
key.mv_size = sKey.size();
key.mv_data = (void*)sKey.data();
rc = mdb_txn_begin(env, NULL, MDB_RDONLY, &txn);
rc = mdb_cursor_open(txn, dbi, &cursor);
if (sKey.empty()) {
std::cout << "Iterating over all values of store!" << std::endl;
while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) == 0) {
const std::string resultKey(static_cast<char*>(key.mv_data), key.mv_size);
const std::string resultValue(static_cast<char*>(data.mv_data), data.mv_size);
// std::cout << "key: " << resultKey << " data: " << resultValue << std::endl;
resultHandler(resultValue);
}
} else {
if ((rc = mdb_cursor_get(cursor, &key, &data, MDB_SET)) == 0) {
const std::string resultKey(static_cast<char*>(key.mv_data), key.mv_size);
const std::string resultValue(static_cast<char*>(data.mv_data), data.mv_size);
// std::cout << "key: " << resultKey << " data: " << resultValue << std::endl;
resultHandler(resultValue);
} else {
std::cout << "couldn't find value " << sKey << std::endl;
}
}
if (rc) {
std::cerr << "mdb_cursor_get: " << rc << mdb_strerror(rc) << std::endl;
}
mdb_cursor_close(cursor);
mdb_txn_abort(txn);
}
|
