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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
#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);
}
ReadTransaction::ReadTransaction(const QString &path)
{
int rc;
//create file
rc = mdb_env_create(&env);
//FIXME MDB_NOTLS required to so we can have multiple read-only transactions per resource?
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;
}
}
ReadTransaction::~ReadTransaction()
{
mdb_txn_abort(txn);
mdb_dbi_close(env, dbi);
mdb_env_close(env);
}
void ReadTransaction::read(const std::string &sKey, const std::function<void(void *data, int size)> &resultHandler)
{
int rc;
MDB_val key;
MDB_val data;
MDB_cursor *cursor;
key.mv_size = sKey.size();
key.mv_data = (void*)sKey.data();
{
//A write transaction is at least required the first time
rc = mdb_txn_begin(env, nullptr, 0, &txn);
//Open the database
//With this we could open multiple named databases if we wanted to
rc = mdb_dbi_open(txn, nullptr, 0, &dbi);
mdb_txn_abort(txn);
}
rc = mdb_txn_begin(env, nullptr, MDB_RDONLY, &txn);
rc = mdb_cursor_open(txn, dbi, &cursor);
if (rc) {
std::cerr << "mdb_cursor_open: " << rc << mdb_strerror(rc) << std::endl;
}
if (sKey.empty()) {
std::cout << "Iterating over all values of store!" << std::endl;
rc = mdb_cursor_get(cursor, &key, &data, MDB_FIRST);
while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) == 0) {
resultHandler(data.mv_data, data.mv_size);
}
//We never find the last value
if (rc == MDB_NOTFOUND) {
rc = 0;
}
} else {
if ((rc = mdb_cursor_get(cursor, &key, &data, MDB_SET)) == 0) {
resultHandler(data.mv_data, data.mv_size);
} 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);
//We keep the transaction open since we want to keep the returned values alive
}
|
