From b2af844f2096c6dfc0b955364a8c543507dce00e Mon Sep 17 00:00:00 2001
From: Xu Peng <xupeng3112@163.com>
Date: Fri, 24 May 2019 18:15:50 +0800
Subject: [PATCH] feat(db): add size api for db

Former-commit-id: 9d20366e22996fc7bb3e7e983ebfa9999591ba6b
---
 cpp/src/db/DB.h       | 2 ++
 cpp/src/db/DBImpl.cpp | 5 +++++
 cpp/src/db/DBImpl.h   | 2 ++
 3 files changed, 9 insertions(+)

diff --git a/cpp/src/db/DB.h b/cpp/src/db/DB.h
index 6f922475..7d976ad8 100644
--- a/cpp/src/db/DB.h
+++ b/cpp/src/db/DB.h
@@ -39,6 +39,8 @@ public:
     virtual Status search(const std::string& group_id, size_t k, size_t nq,
             const float* vectors, const meta::DatesT& dates, QueryResults& results) = 0;
 
+    virtual Status size(long& result) = 0;
+
     virtual Status drop_all() = 0;
 
     virtual Status count(const std::string& group_id, long& result) = 0;
diff --git a/cpp/src/db/DBImpl.cpp b/cpp/src/db/DBImpl.cpp
index eee492ab..971e6daf 100644
--- a/cpp/src/db/DBImpl.cpp
+++ b/cpp/src/db/DBImpl.cpp
@@ -427,6 +427,11 @@ Status DBImpl<EngineT>::count(const std::string& group_id, long& result) {
     return _pMeta->count(group_id, result);
 }
 
+template<typename EngineT>
+Status DBImpl<EngineT>::size(long& result) {
+    return  _pMeta->size(result);
+}
+
 template<typename EngineT>
 DBImpl<EngineT>::~DBImpl() {
     {
diff --git a/cpp/src/db/DBImpl.h b/cpp/src/db/DBImpl.h
index 45f249bd..54c22eb4 100644
--- a/cpp/src/db/DBImpl.h
+++ b/cpp/src/db/DBImpl.h
@@ -55,6 +55,8 @@ public:
 
     virtual Status count(const std::string& group_id, long& result) override;
 
+    virtual Status size(long& result) override;
+
     virtual ~DBImpl();
 
 private:
-- 
GitLab