embulk-filter-add_time 0.1.0

data/src/main/java/org/embulk/filter/add_time/converter/SimpleColumnConverter.java

@@ -0,0 +1,62 @@
+package org.embulk.filter.add_time.converter;
+
+import org.embulk.filter.add_time.reader.ColumnReader;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageReader;
+import org.embulk.spi.Schema;
+
+public class SimpleColumnConverter
+        implements ColumnConverter
+{
+    public static class Builder
+    {
+        private Column column;
+        private ColumnReader reader;
+
+        public Builder()
+        {
+        }
+
+        public Builder setColumn(Column column)
+        {
+            this.column = column;
+            return this;
+        }
+
+        public Builder setColumnReader(ColumnReader reader)
+        {
+            this.reader = reader;
+            return this;
+        }
+
+        public ColumnConverter build()
+        {
+            return new SimpleColumnConverter(column, reader);
+        }
+    }
+
+    private final Column column;
+    private final ColumnReader columnReader;
+
+    private SimpleColumnConverter(Column column, ColumnReader columnReader)
+    {
+        this.column = column;
+        this.columnReader = columnReader;
+    }
+
+    public void update(PageReader pageReader)
+    {
+        columnReader.readValue(column, pageReader);
+    }
+
+    public void convert(PageBuilder pageBuilder)
+    {
+        columnReader.convertValue(column, pageBuilder);
+    }
+
+    public void addColumn(Schema.Builder schemaBuilder)
+    {
+        schemaBuilder.add(column.getName(), column.getType());
+    }
+}

data/src/main/java/org/embulk/filter/add_time/converter/StringValueCastConverter.java

@@ -0,0 +1,33 @@
+package org.embulk.filter.add_time.converter;
+
+import org.embulk.filter.add_time.AddTimeFilterPlugin.FromColumnConfig;
+import org.embulk.filter.add_time.AddTimeFilterPlugin.ToColumnConfig;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.spi.time.TimestampParser;
+
+public class StringValueCastConverter
+        extends ValueCastConverter
+{
+    private final TimestampParser fromTimestampParser;
+
+    public StringValueCastConverter(FromColumnConfig fromColumnConfig, ToColumnConfig toColumnConfig)
+    {
+        super(toColumnConfig);
+        this.fromTimestampParser = new TimestampParser(fromColumnConfig, fromColumnConfig);
+    }
+
+    @Override
+    public void convertValue(final Column column, String value, final PageBuilder pageBuilder)
+    {
+        columnVisitor.setValue(stringToTimestamp(value));
+        columnVisitor.setPageBuilder(pageBuilder);
+        column.visit(columnVisitor);
+    }
+
+    private Timestamp stringToTimestamp(String value)
+    {
+        return fromTimestampParser.parse(value);
+    }
+}

data/src/main/java/org/embulk/filter/add_time/converter/TimestampValueCastConverter.java

@@ -0,0 +1,23 @@
+package org.embulk.filter.add_time.converter;
+
+import org.embulk.filter.add_time.AddTimeFilterPlugin.ToColumnConfig;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.time.Timestamp;
+
+public class TimestampValueCastConverter
+        extends ValueCastConverter
+{
+    public TimestampValueCastConverter(ToColumnConfig toColumnConfig)
+    {
+        super(toColumnConfig);
+    }
+
+    @Override
+    public void convertValue(Column column, final Timestamp value, final PageBuilder pageBuilder)
+    {
+        columnVisitor.setValue(value); // it can use the value directly.
+        columnVisitor.setPageBuilder(pageBuilder);
+        column.visit(columnVisitor);
+    }
+}

data/src/main/java/org/embulk/filter/add_time/converter/ValueCastConverter.java

@@ -0,0 +1,108 @@
+package org.embulk.filter.add_time.converter;
+
+import org.embulk.filter.add_time.AddTimeFilterPlugin.ToColumnConfig;
+import org.embulk.filter.add_time.AddTimeFilterPlugin.UnixTimestampUnit;
+import org.embulk.spi.Column;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.time.Timestamp;
+
+public abstract class ValueCastConverter
+        implements ValueConverter
+{
+    protected final TimestampValueCastVisitor columnVisitor;
+
+    public ValueCastConverter(ToColumnConfig toColumnConfig)
+    {
+        this.columnVisitor = new TimestampValueCastVisitor(UnixTimestampUnit.of(toColumnConfig.getUnixTimestampUnit()));
+    }
+
+    @Override
+    public void convertNull(Column column, PageBuilder pageBuilder)
+    {
+        pageBuilder.setNull(column);
+    }
+
+    @Override
+    public void convertValue(Column column, boolean value, PageBuilder pageBuilder)
+    {
+        throw new AssertionError("Never call.");
+    }
+
+    @Override
+    public void convertValue(Column column, long value, PageBuilder pageBuilder)
+    {
+        throw new AssertionError("Should implement in subclass.");
+    }
+
+    @Override
+    public void convertValue(Column column, double value, PageBuilder pageBuilder)
+    {
+        throw new AssertionError("Never call.");
+    }
+
+    @Override
+    public void convertValue(final Column column, String value, final PageBuilder pageBuilder)
+    {
+        throw new AssertionError("Should implement in subclass.");
+    }
+
+    @Override
+    public void convertValue(Column column, final Timestamp value, final PageBuilder pageBuilder)
+    {
+        throw new AssertionError("Never call.");
+    }
+
+    static class TimestampValueCastVisitor
+            implements ColumnVisitor
+    {
+        private final UnixTimestampUnit toUnixTimestampUnit;
+        private PageBuilder currentPageBuilder;
+        private Timestamp currentValue;
+
+        TimestampValueCastVisitor(UnixTimestampUnit toUnixTimestampUnit)
+        {
+            this.toUnixTimestampUnit = toUnixTimestampUnit;
+        }
+
+        void setValue(Timestamp value)
+        {
+            this.currentValue = value;
+        }
+
+        void setPageBuilder(PageBuilder pageBuilder)
+        {
+            this.currentPageBuilder = pageBuilder;
+        }
+
+        @Override
+        public void booleanColumn(Column column)
+        {
+            throw new AssertionError("Never call.");
+        }
+
+        @Override
+        public void longColumn(Column column)
+        {
+            currentPageBuilder.setLong(column, toUnixTimestampUnit.toLong(currentValue));
+        }
+
+        @Override
+        public void doubleColumn(Column column)
+        {
+            throw new AssertionError("Never call.");
+        }
+
+        @Override
+        public void stringColumn(Column column)
+        {
+            throw new AssertionError("Never call.");
+        }
+
+        @Override
+        public void timestampColumn(Column column)
+        {
+            currentPageBuilder.setTimestamp(column, currentValue);
+        }
+    }
+}

data/src/main/java/org/embulk/filter/add_time/converter/ValueConverter.java

@@ -0,0 +1,22 @@
+package org.embulk.filter.add_time.converter;
+
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.time.Timestamp;
+
+public interface ValueConverter
+{
+    ValueConverter NO_CONV = new ValueNoConverter();
+
+    void convertNull(Column column, PageBuilder pageBuilder);
+
+    void convertValue(Column column, boolean value, PageBuilder pageBuilder);
+
+    void convertValue(Column column, long value, PageBuilder pageBuilder);
+
+    void convertValue(Column column, double value, PageBuilder pageBuilder);
+
+    void convertValue(Column column, String value, PageBuilder pageBuilder);
+
+    void convertValue(Column column, Timestamp value, PageBuilder pageBuilder);
+}

data/src/main/java/org/embulk/filter/add_time/converter/ValueNoConverter.java

@@ -0,0 +1,46 @@
+package org.embulk.filter.add_time.converter;
+
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.time.Timestamp;
+
+public class ValueNoConverter
+        implements ValueConverter
+{
+    @Override
+    public void convertNull(Column column, PageBuilder pageBuilder)
+    {
+        pageBuilder.setNull(column);
+    }
+
+    @Override
+    public void convertValue(Column column, boolean value, PageBuilder pageBuilder)
+    {
+        pageBuilder.setBoolean(column, value);
+    }
+
+    @Override
+    public void convertValue(Column column, long value, PageBuilder pageBuilder)
+    {
+        pageBuilder.setLong(column, value);
+    }
+
+    @Override
+    public void convertValue(Column column, double value, PageBuilder pageBuilder)
+    {
+        pageBuilder.setDouble(column, value);
+    }
+
+    @Override
+    public void convertValue(Column column, String value, PageBuilder pageBuilder)
+    {
+        pageBuilder.setString(column, value);
+    }
+
+    @Override
+    public void convertValue(Column column, Timestamp value, PageBuilder pageBuilder)
+    {
+        pageBuilder.setTimestamp(column, value);
+
+    }
+}

data/src/main/java/org/embulk/filter/add_time/reader/AbstractColumnReader.java

@@ -0,0 +1,55 @@
+package org.embulk.filter.add_time.reader;
+
+import org.embulk.filter.add_time.converter.ValueConverter;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageReader;
+
+public abstract class AbstractColumnReader<T extends AbstractColumnReader>
+        implements ColumnReader<T>
+{
+    protected final ValueConverter valueConverter;
+    protected boolean isNull = false;
+
+    protected AbstractColumnReader(ValueConverter valueConverter)
+    {
+        this.valueConverter = valueConverter;
+    }
+
+    @Override
+    public void readValue(Column column, PageReader pageReader)
+    {
+        if (!(isNull = pageReader.isNull(column))) {
+            readNonNullValue(column, pageReader);
+        }
+    }
+
+    protected abstract void readNonNullValue(Column column, PageReader pageReader);
+
+    @Override
+    public void convertValue(Column column, PageBuilder pageBuilder)
+    {
+        try {
+            if (isNull) {
+                valueConverter.convertNull(column, pageBuilder);
+            }
+            else {
+                convertNonNullValue(column, pageBuilder);
+            }
+        }
+        finally {
+            isNull = false;
+        }
+    }
+
+    protected abstract void convertNonNullValue(Column column, PageBuilder pageBuilder);
+
+    @Override
+    public void copyTo(T columnReader)
+    {
+        columnReader.isNull = this.isNull;
+        copyValueTo(columnReader);
+    }
+
+    protected abstract void copyValueTo(T columnReader);
+}

data/src/main/java/org/embulk/filter/add_time/reader/BooleanColumnReader.java

@@ -0,0 +1,35 @@
+package org.embulk.filter.add_time.reader;
+
+import org.embulk.filter.add_time.converter.ValueConverter;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageReader;
+
+public class BooleanColumnReader
+        extends AbstractColumnReader<BooleanColumnReader>
+{
+    protected boolean value;
+
+    public BooleanColumnReader(ValueConverter valueConverter)
+    {
+        super(valueConverter);
+    }
+
+    @Override
+    public void readNonNullValue(Column column, PageReader pageReader)
+    {
+        value = pageReader.getBoolean(column);
+    }
+
+    @Override
+    public void convertNonNullValue(Column column, PageBuilder pageBuilder)
+    {
+        valueConverter.convertValue(column, value, pageBuilder);
+    }
+
+    @Override
+    public void copyValueTo(BooleanColumnReader columnReader)
+    {
+        columnReader.value = this.value;
+    }
+}

data/src/main/java/org/embulk/filter/add_time/reader/ColumnReader.java

@@ -0,0 +1,14 @@
+package org.embulk.filter.add_time.reader;
+
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageReader;
+
+public interface ColumnReader<T extends ColumnReader>
+{
+    void readValue(Column column, PageReader pageReader);
+
+    void convertValue(Column column, PageBuilder pageBuilder);
+
+    void copyTo(T columnReader);
+}

data/src/main/java/org/embulk/filter/add_time/reader/DoubleColumnReader.java

@@ -0,0 +1,35 @@
+package org.embulk.filter.add_time.reader;
+
+import org.embulk.filter.add_time.converter.ValueConverter;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageReader;
+
+public class DoubleColumnReader
+        extends AbstractColumnReader<DoubleColumnReader>
+{
+    protected double value;
+
+    public DoubleColumnReader(ValueConverter valueConverter)
+    {
+        super(valueConverter);
+    }
+
+    @Override
+    public void readNonNullValue(Column column, PageReader pageReader)
+    {
+        value = pageReader.getDouble(column);
+    }
+
+    @Override
+    public void convertNonNullValue(Column column, PageBuilder pageBuilder)
+    {
+        valueConverter.convertValue(column, value, pageBuilder);
+    }
+
+    @Override
+    public void copyValueTo(DoubleColumnReader columnReader)
+    {
+        columnReader.value = this.value;
+    }
+}

data/src/main/java/org/embulk/filter/add_time/reader/LongColumnReader.java

@@ -0,0 +1,35 @@
+package org.embulk.filter.add_time.reader;
+
+import org.embulk.filter.add_time.converter.ValueConverter;
+import org.embulk.spi.Column;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageReader;
+
+public class LongColumnReader
+        extends AbstractColumnReader<LongColumnReader>
+{
+    protected long value;
+
+    public LongColumnReader(ValueConverter valueConverter)
+    {
+        super(valueConverter);
+    }
+
+    @Override
+    public void readNonNullValue(Column column, PageReader pageReader)
+    {
+        value = pageReader.getLong(column);
+    }
+
+    @Override
+    public void convertNonNullValue(Column column, PageBuilder pageBuilder)
+    {
+        valueConverter.convertValue(column, value, pageBuilder);
+    }
+
+    @Override
+    public void copyValueTo(LongColumnReader columnReader)
+    {
+        columnReader.value = this.value;
+    }
+}