From d6344006e541b640cce9335cfe18a0560c8b81db Mon Sep 17 00:00:00 2001
From: Xu Peng <xupeng3112@163.com>
Date: Tue, 23 Apr 2019 20:03:55 +0800
Subject: [PATCH] feat(db): add drop_all

Former-commit-id: 70720144d39f5ca18d349e475e66fb6ccca7bacf
---
 cpp/src/db/DB.h              | 2 ++
 cpp/src/db/DBImpl.cpp        | 4 ++++
 cpp/src/db/DBImpl.h          | 2 ++
 cpp/src/db/DBMetaImpl.cpp    | 7 +++++++
 cpp/src/db/DBMetaImpl.h      | 2 ++
 cpp/src/db/LocalMetaImpl.cpp | 5 +++++
 cpp/src/db/LocalMetaImpl.h   | 2 ++
 cpp/src/db/Meta.h            | 2 ++
 8 files changed, 26 insertions(+)

diff --git a/cpp/src/db/DB.h b/cpp/src/db/DB.h
index acbe6048..bcc6f90f 100644
--- a/cpp/src/db/DB.h
+++ b/cpp/src/db/DB.h
@@ -30,6 +30,8 @@ public:
     virtual Status search(const std::string& group_id, size_t k, size_t nq,
             const float* vectors, QueryResults& results) = 0;
 
+    virtual Status drop_all() = 0;
+
     DB() = default;
     DB(const DB&) = delete;
     DB& operator=(const DB&) = delete;
diff --git a/cpp/src/db/DBImpl.cpp b/cpp/src/db/DBImpl.cpp
index f18287a0..9f65b7e2 100644
--- a/cpp/src/db/DBImpl.cpp
+++ b/cpp/src/db/DBImpl.cpp
@@ -344,6 +344,10 @@ void DBImpl::background_compaction() {
     }
 }
 
+Status DBImpl::drop_all() {
+    return _pMeta->drop_all();
+}
+
 DBImpl::~DBImpl() {
     {
         std::unique_lock<std::mutex> lock(_mutex);
diff --git a/cpp/src/db/DBImpl.h b/cpp/src/db/DBImpl.h
index 6ecc287d..bb0de8ac 100644
--- a/cpp/src/db/DBImpl.h
+++ b/cpp/src/db/DBImpl.h
@@ -36,6 +36,8 @@ public:
     virtual Status search(const std::string& group_id, size_t k, size_t nq,
             const float* vectors, QueryResults& results) override;
 
+    virtual Status drop_all() override;
+
     virtual ~DBImpl();
 
 private:
diff --git a/cpp/src/db/DBMetaImpl.cpp b/cpp/src/db/DBMetaImpl.cpp
index 3652046e..ca92504f 100644
--- a/cpp/src/db/DBMetaImpl.cpp
+++ b/cpp/src/db/DBMetaImpl.cpp
@@ -427,6 +427,13 @@ Status DBMetaImpl::cleanup() {
     return Status::OK();
 }
 
+Status DBMetaImpl::drop_all() {
+    if (boost::filesystem::is_directory(_options.path)) {
+        boost::filesystem::remove_all(_options.path);
+    }
+    return Status::OK();
+}
+
 DBMetaImpl::~DBMetaImpl() {
     cleanup();
 }
diff --git a/cpp/src/db/DBMetaImpl.h b/cpp/src/db/DBMetaImpl.h
index 7ac04433..e4510597 100644
--- a/cpp/src/db/DBMetaImpl.h
+++ b/cpp/src/db/DBMetaImpl.h
@@ -48,6 +48,8 @@ public:
 
     virtual Status cleanup_ttl_files(uint16_t seconds) override;
 
+    virtual Status drop_all() override;
+
     virtual ~DBMetaImpl();
 
 private:
diff --git a/cpp/src/db/LocalMetaImpl.cpp b/cpp/src/db/LocalMetaImpl.cpp
index f7192a05..07b5c108 100644
--- a/cpp/src/db/LocalMetaImpl.cpp
+++ b/cpp/src/db/LocalMetaImpl.cpp
@@ -246,6 +246,11 @@ Status LocalMetaImpl::cleanup_ttl_files(uint16_t seconds) {
     return Status::OK();
 }
 
+Status LocalMetaImpl::drop_all() {
+    // PXU TODO
+    return Status::OK();
+}
+
 } // namespace meta
 } // namespace engine
 } // namespace vecwise
diff --git a/cpp/src/db/LocalMetaImpl.h b/cpp/src/db/LocalMetaImpl.h
index 6c687500..2175379a 100644
--- a/cpp/src/db/LocalMetaImpl.h
+++ b/cpp/src/db/LocalMetaImpl.h
@@ -43,6 +43,8 @@ public:
 
     virtual Status cleanup_ttl_files(uint16_t seconds) override;
 
+    virtual Status drop_all() override;
+
 private:
 
     Status GetGroupMetaInfoByPath(const std::string& path, GroupSchema& group_info);
diff --git a/cpp/src/db/Meta.h b/cpp/src/db/Meta.h
index 4dbc275f..82c7900e 100644
--- a/cpp/src/db/Meta.h
+++ b/cpp/src/db/Meta.h
@@ -83,6 +83,8 @@ public:
     virtual Status cleanup() = 0;
     virtual Status cleanup_ttl_files(uint16_t) = 0;
 
+    virtual Status drop_all() = 0;
+
     static DateT GetDate(const std::time_t& t);
     static DateT GetDate();
 
-- 
GitLab