fix: decompress will failed when the value is null (#167)

XiaoMi · Sep 26, 2021 · 497f1b1 · 497f1b1
1 parent b31c506
commit 497f1b1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/main/java/com/xiaomi/infra/pegasus/tools/ZstdWrapper.java b/src/main/java/com/xiaomi/infra/pegasus/tools/ZstdWrapper.java
@@ -22,7 +22,7 @@ public static byte[] tryDecompress(byte[] src) {
     byte[] decompressedValue;
     try {
       decompressedValue = decompress(src);
-    } catch (PException e) {
+    } catch (Exception e) {
       // decompress fail
       decompressedValue = src;
     }

diff --git a/src/test/java/com/xiaomi/infra/pegasus/rpc/async/InterceptorTest.java b/src/test/java/com/xiaomi/infra/pegasus/rpc/async/InterceptorTest.java
@@ -53,5 +53,9 @@ public void testCompressionInterceptor() throws PException {
         new String(commonTable.get(hashKey, sortKey, 10000)), new String(compressionValue));
     Assertions.assertEquals(
         new String(compressTable.get(hashKey, sortKey, 10000)), new String(compressionValue));
+
+    // if origin value is null, return null
+    byte[] ret = compressTable.get("not-exist".getBytes(), sortKey, 10000);
+    Assertions.assertNull(ret);
   }
 }