Merge pull request #24 from civitaspo/v0.2.1

V0.2.1

CHENGELOG.md

@@ -1,3 +1,7 @@
-0.2.0 (2016-02-xx)
+0.2.1 (2016-02-25)
+==================
+- [Fix] does not work
+
+0.2.0 (2016-02-15)
 ==================
 - [Add] `decompression` option

build.gradle

@@ -15,7 +15,7 @@ configurations {
     provided
 }
 
-version = "0.2.0"
+version = "0.2.1"
 
 sourceCompatibility = 1.7
 targetCompatibility = 1.7

example/config.yml

@@ -12,11 +12,14 @@ local_fs_example: &local_fs_example
     fs.defaultFS: 'file:///'
     fs.hdfs.impl: 'org.apache.hadoop.fs.LocalFileSystem'
     fs.file.impl: 'org.apache.hadoop.fs.LocalFileSystem'
+    io.compression.codecs: 'org.apache.hadoop.io.compress.GzipCodec,org.apache.hadoop.io.compress.DefaultCodec,org.apache.hadoop.io.compress.BZip2Codec'
 
 in:
   type: hdfs
   <<: *local_fs_example
-  path: example/data.csv
+  path: example/data*
+  skip_header_lines: 1
+  decompression: true
   parser:
     charset: UTF-8
     newline: CRLF

src/main/java/org/embulk/input/hdfs/ConfigurationBuilder.java

@@ -0,0 +1,82 @@
+package org.embulk.input.hdfs;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
+import org.apache.hadoop.conf.Configuration;
+import org.embulk.config.ConfigException;
+import org.embulk.spi.Exec;
+import org.slf4j.Logger;
+
+import java.io.File;
+import java.net.MalformedURLException;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Created by takahiro.nakayama on 2/22/16.
+ */
+public class ConfigurationBuilder
+{
+    private static final Logger logger = Exec.getLogger(ConfigurationBuilder.class);
+    private final ImmutableList.Builder<String> configFilesBuilder;
+    private final ImmutableMap.Builder<String, String> configMapBuilder;
+
+    public ConfigurationBuilder()
+    {
+        this.configFilesBuilder = ImmutableList.builder();
+        this.configMapBuilder = ImmutableMap.builder();
+    }
+
+    public ConfigurationBuilder addConfigFiles(List<String> configFiles)
+    {
+        for (String configFile : configFiles) {
+            addConfigFile(configFile);
+        }
+        return this;
+    }
+
+    public ConfigurationBuilder addConfigFile(String configFile)
+    {
+        configFilesBuilder.add(configFile);
+        return this;
+    }
+
+    public ConfigurationBuilder addConfigMap(Map<String, String> configMap)
+    {
+        for (Map.Entry<String, String> entry : configMap.entrySet()) {
+            addConfig(entry.getKey(), entry.getValue());
+        }
+        return this;
+    }
+
+    public ConfigurationBuilder addConfig(String key, String value)
+    {
+        configMapBuilder.put(key, value);
+        return this;
+    }
+
+    public Configuration build()
+    {
+        Configuration configuration = new Configuration();
+        for (String configFile : configFilesBuilder.build()) {
+            File file = new File(configFile);
+            try {
+                configuration.addResource(file.toURI().toURL());
+            }
+            catch (MalformedURLException e) {
+                throw new ConfigException(e);
+            }
+        }
+        for (Map.Entry<String, String> entry : configMapBuilder.build().entrySet()) {
+            configuration.set(entry.getKey(), entry.getValue());
+        }
+        // For debug
+        for (Map.Entry<String, String> entry : configuration) {
+            logger.trace("{}: {}", entry.getKey(), entry.getValue());
+        }
+        logger.trace("Resource Files: {}", configuration);
+        return configuration;
+    }
+}