diff --git a/data-prepper-plugins/common/src/main/java/org/opensearch/dataprepper/plugins/codec/CompressionOption.java b/data-prepper-plugins/common/src/main/java/org/opensearch/dataprepper/plugins/codec/CompressionOption.java index 14c481a19a..a9fe4a97a8 100644 --- a/data-prepper-plugins/common/src/main/java/org/opensearch/dataprepper/plugins/codec/CompressionOption.java +++ b/data-prepper-plugins/common/src/main/java/org/opensearch/dataprepper/plugins/codec/CompressionOption.java @@ -51,7 +51,9 @@ public static CompressionOption fromFileName(final String fileName) { return CompressionOption.GZIP; } else if(fileName.endsWith(".snappy")){ return CompressionOption.SNAPPY; - }else { + } else if (fileName.endsWith(".zst") || fileName.endsWith(".zstd")) { + return CompressionOption.ZSTD; + } else { return CompressionOption.NONE; } } diff --git a/data-prepper-plugins/common/src/test/java/org/opensearch/dataprepper/plugins/codec/CompressionOptionTest.java b/data-prepper-plugins/common/src/test/java/org/opensearch/dataprepper/plugins/codec/CompressionOptionTest.java index 772476d905..527a8486f8 100644 --- a/data-prepper-plugins/common/src/test/java/org/opensearch/dataprepper/plugins/codec/CompressionOptionTest.java +++ b/data-prepper-plugins/common/src/test/java/org/opensearch/dataprepper/plugins/codec/CompressionOptionTest.java @@ -33,6 +33,16 @@ void testFromFileName_snappy() { assertThat(CompressionOption.fromFileName("temp.snappy"), is(CompressionOption.SNAPPY)); } + @Test + void testFromFileName_zstd() { + assertThat(CompressionOption.fromFileName("temp.zst"), is(CompressionOption.ZSTD)); + } + + @Test + void testFromFileName_zstd_longExtension() { + assertThat(CompressionOption.fromFileName("temp.zstd"), is(CompressionOption.ZSTD)); + } + @Test void testFromFileName_default() { assertThat(CompressionOption.fromFileName("temp.txt"), is(CompressionOption.NONE));