Adding compression codec class to CompressionCodecFactory's confugura…

…tion otherwise ccf.getCodecByClassName won't find the set codec.
pinterest · Feb 12, 2016 · 41566dc · 41566dc
1 parent d8d34f5
commit 41566dc
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);
     }
 }