fix: Move ResetAllDB implementation inside OPENMIND_STORAGE_LEVELDB g…

…uard
ohhmm · Feb 14, 2025 · 936162b · 936162b
1 parent ab6287b
commit 936162b
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/omnn/storage/LevelDbCache.cpp b/omnn/storage/LevelDbCache.cpp
@@ -7,7 +7,7 @@
 #include <string>
 
 
-using namespace omnn::rt::storage; 
+using namespace omnn::rt::storage;
 
 
 LevelDbCache::LevelDbCache(const std::string_view& path)
@@ -40,6 +40,20 @@ LevelDbCache::~LevelDbCache() {
 	delete _db;
 }
 
+bool LevelDbCache::ResetAllDB(const path_str_t& path) {
+    // Delete all keys in the database
+    auto it = _db->NewIterator(leveldb::ReadOptions());
+    for (it->SeekToFirst(); it->Valid(); it->Next()) {
+        auto status = _db->Delete(leveldb::WriteOptions(), it->key());
+        if (!status.ok()) {
+            delete it;
+            return false;
+        }
+    }
+    delete it;
+    return true;
+}
+
 namespace{
 	std::once_flag dbConnectionOptionsInitializedFlag;
 }