diff --git a/cpp/src/db/DB.h b/cpp/src/db/DB.h index acbe60489329b3638f65912fc9fd73a5a6b8bf68..bcc6f90f263ef605fb842efe387a6117a4ce6ad7 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 f18287a0378b3643c1cd956a105a7773765fbc49..9f65b7e2e25f2517826c36fd8b25920e1aae6da3 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 lock(_mutex); diff --git a/cpp/src/db/DBImpl.h b/cpp/src/db/DBImpl.h index 6ecc287dbe102460b60b8118fc6c8393472a6f7f..bb0de8ac5d45d05d50a572ab4f15dddcf953bea0 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 3652046e44f218d7e4b70b69f52659eda9ec82e1..ca92504fac2c50d46aac1cacf6de32071eba0747 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 7ac04433438b17ea2796c38a2d44ecae6b13d183..e4510597b36225e96d2e810c72ba477295ec50e7 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 f7192a05b7c72267df231b0114878c1ed681961a..07b5c1086502a4c404a63a67cb130f75af19152c 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 6c687500e1c5393e1061d3639f5114edec2040ce..2175379a01d702e7b28aa281d79a2c16eaedd0a9 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 4dbc275f656a3804ba34b7d8eb2c2aa3eba4b443..82c7900e47af464fb66bc7153c51ddeff0b9f73c 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();