embulk-output-jdbc 0.2.4 → 0.3.0

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

@@ -0,0 +1,59 @@
+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.spi.time.TimestampFormatter;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class NStringColumnSetter
+        extends ColumnSetter
+{
+    private final TimestampFormatter timestampFormatter;
+
+    public NStringColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue,
+            TimestampFormatter timestampFormatter)
+    {
+        super(batch, column, defaultValue);
+        this.timestampFormatter = timestampFormatter;
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setNString();
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setNString(Boolean.toString(v));
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        batch.setNString(Long.toString(v));
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        batch.setNString(Double.toString(v));
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        batch.setNString(v);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        batch.setNString(timestampFormatter.format(v));
+    }
+}

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

@@ -1,43 +1,53 @@
-package org.embulk.output.jdbc.setter;
-
-import java.io.IOException;
-import java.sql.SQLException;
-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 NullColumnSetter
-        extends ColumnSetter
-{
-    public NullColumnSetter(BatchInsert batch, PageReader pageReader,
-            JdbcColumn column)
-    {
-        super(batch, pageReader, column);
-    }
-
-    protected void booleanValue(boolean v) throws IOException, SQLException
-    {
-        nullValue();
-    }
-
-    protected void longValue(long v) throws IOException, SQLException
-    {
-        nullValue();
-    }
-
-    protected void doubleValue(double v) throws IOException, SQLException
-    {
-        nullValue();
-    }
-
-    protected void stringValue(String v) throws IOException, SQLException
-    {
-        nullValue();
-    }
-
-    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.time.Timestamp;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class NullColumnSetter
+        extends ColumnSetter
+{
+    public NullColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        defaultValue.setNull();
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        defaultValue.setNull();
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        defaultValue.setNull();
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        defaultValue.setNull();
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        defaultValue.setNull();
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setNull();
+    }
+}

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

@@ -0,0 +1,105 @@
+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 class NullDefaultValueSetter
+        extends DefaultValueSetter
+{
+    public NullDefaultValueSetter(BatchInsert batch, JdbcColumn column)
+    {
+        super(batch, column);
+    }
+
+    @Override
+    public void setNull() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setBoolean() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setByte() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setShort() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setInt() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setLong() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setFloat() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setDouble() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setBigDecimal() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setString() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setNString() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setBytes() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setSqlDate() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setSqlTime() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void setSqlTimestamp() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+}

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

@@ -0,0 +1,56 @@
+package org.embulk.output.jdbc.setter;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.spi.time.TimestampFormatter;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class PassThroughColumnSetter
+        extends ColumnSetter
+{
+    public PassThroughColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        batch.setNull(column.getSqlType());
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setBoolean(v);
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        batch.setLong(v);
+    }
+
+    @Override
+    public void doubleValue(double v) throws IOException, SQLException
+    {
+        batch.setDouble(v);
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        batch.setString(v);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        java.sql.Timestamp t = new java.sql.Timestamp(v.toEpochMilli());
+        t.setNanos(v.getNano());
+        batch.setSqlTimestamp(t, getSqlType());
+    }
+}

data/src/main/java/org/embulk/output/jdbc/setter/ShortColumnSetter.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 ShortColumnSetter
+        extends ColumnSetter
+{
+    public ShortColumnSetter(BatchInsert batch, JdbcColumn column,
+            DefaultValueSetter defaultValue)
+    {
+        super(batch, column, defaultValue);
+    }
+
+    @Override
+    public void nullValue() throws IOException, SQLException
+    {
+        defaultValue.setShort();
+    }
+
+    @Override
+    public void booleanValue(boolean v) throws IOException, SQLException
+    {
+        batch.setShort(v ? (short) 1 : (short) 0);
+    }
+
+    @Override
+    public void longValue(long v) throws IOException, SQLException
+    {
+        if (v > Short.MAX_VALUE || v < Short.MIN_VALUE) {
+            defaultValue.setShort();
+        } else {
+            batch.setShort((short) 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.setShort();
+            return;
+        }
+        longValue(lv);
+    }
+
+    @Override
+    public void stringValue(String v) throws IOException, SQLException
+    {
+        short sv;
+        try {
+            sv = Short.parseShort(v);
+        } catch (NumberFormatException e) {
+            defaultValue.setShort();
+            return;
+        }
+        batch.setShort(sv);
+    }
+
+    @Override
+    public void timestampValue(Timestamp v) throws IOException, SQLException
+    {
+        defaultValue.setShort();
+    }
+}

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

@@ -1,38 +1,43 @@
-package org.embulk.output.jdbc.setter;
-
-import org.embulk.spi.PageReader;
-import org.embulk.spi.time.Timestamp;
-import org.embulk.output.jdbc.BatchInsert;
-
-public class SkipColumnSetter
-        extends ColumnSetter
-{
-    public SkipColumnSetter(BatchInsert batch, PageReader pageReader)
-    {
-        super(batch, pageReader, null);
-    }
-
-    protected void booleanValue(boolean v)
-    {
-    }
-
-    protected void longValue(long v)
-    {
-    }
-
-    protected void doubleValue(double v)
-    {
-    }
-
-    protected void stringValue(String v)
-    {
-    }
-
-    protected void timestampValue(Timestamp v)
-    {
-    }
-
-    protected void nullValue()
-    {
-    }
-}
+package org.embulk.output.jdbc.setter;
+
+import org.embulk.spi.time.Timestamp;
+import org.embulk.output.jdbc.BatchInsert;
+
+public class SkipColumnSetter
+        extends ColumnSetter
+{
+    public SkipColumnSetter(BatchInsert batch)
+    {
+        super(batch, null, null);
+    }
+
+    @Override
+    public void booleanValue(boolean v)
+    {
+    }
+
+    @Override
+    public void longValue(long v)
+    {
+    }
+
+    @Override
+    public void doubleValue(double v)
+    {
+    }
+
+    @Override
+    public void stringValue(String v)
+    {
+    }
+
+    @Override
+    public void timestampValue(Timestamp v)
+    {
+    }
+
+    @Override
+    public void nullValue()
+    {
+    }
+}