Add factory to create a (de)compressor based on native availability

airlift · Jul 19, 2024 · a0491d4 · a0491d4
1 parent f1f65a8
commit a0491d4
Show file tree

Hide file tree

Showing 6 changed files with 48 additions and 0 deletions.
diff --git a/src/main/java/io/airlift/compress/v2/lz4/Lz4Compressor.java b/src/main/java/io/airlift/compress/v2/lz4/Lz4Compressor.java
@@ -22,4 +22,12 @@ public sealed interface Lz4Compressor
         permits Lz4JavaCompressor, Lz4NativeCompressor
 {
     int compress(MemorySegment input, MemorySegment output);
+
+    static Lz4Compressor create()
+    {
+        if (Lz4NativeCompressor.isEnabled()) {
+            return new Lz4NativeCompressor();
+        }
+        return new Lz4JavaCompressor();
+    }
 }
diff --git a/src/main/java/io/airlift/compress/v2/lz4/Lz4Decompressor.java b/src/main/java/io/airlift/compress/v2/lz4/Lz4Decompressor.java
@@ -22,4 +22,12 @@ public sealed interface Lz4Decompressor
         permits Lz4JavaDecompressor, Lz4NativeDecompressor
 {
     int decompress(MemorySegment input, MemorySegment output);
+
+    static Lz4Decompressor create()
+    {
+        if (Lz4NativeDecompressor.isEnabled()) {
+            return new Lz4NativeDecompressor();
+        }
+        return new Lz4JavaDecompressor();
+    }
 }
diff --git a/src/main/java/io/airlift/compress/v2/snappy/SnappyCompressor.java b/src/main/java/io/airlift/compress/v2/snappy/SnappyCompressor.java
@@ -22,4 +22,12 @@ public sealed interface SnappyCompressor
         permits SnappyJavaCompressor, SnappyNativeCompressor
 {
     int compress(MemorySegment input, MemorySegment output);
+
+    static SnappyCompressor create()
+    {
+        if (SnappyNativeCompressor.isEnabled()) {
+            return new SnappyNativeCompressor();
+        }
+        return new SnappyJavaCompressor();
+    }
 }
diff --git a/src/main/java/io/airlift/compress/v2/snappy/SnappyDecompressor.java b/src/main/java/io/airlift/compress/v2/snappy/SnappyDecompressor.java
@@ -20,4 +20,12 @@ public sealed interface SnappyDecompressor
         permits SnappyJavaDecompressor, SnappyNativeDecompressor
 {
     int getUncompressedLength(byte[] compressed, int compressedOffset);
+
+    static SnappyDecompressor create()
+    {
+        if (SnappyNativeDecompressor.isEnabled()) {
+            return new SnappyNativeDecompressor();
+        }
+        return new SnappyJavaDecompressor();
+    }
 }
diff --git a/src/main/java/io/airlift/compress/v2/zstd/ZstdCompressor.java b/src/main/java/io/airlift/compress/v2/zstd/ZstdCompressor.java
@@ -21,4 +21,12 @@ public interface ZstdCompressor
         extends Compressor
 {
     int compress(MemorySegment input, MemorySegment output);
+
+    static ZstdCompressor create()
+    {
+        if (ZstdNativeCompressor.isEnabled()) {
+            return new ZstdNativeCompressor();
+        }
+        return new ZstdJavaCompressor();
+    }
 }
diff --git a/src/main/java/io/airlift/compress/v2/zstd/ZstdDecompressor.java b/src/main/java/io/airlift/compress/v2/zstd/ZstdDecompressor.java
@@ -19,4 +19,12 @@ public interface ZstdDecompressor
         extends Decompressor
 {
     long getDecompressedSize(byte[] input, int offset, int length);
+
+    static ZstdDecompressor create()
+    {
+        if (ZstdNativeDecompressor.isEnabled()) {
+            return new ZstdNativeDecompressor();
+        }
+        return new ZstdJavaDecompressor();
+    }
 }