diff --git a/java/bench/core/src/java/org/apache/orc/bench/core/CompressionKind.java b/java/bench/core/src/java/org/apache/orc/bench/core/CompressionKind.java index 6eda6ad26c..b9dc36f1fe 100644 --- a/java/bench/core/src/java/org/apache/orc/bench/core/CompressionKind.java +++ b/java/bench/core/src/java/org/apache/orc/bench/core/CompressionKind.java @@ -18,6 +18,7 @@ package org.apache.orc.bench.core; +import io.airlift.compress.lz4.Lz4Codec; import io.airlift.compress.snappy.SnappyCodec; import org.apache.hadoop.fs.Path; @@ -34,6 +35,7 @@ public enum CompressionKind { NONE("none"), ZLIB("gz"), SNAPPY("snappy"), + LZ4("lz4"), ZSTD("zstd"); CompressionKind(String extension) { @@ -54,6 +56,8 @@ public OutputStream create(OutputStream out) throws IOException { return new GZIPOutputStream(out); case SNAPPY: return new SnappyCodec().createOutputStream(out); + case LZ4: + return new Lz4Codec().createOutputStream(out); default: throw new IllegalArgumentException("Unhandled kind " + this); } @@ -67,6 +71,8 @@ public InputStream read(InputStream in) throws IOException { return new GZIPInputStream(in); case SNAPPY: return new SnappyCodec().createInputStream(in); + case LZ4: + return new Lz4Codec().createInputStream(in); default: throw new IllegalArgumentException("Unhandled kind " + this); } diff --git a/java/bench/core/src/java/org/apache/orc/bench/core/Utilities.java b/java/bench/core/src/java/org/apache/orc/bench/core/Utilities.java index 1d25746d4f..2ad51472bb 100644 --- a/java/bench/core/src/java/org/apache/orc/bench/core/Utilities.java +++ b/java/bench/core/src/java/org/apache/orc/bench/core/Utilities.java @@ -58,6 +58,8 @@ public static org.apache.orc.CompressionKind getCodec(CompressionKind compressio return org.apache.orc.CompressionKind.ZLIB; case SNAPPY: return org.apache.orc.CompressionKind.SNAPPY; + case LZ4: + return org.apache.orc.CompressionKind.LZ4; case ZSTD: return org.apache.orc.CompressionKind.ZSTD; default: