Fixed a bug in HaloDBStore.truncate not always clearing every value

outr · Nov 14, 2024 · decbe6c · decbe6c
1 parent 796b56e
commit decbe6c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/halodb/src/main/scala/lightdb/halodb/HaloDBStore.scala b/halodb/src/main/scala/lightdb/halodb/HaloDBStore.scala
@@ -96,8 +96,12 @@ class HaloDBStore[Doc <: Document[Doc], Model <: DocumentModel[Doc]](directory:
 
   override def truncate()(implicit transaction: Transaction[Doc]): Int = {
     val size = count
-    instance.newIterator().asScala.foreach(r => instance.delete(r.getKey))
-    size
+    if (size == 0) {
+      0
+    } else {
+      instance.newIterator().asScala.foreach(r => instance.delete(r.getKey))
+      size + truncate()
+    }
   }
 
   override def dispose(): Unit = {