1 files changed, 90 insertions, 0 deletions

diff --git a/store/database.cpp b/store/database.cpp
new file mode 100644
index 0000000..d6fe7c0
--- /dev/null
+++ b/store/database.cpp
@@ -0,0 +1,90 @@
+#include "database.h"
+
+#include <iostream>
+#include <QDir>
+#include <QString>
+#include <QTime>
+#include <qdebug.h>
+
+Database::Database()
+{
+    int rc;
+
+    QDir dir;
+    dir.mkdir("./testdb");
+
+    //create file
+    rc = mdb_env_create(&env);
+    rc = mdb_env_open(env, "./testdb", 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)
+{
+    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);
+    // while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) == 0) {
+    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;
+    } else {
+        std::cout << "couldn't find value " << sKey << std::endl;
+    }
+    mdb_cursor_close(cursor);
+    mdb_txn_abort(txn);
+}
+