add prioritized cache wrapper

Signed-off-by: haoxiang47 <[email protected]>

include/titan/prioritized_cache.h

@@ -0,0 +1,29 @@
+#pragma once
+#include "rocksdb/cache.h"
+
+namespace rocksdb {
+namespace titandb {
+
+class PrioritizedCache : public Cache {
+ public:
+  // Constructs a PrioritizedCache, which is a wrapper
+  // of rocksdb Cache
+  PrioritizedCache(std::shared_ptr<Cache> cache, Cache::Priority priority);
+  ~PrioritizedCache();
+
+  // Get the Cache ptr
+  std::shared_ptr<Cache> GetCache();
+
+  // always insert into the cache with the priority when init,
+  // regardless of user provided option
+  Status Insert(const Slice& key, void* value, size_t charge,
+                void (*deleter)(const Slice& key, void* value),
+                Cache::Handle** handle, Cache::Priority /*priority*/) override;
+
+ private:
+  std::shared_ptr<Cache> cache_;
+  Cache::Priority priority_;
+};
+
+}  // namespace titandb
+}  // namespace rocksdb

src/prioritized_cache.cc

@@ -0,0 +1,23 @@
+#include "titan/prioritized_cache.h"
+
+namespace rocksdb {
+namespace titandb {
+
+PrioritizedCache::PrioritizedCache(std::shared_ptr<Cache> cache,
+                                   Cache::Priority priority)
+    : cache_(cache), priority_(priority) {}
+
+PrioritizedCache::~PrioritizedCache() {}
+
+std::shared_ptr<Cache> PrioritizedCache::GetCache() { return cache_; }
+
+Status PrioritizedCache::Insert(const Slice& key, void* value, size_t charge,
+                                void (*deleter)(const Slice& key, void* value),
+                                Cache::Handle** handle,
+                                Cache::Priority /*priority*/) {
+  assert(cache_);
+  return cache_->Insert(key, value, charge, deleter, handle, priority_);
+}
+
+}  // namespace titandb
+}  // namespace rocksdb