1 files changed, 190 insertions, 0 deletions
diff --git a/src/test/ObjectMap/KeyValueDBMemory.h b/src/test/ObjectMap/KeyValueDBMemory.h
new file mode 100644
index 00000000..415399e0
--- /dev/null
+++ b/src/test/ObjectMap/KeyValueDBMemory.h
@@ -0,0 +1,190 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+#include <map>
+#include <set>
+#include <string>
+
+#include "kv/KeyValueDB.h"
+#include "include/buffer.h"
+#include "include/Context.h"
+
+using std::string;
+
+class KeyValueDBMemory : public KeyValueDB {
+public:
+  std::map<std::pair<string,string>,bufferlist> db;
+
+  KeyValueDBMemory() { }
+  explicit KeyValueDBMemory(KeyValueDBMemory *db) : db(db->db) { }
+  ~KeyValueDBMemory() override { }
+
+  int init(string _opt) override {
+    return 0;
+  }
+  int open(std::ostream &out, const vector<ColumnFamily>& cfs = {}) override {
+    return 0;
+  }
+  int create_and_open(ostream &out, const vector<ColumnFamily>& cfs = {}) override {
+    return 0;
+  }
+
+  int get(
+    const string &prefix,
+    const std::set<string> &key,
+    std::map<string, bufferlist> *out
+    ) override;
+  using KeyValueDB::get;
+
+  int get_keys(
+    const string &prefix,
+    const std::set<string> &key,
+    std::set<string> *out
+    );
+
+  int set(
+    const string &prefix,
+    const string &key,
+    const bufferlist &bl
+    );
+
+  int rmkey(
+    const string &prefix,
+    const string &key
+    );
+
+  int rmkeys_by_prefix(
+    const string &prefix
+    );
+
+  int rm_range_keys(
+      const string &prefix,
+      const string &start,
+      const string &end
+      );
+
+  class TransactionImpl_ : public TransactionImpl {
+  public:
+    list<Context *> on_commit;
+    KeyValueDBMemory *db;
+
+    explicit TransactionImpl_(KeyValueDBMemory *db) : db(db) {}
+
+
+    struct SetOp : public Context {
+      KeyValueDBMemory *db;
+      std::pair<string,string> key;
+      bufferlist value;
+      SetOp(KeyValueDBMemory *db,
+	    const std::pair<string,string> &key,
+	    const bufferlist &value)
+	: db(db), key(key), value(value) {}
+      void finish(int r) override {
+	db->set(key.first, key.second, value);
+      }
+    };
+
+    void set(const string &prefix, const string &k, const bufferlist& bl) override {
+      on_commit.push_back(new SetOp(db, std::make_pair(prefix, k), bl));
+    }
+
+    struct RmKeysOp : public Context {
+      KeyValueDBMemory *db;
+      std::pair<string,string> key;
+      RmKeysOp(KeyValueDBMemory *db,
+	       const std::pair<string,string> &key)
+	: db(db), key(key) {}
+      void finish(int r) override {
+	db->rmkey(key.first, key.second);
+      }
+    };
+
+    using KeyValueDB::TransactionImpl::rmkey;
+    using KeyValueDB::TransactionImpl::set;
+    void rmkey(const string &prefix, const string &key) override {
+      on_commit.push_back(new RmKeysOp(db, std::make_pair(prefix, key)));
+    }
+
+    struct RmKeysByPrefixOp : public Context {
+      KeyValueDBMemory *db;
+      string prefix;
+      RmKeysByPrefixOp(KeyValueDBMemory *db,
+		       const string &prefix)
+	: db(db), prefix(prefix) {}
+      void finish(int r) override {
+	db->rmkeys_by_prefix(prefix);
+      }
+    };
+    void rmkeys_by_prefix(const string &prefix) override {
+      on_commit.push_back(new RmKeysByPrefixOp(db, prefix));
+    }
+
+    struct RmRangeKeys: public Context {
+      KeyValueDBMemory *db;
+      string prefix, start, end;
+      RmRangeKeys(KeyValueDBMemory *db, const string &prefix, const string &s, const string &e)
+	: db(db), prefix(prefix), start(s), end(e) {}
+      void finish(int r) {
+	db->rm_range_keys(prefix, start, end);
+      }
+    };
+
+    void rm_range_keys(const string &prefix, const string &start, const string &end) {
+      on_commit.push_back(new RmRangeKeys(db, prefix, start, end));
+    }
+
+    int complete() {
+      for (list<Context *>::iterator i = on_commit.begin();
+	   i != on_commit.end();
+	   on_commit.erase(i++)) {
+	(*i)->complete(0);
+      }
+      return 0;
+    }
+
+    ~TransactionImpl_() override {
+      for (list<Context *>::iterator i = on_commit.begin();
+	   i != on_commit.end();
+	   on_commit.erase(i++)) {
+	delete *i;
+      }
+    }
+  };
+
+  Transaction get_transaction() override {
+    return Transaction(new TransactionImpl_(this));
+  }
+
+  int submit_transaction(Transaction trans) override {
+    return static_cast<TransactionImpl_*>(trans.get())->complete();
+  }
+
+  uint64_t get_estimated_size(map<string,uint64_t> &extras) override {
+    uint64_t total_size = 0;
+
+    for (map<pair<string,string>,bufferlist>::iterator p = db.begin();
+         p != db.end(); ++p) {
+      string prefix = p->first.first;
+      bufferlist &bl = p->second;
+
+      uint64_t sz = bl.length();
+      total_size += sz;
+      if (extras.count(prefix) == 0)
+        extras[prefix] = 0;
+      extras[prefix] += sz;
+    }
+
+    return total_size;
+  }
+
+private:
+  bool exists_prefix(const string &prefix) {
+    std::map<std::pair<string,string>,bufferlist>::iterator it;
+    it = db.lower_bound(std::make_pair(prefix, ""));
+    return ((it != db.end()) && ((*it).first.first == prefix));
+  }
+
+  friend class WholeSpaceMemIterator;
+
+public:
+  WholeSpaceIterator get_wholespace_iterator() override;
+};