HaloDB.contains method

yahoo · May 30, 2020 · aed0d27 · aed0d27
1 parent 741b4f7
commit aed0d27
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 1 deletion.
diff --git a/src/main/java/com/oath/halodb/HaloDB.java b/src/main/java/com/oath/halodb/HaloDB.java
@@ -56,6 +56,10 @@ public void delete(byte[] key) throws HaloDBException {
         }
     }
 
+    public boolean contains(byte[] key) {
+        return dbInternal.contains(key);
+    }
+
     public void close() throws HaloDBException {
         try {
             dbInternal.close();

diff --git a/src/main/java/com/oath/halodb/HaloDBInternal.java b/src/main/java/com/oath/halodb/HaloDBInternal.java
@@ -381,13 +381,17 @@ boolean clearSnapshot() {
                 return false;
             }
 
-            return  true;
+            return true;
         } else {
             logger.info("snapshot not existed");
             return true;
         }
     }
 
+    boolean contains(byte[] key) {
+        return inMemoryIndex.containsKey(key);
+    }
+
     void delete(byte[] key) throws IOException {
         writeLock.lock();
         try {

diff --git a/src/test/java/com/oath/halodb/HaloDBTest.java b/src/test/java/com/oath/halodb/HaloDBTest.java
@@ -76,6 +76,33 @@ public void testPutUpdateAndGetDB(HaloDBOptions options) throws HaloDBException
         });
     }
 
+    @Test(dataProvider = "Options")
+    public void testPutAndContainsDB(HaloDBOptions options) throws HaloDBException {
+        String directory = TestUtils.getTestDirectory("HaloDBTest", "testPutAndGetDB");
+
+        options.setCompactionDisabled(true);
+
+        HaloDB db = getTestDB(directory, options);
+
+        int noOfRecords = 10_000;
+        List<Record> records = TestUtils.insertRandomRecordsOfSize(db, noOfRecords, 128);
+
+        List<byte[]> missingKeys = new ArrayList<>();
+        for (int i = 0; i < noOfRecords; i++) {
+            missingKeys.add(TestUtils.generateRandomByteArray(64));
+        }
+
+        List<Record> actual = new ArrayList<>();
+        db.newIterator().forEachRemaining(actual::add);
+
+        Assert.assertTrue(actual.containsAll(records) && records.containsAll(actual));
+
+        records.forEach(record -> Assert.assertTrue(db.contains(record.getKey())));
+
+        missingKeys.forEach(key -> Assert.assertFalse(db.contains(key)));
+    }
+
+
     @Test(dataProvider = "Options")
     public void testCreateCloseAndOpenDB(HaloDBOptions options) throws HaloDBException {