From 617adf4090fbaac3a3b2d943548eeb6516fb9bdd Mon Sep 17 00:00:00 2001
From: Chris Xiong <chirs241097@gmail.com>
Date: Mon, 26 Sep 2022 23:38:30 -0400
Subject: Add show db engine memory usage.

---
 xsig/include/signature_db.hpp | 2 ++
 xsig/src/signature_db.cpp     | 7 +++++++
 2 files changed, 9 insertions(+)

(limited to 'xsig')

diff --git a/xsig/include/signature_db.hpp b/xsig/include/signature_db.hpp
index 23e3aae..b98b22a 100644
--- a/xsig/include/signature_db.hpp
+++ b/xsig/include/signature_db.hpp
@@ -94,6 +94,8 @@ public:
 
     std::vector<std::pair<size_t, double>> search_image(const fs::path &path, const populate_cfg_t &cfg, bool insert = false);
 
+    int64_t db_memory_usage();
+
     //disjoint set for keeping similar images in the same group
     //some of these probably shouldn't be public. TBD...
     void ds_init();
diff --git a/xsig/src/signature_db.cpp b/xsig/src/signature_db.cpp
index 3c4187b..105106b 100644
--- a/xsig/src/signature_db.cpp
+++ b/xsig/src/signature_db.cpp
@@ -479,6 +479,13 @@ std::vector<std::pair<size_t, double>> signature_db::search_image(const fs::path
     return ret;
 }
 
+int64_t signature_db::db_memory_usage()
+{
+    sqlite3_int64 r, dummy;
+    sqlite3_status64(SQLITE_STATUS_MEMORY_USED, &r, &dummy, 0);
+    return static_cast<int64_t>(r);
+}
+
 void signature_db::ds_init()
 {
     sqlite3_exec(p->db, R"sql(
-- 
cgit v1.2.3