From 64e82fe5ccfa10e1652c149d10e89b08a5aae165 Mon Sep 17 00:00:00 2001
From: Christian Mollekopf <chrigi_1@fastmail.fm>
Date: Sat, 22 Nov 2014 12:05:52 +0100
Subject: Buffertest + store

---
 store/database.cpp | 90 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 store/database.h   | 16 ++++++++++
 2 files changed, 106 insertions(+)
 create mode 100644 store/database.cpp
 create mode 100644 store/database.h

(limited to 'store')

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);
+}
+
diff --git a/store/database.h b/store/database.h
new file mode 100644
index 0000000..b42b9f7
--- /dev/null
+++ b/store/database.h
@@ -0,0 +1,16 @@
+#include <lmdb.h>
+#include <string>
+
+class Database {
+public:
+    Database();
+    ~Database();
+    MDB_txn *startTransaction();
+    void endTransaction(MDB_txn *transaction);
+    void write(const std::string &sKey, const std::string &sValue, MDB_txn *transaction);
+    void read(const std::string &sKey);
+
+private:
+    MDB_env *env;
+    MDB_dbi dbi;
+};
-- 
cgit v1.2.3