feat: optimize zstd compress (#246)

Author: YongwuHe

arex-instrumentation-foundation/src/main/java/io/arex/foundation/util/CompressUtil.java

@@ -1,7 +1,8 @@
 package io.arex.foundation.util;
 
-import com.github.luben.zstd.ZstdInputStream;
-import com.github.luben.zstd.ZstdOutputStream;
+import com.github.luben.zstd.RecyclingBufferPool;
+import com.github.luben.zstd.ZstdInputStreamNoFinalizer;
+import com.github.luben.zstd.ZstdOutputStreamNoFinalizer;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -35,7 +36,8 @@ public static byte[] zstdCompress(byte[] original) {
 
         try (ByteArrayInputStream byteInputStream = new ByteArrayInputStream(original);
             ByteArrayOutputStream byteOutputStream = new ByteArrayOutputStream(byteInputStream.available());
-            ZstdOutputStreamNoFinalizer zstdOutputStream = new ZstdOutputStreamNoFinalizer(byteOutputStream)) {
+            ZstdOutputStreamNoFinalizer zstdOutputStream = new ZstdOutputStreamNoFinalizer(byteOutputStream,
+                    RecyclingBufferPool.INSTANCE)) {
 
             byte[] buffer = new byte[BYTES_BUFFER_LENGTH];
             for (int length; (length = byteInputStream.read(buffer, 0, BYTES_BUFFER_LENGTH)) != -1; ) {
@@ -52,7 +54,8 @@ public static byte[] zstdCompress(byte[] original) {
     }
 
     public static String zstdDecompress(InputStream inputStream, Charset charsetName) {
-        try (ZstdInputStreamNoFinalizer zstdInputStream = new ZstdInputStreamNoFinalizer(inputStream);
+        try (ZstdInputStreamNoFinalizer zstdInputStream = new ZstdInputStreamNoFinalizer(inputStream,
+                RecyclingBufferPool.INSTANCE);
             ByteArrayOutputStream byteOutputStream = new ByteArrayOutputStream(inputStream.available())) {
 
             byte[] buffer = new byte[BYTES_BUFFER_LENGTH];
@@ -70,22 +73,4 @@ public static String zstdDecompress(InputStream inputStream, Charset charsetName
     public static String zstdDecompress(byte[] bytes, Charset charsetName) {
         return zstdDecompress(new ByteArrayInputStream(bytes), charsetName);
     }
-
-    public static class ZstdInputStreamNoFinalizer extends ZstdInputStream {
-        ZstdInputStreamNoFinalizer(InputStream inputStream) throws IOException {
-            super(inputStream);
-        }
-
-        @Override
-        public void finalize() { }
-    }
-
-    public static class ZstdOutputStreamNoFinalizer extends ZstdOutputStream {
-        public ZstdOutputStreamNoFinalizer(OutputStream outStream) throws IOException {
-            super(outStream);
-        }
-
-        @Override
-        public void finalize() { }
-    }
 }

arex-instrumentation-foundation/src/test/java/io/arex/foundation/util/CompressUtilTest.java

@@ -2,8 +2,8 @@
 
 import static org.junit.jupiter.api.Assertions.assertArrayEquals;
 import static org.junit.jupiter.api.Assertions.assertEquals;
-
 import java.nio.charset.StandardCharsets;
+import java.util.Base64;
 import org.junit.jupiter.api.Test;
 
 class CompressUtilTest {
@@ -17,4 +17,16 @@ void test() {
         String expected = CompressUtil.zstdDecompress(actualBytes, StandardCharsets.UTF_8);
         assertEquals("hello AREX", expected);
     }
+
+    @Test
+    void testCompatible() {
+        String oldCompressString = "KLUv/QBYJAgAltE3JDCLOAcQDrWkTCPhBm9L2vigBeJYVw+ES3bvwu/pHjBwMDDAASsAMQAtAAzxS6NLBfbeBg8o7tWvx5hO8bSXSw2cXV/"
+                + "uI+wqUQ6KYaSdz5dz07JD7ZG7OKMoAIimCVASSbMsidMsCYNIlCVJWDQEzg4xIIaPMoNKJyl/VTpxsy+MjVN6UCG106Nxd1IQfDzC3lGnBm20Rpz91TiM4jiMg1i"
+                + "cxMKurzkWtk7l50th79W3kBsk1WNzrhXoa45eZZ6ew9OOFEIasErwq3q0kOAQbbpDQnLhv8aFUqycm7ajdgwARRmAwGN29RL2nXEYEDb8ug9tmqh5KPj4mTh9ehy1EAUBAAA=";
+        final byte[] oldCompressBytes = Base64.getDecoder().decode(oldCompressString);
+        final String newDeCompressString = CompressUtil.zstdDecompress(oldCompressBytes, StandardCharsets.UTF_8);
+        final byte[] newCompressBytes = CompressUtil.zstdCompress(newDeCompressString, StandardCharsets.UTF_8);
+        String newCompressString = Base64.getEncoder().encodeToString(newCompressBytes);
+        assertEquals(oldCompressString, newCompressString);
+    }
 }