embulk-output-jdbc 0.2.4 → 0.3.0

data/src/main/java/org/embulk/output/jdbc/setter/ColumnSetterVisitor.java

@@ -0,0 +1,96 @@
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import org.embulk.spi.Column;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.PageReader;
+import org.embulk.spi.time.Timestamp;
+
+public class ColumnSetterVisitor
+        implements ColumnVisitor
+{
+    private final PageReader pageReader;
+    private final ColumnSetter setter;
+
+    public ColumnSetterVisitor(PageReader pageReader, ColumnSetter setter)
+    {
+        this.pageReader = pageReader;
+        this.setter = setter;
+    }
+
+    @Override
+    public void booleanColumn(Column column)
+    {
+        try {
+            if (pageReader.isNull(column)) {
+                setter.nullValue();
+            } else {
+                setter.booleanValue(pageReader.getBoolean(column));
+            }
+        } catch (IOException | SQLException ex) {
+            // TODO exception class
+            throw new RuntimeException(ex);
+        }
+    }
+
+    @Override
+    public void longColumn(Column column)
+    {
+        try {
+            if (pageReader.isNull(column)) {
+                setter.nullValue();
+            } else {
+                setter.longValue(pageReader.getLong(column));
+            }
+        } catch (IOException | SQLException ex) {
+            // TODO exception class
+            throw new RuntimeException(ex);
+        }
+    }
+
+    @Override
+    public void doubleColumn(Column column)
+    {
+        try {
+            if (pageReader.isNull(column)) {
+                setter.nullValue();
+            } else {
+                setter.doubleValue(pageReader.getDouble(column));
+            }
+        } catch (IOException | SQLException ex) {
+            // TODO exception class
+            throw new RuntimeException(ex);
+        }
+    }
+
+    @Override
+    public void stringColumn(Column column)
+    {
+        try {
+            if (pageReader.isNull(column)) {
+                setter.nullValue();
+            } else {
+                setter.stringValue(pageReader.getString(column));
+            }
+        } catch (IOException | SQLException ex) {
+            // TODO exception class
+            throw new RuntimeException(ex);
+        }
+    }
+
+    @Override
+    public void timestampColumn(Column column)
+    {
+        try {
+            if (pageReader.isNull(column)) {
+                setter.nullValue();
+            } else {
+                setter.timestampValue(pageReader.getTimestamp(column));
+            }
+        } catch (IOException | SQLException ex) {
+            // TODO exception class
+            throw new RuntimeException(ex);
+        }
+    }
+}

data/src/main/java/org/embulk/output/jdbc/setter/DefaultValueSetter.java

@@ -0,0 +1,48 @@
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public abstract class DefaultValueSetter
+{
+    protected final BatchInsert batch;
+    protected final JdbcColumn column;
+
+    public DefaultValueSetter(BatchInsert batch, JdbcColumn column)
+    {
+        this.batch = batch;
+        this.column = column;
+    }
+
+    public abstract void setNull() throws IOException, SQLException;
+
+    public abstract void setBoolean() throws IOException, SQLException;
+
+    public abstract void setByte() throws IOException, SQLException;
+
+    public abstract void setShort() throws IOException, SQLException;
+
+    public abstract void setInt() throws IOException, SQLException;
+
+    public abstract void setLong() throws IOException, SQLException;
+
+    public abstract void setFloat() throws IOException, SQLException;
+
+    public abstract void setDouble() throws IOException, SQLException;
+
+    public abstract void setBigDecimal() throws IOException, SQLException;
+
+    public abstract void setString() throws IOException, SQLException;
+
+    public abstract void setNString() throws IOException, SQLException;
+
+    public abstract void setBytes() throws IOException, SQLException;
+
+    public abstract void setSqlDate() throws IOException, SQLException;
+
+    public abstract void setSqlTime() throws IOException, SQLException;
+
+    public abstract void setSqlTimestamp() throws IOException, SQLException;
+}

data/src/main/java/org/embulk/output/jdbc/setter/DoubleColumnSetter.java

@@ -1,51 +1,61 @@
-package org.embulk.output.jdbc.setter;
-
-import java.io.IOException;
-import java.sql.SQLException;
-import org.embulk.spi.ColumnVisitor;
-import org.embulk.spi.PageReader;
-import org.embulk.spi.time.Timestamp;
-import org.embulk.output.jdbc.JdbcColumn;
-import org.embulk.output.jdbc.BatchInsert;
-
-public class DoubleColumnSetter
-        extends ColumnSetter
-{
-    public DoubleColumnSetter(BatchInsert batch, PageReader pageReader,
-            JdbcColumn column)
-    {
-        super(batch, pageReader, column);
-    }
-
-    protected void booleanValue(boolean v) throws IOException, SQLException
-    {
-        batch.setDouble(v ? 1.0 : 0.0);
-    }
-
-    protected void longValue(long v) throws IOException, SQLException
-    {
-        batch.setDouble((double) v);
-    }
-
-    protected void doubleValue(double v) throws IOException, SQLException
-    {
-        batch.setDouble(v);
-    }
-
-    protected void stringValue(String v) throws IOException, SQLException
-    {
-        double dv;
-        try {
-            dv = Double.parseDouble(v);
-        } catch (NumberFormatException e) {
-            nullValue();
-            return;
-        }
-        batch.setDouble(dv);
-    }
-
-    protected void timestampValue(Timestamp v) throws IOException, SQLException
-    {
-        nullValue();
-    }
-}
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class DoubleColumnSetter
+        extends ColumnSetter
+{
+    public DoubleColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setDouble();
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setDouble(v ? 1.0 : 0.0);
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        batch.setDouble((double) v);
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        batch.setDouble(v);
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        double dv;
+        try {
+            dv = Double.parseDouble(v);
+        } catch (NumberFormatException e) {
+            defaultValue.setDouble();
+            return;
+        }
+        batch.setDouble(dv);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        defaultValue.setDouble();
+    }
+}

data/src/main/java/org/embulk/output/jdbc/setter/FloatColumnSetter.java

@@ -0,0 +1,61 @@
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class FloatColumnSetter
+        extends ColumnSetter
+{
+    public FloatColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setFloat();
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setFloat(v ? (float) 1.0 : (float) 0.0);
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        batch.setFloat((float) v);
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        batch.setFloat((float) v);
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        float fv;
+        try {
+            fv = Float.parseFloat(v);
+        } catch (NumberFormatException e) {
+            defaultValue.setFloat();
+            return;
+        }
+        batch.setFloat(fv);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        defaultValue.setFloat();
+    }
+}

data/src/main/java/org/embulk/output/jdbc/setter/IntColumnSetter.java

@@ -0,0 +1,76 @@
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import java.math.RoundingMode;
+import com.google.common.math.DoubleMath;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class IntColumnSetter
+        extends ColumnSetter
+{
+    public IntColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setInt();
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setInt(v ? 1 : 0);
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        if (v > Integer.MAX_VALUE || v < Integer.MIN_VALUE) {
+            defaultValue.setInt();
+        } else {
+            batch.setInt((int) v);
+        }
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        long lv;
+        try {
+            // TODO configurable rounding mode
+            lv = DoubleMath.roundToLong(v, RoundingMode.HALF_UP);
+        } catch (ArithmeticException ex) {
+            // NaN / Infinite / -Infinite
+            defaultValue.setInt();
+            return;
+        }
+        longValue(lv);
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        int iv;
+        try {
+            iv = Integer.parseInt(v);
+        } catch (NumberFormatException e) {
+            defaultValue.setInt();
+            return;
+        }
+        batch.setInt(iv);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        defaultValue.setInt();
+    }
+}

data/src/main/java/org/embulk/output/jdbc/setter/LongColumnSetter.java

@@ -1,62 +1,72 @@
-package org.embulk.output.jdbc.setter;
-
-import java.io.IOException;
-import java.sql.SQLException;
-import java.math.RoundingMode;
-import com.google.common.math.DoubleMath;
-import org.embulk.spi.ColumnVisitor;
-import org.embulk.spi.PageReader;
-import org.embulk.spi.time.Timestamp;
-import org.embulk.output.jdbc.JdbcColumn;
-import org.embulk.output.jdbc.BatchInsert;
-
-public class LongColumnSetter
-        extends ColumnSetter
-{
-    public LongColumnSetter(BatchInsert batch, PageReader pageReader,
-            JdbcColumn column)
-    {
-        super(batch, pageReader, column);
-    }
-
-    protected void booleanValue(boolean v) throws IOException, SQLException
-    {
-        batch.setLong(v ? 1L : 0L);
-    }
-
-    protected void longValue(long v) throws IOException, SQLException
-    {
-        batch.setLong(v);
-    }
-
-    protected void doubleValue(double v) throws IOException, SQLException
-    {
-        long lv;
-        try {
-            // TODO configurable rounding mode
-            lv = DoubleMath.roundToLong(v, RoundingMode.HALF_UP);
-        } catch (ArithmeticException ex) {
-            // NaN / Infinite / -Infinite
-            nullValue();
-            return;
-        }
-        batch.setLong(lv);
-    }
-
-    protected void stringValue(String v) throws IOException, SQLException
-    {
-        long lv;
-        try {
-            lv = Long.parseLong(v);
-        } catch (NumberFormatException e) {
-            nullValue();
-            return;
-        }
-        batch.setLong(lv);
-    }
-
-    protected void timestampValue(Timestamp v) throws IOException, SQLException
-    {
-        nullValue();
-    }
-}
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import java.math.RoundingMode;
+import com.google.common.math.DoubleMath;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class LongColumnSetter
+        extends ColumnSetter
+{
+    public LongColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setLong();
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setLong(v ? 1L : 0L);
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        batch.setLong(v);
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        long lv;
+        try {
+            // TODO configurable rounding mode
+            lv = DoubleMath.roundToLong(v, RoundingMode.HALF_UP);
+        } catch (ArithmeticException ex) {
+            // NaN / Infinite / -Infinite
+            defaultValue.setLong();
+            return;
+        }
+        batch.setLong(lv);
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        long lv;
+        try {
+            lv = Long.parseLong(v);
+        } catch (NumberFormatException e) {
+            defaultValue.setLong();
+            return;
+        }
+        batch.setLong(lv);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        defaultValue.setLong();
+    }
+}