Merge pull request #178 from prezi/master

Fix for custom compression codec
pinterest · Feb 12, 2016 · 77fab6a · 77fab6a
2 parents d8d34f5 + 41566dc
commit 77fab6a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/com/pinterest/secor/util/CompressionUtil.java b/src/main/java/com/pinterest/secor/util/CompressionUtil.java
@@ -16,6 +16,9 @@
  */
 package com.pinterest.secor.util;
 
+import java.util.Collections;
+import java.util.LinkedList;
+
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.io.compress.CompressionCodec;
 import org.apache.hadoop.io.compress.CompressionCodecFactory;
@@ -31,8 +34,10 @@ public class CompressionUtil {
 
     public static CompressionCodec createCompressionCodec(String className)
             throws Exception {
+        Configuration configuration = new Configuration();
+        CompressionCodecFactory.setCodecClasses(configuration,new LinkedList<Class>(Collections.singletonList(Class.forName(className))));
         CompressionCodecFactory ccf = new CompressionCodecFactory(
-                new Configuration());
+                configuration);
         return ccf.getCodecByClassName(className);
     }
 }