From 713d1ec6b798d697accd1f5a7a27885e406cf66a Mon Sep 17 00:00:00 2001
From: Xu Peng <xupeng3112@163.com>
Date: Tue, 30 Apr 2019 15:50:24 +0800
Subject: [PATCH] feat(db): add cache

Former-commit-id: 2dd8078d8df484ddac54379bc6c84c015c1fc530
---
 cpp/src/db/FaissSerializer.cpp | 7 +++++++
 cpp/src/db/FaissSerializer.h   | 2 ++
 cpp/src/db/Serializer.h        | 2 ++
 3 files changed, 11 insertions(+)

diff --git a/cpp/src/db/FaissSerializer.cpp b/cpp/src/db/FaissSerializer.cpp
index fee7750469..b2abb3a19c 100644
--- a/cpp/src/db/FaissSerializer.cpp
+++ b/cpp/src/db/FaissSerializer.cpp
@@ -1,6 +1,7 @@
 #include <easylogging++.h>
 #include <faiss/AutoTune.h>
 #include <wrapper/Index.h>
+#include <cache/CpuCacheMgr.h>
 
 #include "FaissSerializer.h"
 
@@ -33,6 +34,12 @@ Status FaissSerializer::Serialize() {
     return Status::OK();
 }
 
+Status FaissSerializer::Cache() {
+    zilliz::vecwise::cache::CpuCacheMgr::GetInstance(
+            )->InsertItem(location_, std::make_shared<Index>(pIndex_));
+
+    return Status::OK();
+}
 
 } // namespace engine
 } // namespace vecwise
diff --git a/cpp/src/db/FaissSerializer.h b/cpp/src/db/FaissSerializer.h
index a56779996e..d3a255a0bb 100644
--- a/cpp/src/db/FaissSerializer.h
+++ b/cpp/src/db/FaissSerializer.h
@@ -24,6 +24,8 @@ public:
 
     virtual Status Serialize() override;
 
+    virtual Status Cache() override;
+
 protected:
     std::shared_ptr<faiss::Index> pIndex_;
     std::string location_;
diff --git a/cpp/src/db/Serializer.h b/cpp/src/db/Serializer.h
index cb2891be2e..dcb32e0b05 100644
--- a/cpp/src/db/Serializer.h
+++ b/cpp/src/db/Serializer.h
@@ -22,6 +22,8 @@ public:
 
     virtual Status Serialize() = 0;
 
+    virtual Status Cache() = 0;
+
     virtual ~Serializer() {}
 };