minor fixes

fingltd · Oct 20, 2014 · bdef4fc · bdef4fc
1 parent d52b152
commit bdef4fc
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/FourMcInputStream.java b/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/FourMcInputStream.java
@@ -39,6 +39,7 @@
 import org.apache.hadoop.fs.FileSystem;
 import org.apache.hadoop.fs.Path;
 import org.apache.hadoop.io.compress.BlockDecompressorStream;
+import org.apache.hadoop.io.compress.CodecPool;
 import org.apache.hadoop.io.compress.Decompressor;
 
 import java.io.EOFException;
@@ -331,9 +332,10 @@ public long getCompressedBytesRead() {
     public void close() throws IOException {
         byte[] b = new byte[4096];
         while (!decompressor.finished()) {
-            decompressor.decompress(b, 0, b.length);
+          decompressor.decompress(b, 0, b.length);
         }
         super.close();
+        CodecPool.returnDecompressor(decompressor);
     }
 }
 
diff --git a/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/FourMcOutputStream.java b/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/FourMcOutputStream.java
@@ -36,6 +36,7 @@
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.apache.hadoop.io.DataOutputBuffer;
+import org.apache.hadoop.io.compress.CodecPool;
 import org.apache.hadoop.io.compress.Compressor;
 import org.apache.hadoop.io.compress.CompressorStream;
 
@@ -122,6 +123,8 @@ public void close() throws IOException {
 
         out.close();
         closed = true;
+
+        CodecPool.returnCompressor(compressor);
     }
 
     @Override

diff --git a/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/Lz4Compressor.java b/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/Lz4Compressor.java
@@ -44,7 +44,7 @@
 /**
  * LZ4 Compressor (FAST).
  */
-class Lz4Compressor implements Compressor {
+public class Lz4Compressor implements Compressor {
     private static final Log LOG = LogFactory.getLog(Lz4Compressor.class.getName());
 
     private int directBufferSize;

diff --git a/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/Lz4Decompressor.java b/java/hadoop-4mc/src/main/java/com/hadoop/compression/fourmc/Lz4Decompressor.java
@@ -44,7 +44,7 @@
 /**
  * LZ4 Decompressor, tuned for 4mc purposes.
  */
-class Lz4Decompressor implements Decompressor {
+public class Lz4Decompressor implements Decompressor {
 
     private static final Log LOG = LogFactory.getLog(Lz4Decompressor.class.getName());