accessors for custom compression modes

Signed-off-by: Sarthak Aggarwal <sarthagg@amazon.com>
opensearch-project · Nov 30, 2023 · 07d42d8 · 07d42d8
1 parent 82a4fea
commit 07d42d8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormat.java b/src/main/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormat.java
@@ -132,4 +132,13 @@ Lucene95CustomCodec.Mode getMode() {
     public int getCompressionLevel() {
         return compressionLevel;
     }
+
+
+    public CompressionMode getZstdCompressionMode() {
+        return zstdCompressionMode;
+    }
+
+    public CompressionMode getZstdNoDictCompressionMode() {
+        return zstdNoDictCompressionMode;
+    }
 }
diff --git a/...t/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormatTests.java b/...t/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormatTests.java
@@ -44,4 +44,9 @@ public void testZstdNoDictLucene95CustomCodecModeWithCompressionLevel() {
         assertEquals(randomCompressionLevel, lucene95CustomStoredFieldsFormat.getCompressionLevel());
     }
 
+    public void testCompressionModes(){
+        Lucene95CustomStoredFieldsFormat lucene95CustomStoredFieldsFormat = new Lucene95CustomStoredFieldsFormat();
+        assertTrue(lucene95CustomStoredFieldsFormat.getZstdCompressionMode() instanceof ZstdCompressionMode);
+        assertTrue(lucene95CustomStoredFieldsFormat.getZstdNoDictCompressionMode() instanceof ZstdNoDictCompressionMode);
+    }
 }