1 files changed, 188 insertions, 0 deletions
diff --git a/src/test/ObjectMap/KeyValueDBMemory.h b/src/test/ObjectMap/KeyValueDBMemory.h
new file mode 100644
index 000000000..de84ede90
--- /dev/null
+++ b/src/test/ObjectMap/KeyValueDBMemory.h
@@ -0,0 +1,188 @@
+// -*- 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 std::string& cfs="") override {
+    return 0;
+  }
+  int create_and_open(std::ostream &out, const std::string& cfs="") override {
+    return 0;
+  }
+
+  int get(
+    const std::string &prefix,
+    const std::set<std::string> &key,
+    std::map<std::string, bufferlist> *out
+    ) override;
+  using KeyValueDB::get;
+
+  int get_keys(
+    const std::string &prefix,
+    const std::set<std::string> &key,
+    std::set<std::string> *out
+    );
+
+  int set(
+    const std::string &prefix,
+    const std::string &key,
+    const bufferlist &bl
+    );
+
+  int rmkey(
+    const std::string &prefix,
+    const std::string &key
+    );
+
+  int rmkeys_by_prefix(
+    const std::string &prefix
+    );
+
+  int rm_range_keys(
+      const std::string &prefix,
+      const std::string &start,
+      const std::string &end
+      );
+
+  class TransactionImpl_ : public TransactionImpl {
+  public:
+    std::list<Context *> on_commit;
+    KeyValueDBMemory *db;
+
+    explicit TransactionImpl_(KeyValueDBMemory *db) : db(db) {}
+
+
+    struct SetOp : public Context {
+      KeyValueDBMemory *db;
+      std::pair<std::string,std::string> key;
+      bufferlist value;
+      SetOp(KeyValueDBMemory *db,
+	    const std::pair<std::string,std::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 std::string &prefix, const std::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<std::string,std::string> key;
+      RmKeysOp(KeyValueDBMemory *db,
+	       const std::pair<std::string,std::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 std::string &prefix, const std::string &key) override {
+      on_commit.push_back(new RmKeysOp(db, std::make_pair(prefix, key)));
+    }
+
+    struct RmKeysByPrefixOp : public Context {
+      KeyValueDBMemory *db;
+      std::string prefix;
+      RmKeysByPrefixOp(KeyValueDBMemory *db,
+		       const std::string &prefix)
+	: db(db), prefix(prefix) {}
+      void finish(int r) override {
+	db->rmkeys_by_prefix(prefix);
+      }
+    };
+    void rmkeys_by_prefix(const std::string &prefix) override {
+      on_commit.push_back(new RmKeysByPrefixOp(db, prefix));
+    }
+
+    struct RmRangeKeys: public Context {
+      KeyValueDBMemory *db;
+      std::string prefix, start, end;
+      RmRangeKeys(KeyValueDBMemory *db, const std::string &prefix, const std::string &s, const std::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 std::string &prefix, const std::string &start, const std::string &end) {
+      on_commit.push_back(new RmRangeKeys(db, prefix, start, end));
+    }
+
+    int complete() {
+      for (auto i = on_commit.begin();
+	   i != on_commit.end();
+	   on_commit.erase(i++)) {
+	(*i)->complete(0);
+      }
+      return 0;
+    }
+
+    ~TransactionImpl_() override {
+      for (auto 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(std::map<std::string,uint64_t> &extras) override {
+    uint64_t total_size = 0;
+
+    for (auto& [key, bl] : db) {
+      string prefix = key.first;
+
+      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 std::string &prefix) {
+    std::map<std::pair<std::string,std::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(IteratorOpts opts = 0) override;
+};