embulk-output-orc 0.3.0 → 0.3.5

data/.travis.yml

@@ -1,14 +0,0 @@
-language: java
-jdk:
-  - oraclejdk8
-
-cache:
-  directories:  # run "travis cache --delete" to delete caches
-    - $HOME/.gradle
-
-sudo: false
-script:
-  - ./gradlew --info checkstyle
-  - ./gradlew --info check
-
-after_success: ./gradlew sonarqube -Dsonar.organization=yuokada-github -Dsonar.host.url=https://sonarcloud.io -Dsonar.login=${SONAR_TOKEN}

data/src/main/java/org/embulk/output/orc/OrcColumnVisitor.java

@@ -1,82 +0,0 @@
-package org.embulk.output.orc;
-
-import org.apache.hadoop.hive.ql.exec.vector.BytesColumnVector;
-import org.apache.hadoop.hive.ql.exec.vector.DoubleColumnVector;
-import org.apache.hadoop.hive.ql.exec.vector.LongColumnVector;
-import org.apache.hadoop.hive.ql.exec.vector.TimestampColumnVector;
-import org.apache.hadoop.hive.ql.exec.vector.VectorizedRowBatch;
-import org.embulk.spi.Column;
-import org.embulk.spi.ColumnVisitor;
-import org.embulk.spi.PageReader;
-import org.embulk.spi.time.Timestamp;
-
-import java.nio.charset.StandardCharsets;
-
-public class OrcColumnVisitor
-        implements ColumnVisitor
-{
-    private final PageReader reader;
-    private final VectorizedRowBatch batch;
-    private final Integer i;
-
-    public OrcColumnVisitor(PageReader pageReader, VectorizedRowBatch rowBatch, Integer i)
-    {
-        this.reader = pageReader;
-        this.batch = rowBatch;
-        this.i = i;
-    }
-
-    @Override
-    public void booleanColumn(Column column)
-    {
-        if (reader.isNull(column)) {
-            ((LongColumnVector) batch.cols[column.getIndex()]).vector[i] = 0;
-        }
-        else {
-            if (reader.getBoolean(column)) {
-                ((LongColumnVector) batch.cols[column.getIndex()]).vector[i] = 1;
-            }
-            else {
-                ((LongColumnVector) batch.cols[column.getIndex()]).vector[i] = 0;
-            }
-        }
-    }
-
-    @Override
-    public void longColumn(Column column)
-    {
-        ((LongColumnVector) batch.cols[column.getIndex()]).vector[i] = reader.getLong(column);
-    }
-
-    @Override
-    public void doubleColumn(Column column)
-    {
-        ((DoubleColumnVector) batch.cols[column.getIndex()]).vector[i] = reader.getDouble(column);
-    }
-
-    @Override
-    public void stringColumn(Column column)
-    {
-        ((BytesColumnVector) batch.cols[column.getIndex()]).setVal(i,
-                reader.getString(column).getBytes(StandardCharsets.UTF_8));
-    }
-
-    @Override
-    public void timestampColumn(Column column)
-    {
-        if (reader.isNull(column)) {
-            ((TimestampColumnVector) batch.cols[column.getIndex()]).setNullValue(i);
-        }
-        else {
-            Timestamp timestamp = reader.getTimestamp(column);
-            java.sql.Timestamp ts = new java.sql.Timestamp(timestamp.getEpochSecond() * 1000);
-            ((TimestampColumnVector) batch.cols[column.getIndex()]).set(i, ts);
-        }
-    }
-
-    @Override
-    public void jsonColumn(Column column)
-    {
-        throw new UnsupportedOperationException("orc output plugin does not support json type");
-    }
-}

data/src/main/java/org/embulk/output/orc/OrcOutputPlugin.java

@@ -1,249 +0,0 @@
-package org.embulk.output.orc;
-
-import com.google.common.base.Throwables;
-import org.apache.hadoop.conf.Configuration;
-import org.apache.hadoop.fs.LocalFileSystem;
-import org.apache.hadoop.fs.Path;
-import org.apache.hadoop.hdfs.DistributedFileSystem;
-import org.apache.hadoop.hive.ql.exec.vector.VectorizedRowBatch;
-import org.apache.hadoop.util.VersionInfo;
-import org.apache.orc.CompressionKind;
-import org.apache.orc.OrcFile;
-import org.apache.orc.TypeDescription;
-import org.apache.orc.Writer;
-import org.embulk.config.ConfigDiff;
-import org.embulk.config.ConfigSource;
-import org.embulk.config.TaskReport;
-import org.embulk.config.TaskSource;
-import org.embulk.spi.Column;
-import org.embulk.spi.Exec;
-import org.embulk.spi.OutputPlugin;
-import org.embulk.spi.Page;
-import org.embulk.spi.PageReader;
-import org.embulk.spi.Schema;
-import org.embulk.spi.TransactionalPageOutput;
-import org.embulk.spi.time.TimestampFormatter;
-import org.embulk.spi.type.Type;
-import org.embulk.spi.util.Timestamps;
-import org.embulk.util.aws.credentials.AwsCredentials;
-
-import java.io.IOException;
-import java.util.List;
-
-public class OrcOutputPlugin
-        implements OutputPlugin
-{
-    @Override
-    public ConfigDiff transaction(ConfigSource config,
-            Schema schema, int taskCount,
-            OutputPlugin.Control control)
-    {
-        PluginTask task = config.loadConfig(PluginTask.class);
-
-        // retryable (idempotent) output:
-        // return resume(task.dump(), schema, taskCount, control);
-
-        // non-retryable (non-idempotent) output:
-        control.run(task.dump());
-        return Exec.newConfigDiff();
-    }
-
-    @Override
-    public ConfigDiff resume(TaskSource taskSource,
-            Schema schema, int taskCount,
-            OutputPlugin.Control control)
-    {
-        throw new UnsupportedOperationException("orc output plugin does not support resuming");
-    }
-
-    @Override
-    public void cleanup(TaskSource taskSource,
-            Schema schema, int taskCount,
-            List<TaskReport> successTaskReports)
-
-    {
-    }
-
-    @Override
-    public TransactionalPageOutput open(TaskSource taskSource, Schema schema, int taskIndex)
-    {
-        PluginTask task = taskSource.loadTask(PluginTask.class);
-
-        if (task.getOverwrite()) {
-            OrcOutputPluginHelper.removeOldFile(buildPath(task, taskIndex));
-        }
-
-        final PageReader reader = new PageReader(schema);
-        Writer writer = createWriter(task, schema, taskIndex);
-
-        return new OrcTransactionalPageOutput(reader, writer, task);
-    }
-
-    private String buildPath(PluginTask task, int processorIndex)
-    {
-        final String pathPrefix = task.getPathPrefix();
-        final String pathSuffix = task.getFileNameExtension();
-        final String sequenceFormat = task.getSequenceFormat();
-        return pathPrefix + String.format(sequenceFormat, processorIndex) + pathSuffix;
-    }
-
-    private TypeDescription getSchema(Schema schema)
-    {
-        TypeDescription oschema = TypeDescription.createStruct();
-        for (int i = 0; i < schema.size(); i++) {
-            Column column = schema.getColumn(i);
-            Type type = column.getType();
-            switch (type.getName()) {
-                case "long":
-                    oschema.addField(column.getName(), TypeDescription.createLong());
-                    break;
-                case "double":
-                    oschema.addField(column.getName(), TypeDescription.createDouble());
-                    break;
-                case "boolean":
-                    oschema.addField(column.getName(), TypeDescription.createBoolean());
-                    break;
-                case "string":
-                    oschema.addField(column.getName(), TypeDescription.createString());
-                    break;
-                case "timestamp":
-                    oschema.addField(column.getName(), TypeDescription.createTimestamp());
-                    break;
-                default:
-                    System.out.println("Unsupported type");
-                    break;
-            }
-        }
-        return oschema;
-    }
-
-    private Configuration getHadoopConfiguration(PluginTask task)
-    {
-        Configuration conf = new Configuration();
-
-        // see: https://stackoverflow.com/questions/17265002/hadoop-no-filesystem-for-scheme-file
-        conf.set("fs.hdfs.impl", DistributedFileSystem.class.getName());
-        conf.set("fs.file.impl", LocalFileSystem.class.getName());
-        // see: https://stackoverflow.com/questions/20833444/how-to-set-objects-in-hadoop-configuration
-
-        AwsCredentials.getAWSCredentialsProvider(task);
-        if (task.getAccessKeyId().isPresent()) {
-            conf.set("fs.s3a.access.key", task.getAccessKeyId().get());
-            conf.set("fs.s3n.awsAccessKeyId", task.getAccessKeyId().get());
-        }
-        if (task.getSecretAccessKey().isPresent()) {
-            conf.set("fs.s3a.secret.key", task.getSecretAccessKey().get());
-            conf.set("fs.s3n.awsSecretAccessKey", task.getSecretAccessKey().get());
-        }
-        if (task.getEndpoint().isPresent()) {
-            conf.set("fs.s3a.endpoint", task.getEndpoint().get());
-        }
-        return conf;
-    }
-
-    private Writer createWriter(PluginTask task, Schema schema, int processorIndex)
-    {
-        final TimestampFormatter[] timestampFormatters = Timestamps
-                .newTimestampColumnFormatters(task, schema, task.getColumnOptions());
-
-        Configuration conf = getHadoopConfiguration(task);
-        TypeDescription oschema = getSchema(schema);
-
-        // see: https://groups.google.com/forum/#!topic/vertx/lLb-slzpWVg
-        Thread.currentThread().setContextClassLoader(VersionInfo.class.getClassLoader());
-
-        Writer writer = null;
-        try {
-            // Make writerOptions
-            OrcFile.WriterOptions writerOptions = createWriterOptions(task, conf);
-            // see: https://stackoverflow.com/questions/9256733/how-to-connect-hive-in-ireport
-            // see: https://community.hortonworks.com/content/kbentry/73458/connecting-dbvisualizer-and-datagrip-to-hive-with.html
-            writer = OrcFile.createWriter(new Path(buildPath(task, processorIndex)),
-                    writerOptions.setSchema(oschema)
-                            .version(OrcFile.Version.V_0_12));
-        }
-        catch (IOException e) {
-            Throwables.propagate(e);
-        }
-        return writer;
-    }
-
-    private OrcFile.WriterOptions createWriterOptions(PluginTask task, Configuration conf)
-    {
-        final Integer bufferSize = task.getBufferSize();
-        final Integer stripSize = task.getStripSize();
-        final Integer blockSize = task.getBlockSize();
-        final String kindString = task.getCompressionKind();
-        CompressionKind kind = CompressionKind.valueOf(kindString);
-        return OrcFile.writerOptions(conf)
-                .bufferSize(bufferSize)
-                .blockSize(blockSize)
-                .stripeSize(stripSize)
-                .compress(kind);
-    }
-
-    class OrcTransactionalPageOutput
-            implements TransactionalPageOutput
-    {
-        private final PageReader reader;
-        private final Writer writer;
-
-        public OrcTransactionalPageOutput(PageReader reader, Writer writer, PluginTask task)
-        {
-            this.reader = reader;
-            this.writer = writer;
-        }
-
-        @Override
-        public void add(Page page)
-        {
-            int size = page.getStringReferences().size();
-            final TypeDescription schema = getSchema(reader.getSchema());
-            final VectorizedRowBatch batch = schema.createRowBatch();
-            batch.size = size;
-
-            reader.setPage(page);
-            int i = 0;
-            while (reader.nextRecord()) {
-                reader.getSchema().visitColumns(
-                        new OrcColumnVisitor(reader, batch, i)
-                );
-                i++;
-            }
-            try {
-                writer.addRowBatch(batch);
-                batch.reset();
-            }
-            catch (IOException e) {
-                e.printStackTrace();
-            }
-        }
-
-        @Override
-        public void finish()
-        {
-            try {
-                writer.close();
-            }
-            catch (IOException e) {
-                Throwables.propagate(e);
-            }
-        }
-
-        @Override
-        public void close()
-        {
-        }
-
-        @Override
-        public void abort()
-        {
-        }
-
-        @Override
-        public TaskReport commit()
-        {
-            return Exec.newTaskReport();
-        }
-    }
-}

data/src/main/java/org/embulk/output/orc/OrcOutputPluginHelper.java

@@ -1,28 +0,0 @@
-package org.embulk.output.orc;
-
-import com.google.common.base.Throwables;
-
-import java.io.IOException;
-import java.nio.file.Files;
-import java.nio.file.Path;
-import java.nio.file.Paths;
-
-class OrcOutputPluginHelper
-{
-    protected OrcOutputPluginHelper()
-    {
-        throw new UnsupportedOperationException();
-    }
-
-    static void removeOldFile(String fpath)
-    {
-        Path path = Paths.get(fpath);
-        // TODO: Check local file. not HDFS or S3.
-        try {
-            Files.deleteIfExists(path);
-        }
-        catch (IOException e) {
-            Throwables.propagate(e);
-        }
-    }
-}

data/src/main/java/org/embulk/output/orc/PluginTask.java

@@ -1,60 +0,0 @@
-package org.embulk.output.orc;
-
-import com.google.common.base.Optional;
-import org.embulk.config.Config;
-import org.embulk.config.ConfigDefault;
-import org.embulk.config.Task;
-import org.embulk.spi.time.TimestampFormatter;
-import org.embulk.util.aws.credentials.AwsCredentialsTask;
-import org.joda.time.DateTimeZone;
-
-import java.util.Map;
-
-public interface PluginTask
-        extends Task, TimestampFormatter.Task, AwsCredentialsTask
-{
-    @Config("path_prefix")
-    String getPathPrefix();
-
-    @Config("file_ext")
-    @ConfigDefault("\".orc\"")
-    String getFileNameExtension();
-
-    @Config("column_options")
-    @ConfigDefault("{}")
-    Map<String, TimestampColumnOption> getColumnOptions();
-
-    @Config("sequence_format")
-    @ConfigDefault("\".%03d\"")
-    String getSequenceFormat();
-
-    // see: https://orc.apache.org/docs/hive-config.html
-    // ORC File options
-    @Config("strip_size")
-    @ConfigDefault("67108864") // 64MB
-    Integer getStripSize();
-
-    @Config("buffer_size")
-    @ConfigDefault("262144") // 256KB
-    Integer getBufferSize();
-
-    @Config("block_size")
-    @ConfigDefault("268435456") // 256MB
-    Integer getBlockSize();
-
-    @Config("compression_kind")
-    @ConfigDefault("ZLIB")
-    public String getCompressionKind();
-
-    @Config("overwrite")
-    @ConfigDefault("false")
-    boolean getOverwrite();
-
-    @Config("default_from_timezone")
-    @ConfigDefault("\"UTC\"")
-    DateTimeZone getDefaultFromTimeZone();
-
-    @Config("endpoint")
-    @ConfigDefault("null")
-    Optional<String> getEndpoint();
-}

data/src/main/java/org/embulk/output/orc/TimestampColumnOption.java

@@ -1,22 +0,0 @@
-package org.embulk.output.orc;
-
-import com.google.common.base.Optional;
-import org.embulk.config.Config;
-import org.embulk.config.ConfigDefault;
-import org.embulk.config.Task;
-import org.embulk.spi.time.TimestampFormatter;
-import org.joda.time.DateTimeZone;
-
-import java.util.List;
-
-public interface TimestampColumnOption
-        extends Task, TimestampFormatter.TimestampColumnOption
-{
-    @Config("from_timezone")
-    @ConfigDefault("null")
-    Optional<DateTimeZone> getFromTimeZone();
-
-    @Config("from_format")
-    @ConfigDefault("null")
-    Optional<List<String>> getFromFormat();
-}