minor

tomfran · Oct 13, 2023 · fde2dc0 · fde2dc0
1 parent 98b8593
commit fde2dc0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/src/main/java/com/tomfran/lsm/memtable/Memtable.java b/src/main/java/com/tomfran/lsm/memtable/Memtable.java
@@ -1,6 +1,7 @@
 package com.tomfran.lsm.memtable;
 
 import com.tomfran.lsm.types.ByteArrayPair;
+import com.tomfran.lsm.utils.UniqueSortedIterator;
 
 import java.util.Iterator;
 
@@ -34,7 +35,7 @@ public int size() {
 
     @Override
     public Iterator<ByteArrayPair> iterator() {
-        return list.iterator();
+        return new UniqueSortedIterator<>(list.iterator());
     }
 
 }
diff --git a/src/main/java/com/tomfran/lsm/sstable/SSTable.java b/src/main/java/com/tomfran/lsm/sstable/SSTable.java
@@ -12,6 +12,7 @@
 
 import java.io.File;
 import java.util.Iterator;
+import java.util.List;
 
 import static com.tomfran.lsm.comparator.ByteArrayComparator.compare;
 import static java.util.Arrays.stream;
@@ -257,9 +258,8 @@ private void writeItems(String filename, Iterator<ByteArrayPair> items, int samp
     }
 
     public void deleteFiles() {
-        new File(filename + DATA_FILE_EXTENSION).delete();
-        new File(filename + INDEX_FILE_EXTENSION).delete();
-        new File(filename + BLOOM_FILE_EXTENSION).delete();
+        for (var extension : List.of(DATA_FILE_EXTENSION, INDEX_FILE_EXTENSION, BLOOM_FILE_EXTENSION))
+            new File(filename + extension).delete();
     }
 
     private static class SSTableIterator implements Iterator<ByteArrayPair> {

diff --git a/src/test/java/com/tomfran/lsm/bloom/BloomFilterAccuracyTest.java b/src/test/java/com/tomfran/lsm/bloom/BloomFilterAccuracyTest.java
@@ -21,7 +21,7 @@ static void tearDown() {
                 "False positive rate is not close to the expected value: " + avg;
     }
 
-    @RepeatedTest(10)
+    @RepeatedTest(5)
     void testAdd() {
         b = new BloomFilter(INSERTIONS, FALSE_POSITIVE_RATE);