ByteBufs are not released properly in SnappyCodec and LZ4Codec #1046

redisson/src/main/java/org/redisson/codec/LZ4Codec.java

@@ -56,6 +56,10 @@ public class LZ4Codec implements Codec {
         this.innerCodec = innerCodec;
     }
     
+    public LZ4Codec(ClassLoader classLoader) {
+        this(new FstCodec(classLoader));
+    }
+
     private final Decoder<Object> decoder = new Decoder<Object>() {
         @Override
         public Object decode(ByteBuf buf, State state) throws IOException {
@@ -79,8 +83,10 @@ public class LZ4Codec implements Codec {
 
         @Override
         public ByteBuf encode(Object in) throws IOException {
+            ByteBuf bytes = null;
+            try {
                 LZ4Compressor compressor = factory.fastCompressor();
-            ByteBuf bytes = innerCodec.getValueEncoder().encode(in);
+                bytes = innerCodec.getValueEncoder().encode(in);
                 ByteBuffer srcBuf = bytes.internalNioBuffer(bytes.readerIndex(), bytes.readableBytes());
                 
                 int outMaxLength = compressor.maxCompressedLength(bytes.readableBytes());
@@ -94,6 +100,11 @@ public class LZ4Codec implements Codec {
                 int compressedLength = outBuf.position() - pos;
                 out.writerIndex(out.writerIndex() + compressedLength);
                 return out;
+            } finally {
+                if (bytes != null) {
+                    bytes.release();
+                }
+            }
         }
     };
 

redisson/src/main/java/org/redisson/codec/SnappyCodec.java

@@ -60,6 +60,10 @@ public class SnappyCodec implements Codec {
         this.innerCodec = innerCodec;
     }
 
+    public SnappyCodec(ClassLoader classLoader) {
+        this(new FstCodec(classLoader));
+    }
+    
     private final Decoder<Object> decoder = new Decoder<Object>() {
         
         @Override
@@ -67,9 +71,9 @@ public class SnappyCodec implements Codec {
             ByteBuf out = ByteBufAllocator.DEFAULT.buffer();
             try {
                 snappyDecoder.get().decode(buf, out);
-                snappyDecoder.get().reset();
                 return innerCodec.getValueDecoder().decode(out, state);
             } finally {
+                snappyDecoder.get().reset();
                 out.release();
             }
         }
@@ -80,10 +84,14 @@ public class SnappyCodec implements Codec {
         @Override
         public ByteBuf encode(Object in) throws IOException {
             ByteBuf buf = innerCodec.getValueEncoder().encode(in);
-            ByteBuf out = ByteBufAllocator.DEFAULT.buffer(1024*100);
+            ByteBuf out = ByteBufAllocator.DEFAULT.buffer(buf.readableBytes() + 128);
+            try {
                 snappyEncoder.get().encode(buf, out, buf.readableBytes());
-            snappyEncoder.get().reset();
                 return out;
+            } finally {
+                buf.release();
+                snappyEncoder.get().reset();
+            }
         }
     };