embulk-output-teradata 0.1.2 → 0.1.3

data/src/main/java/org/embulk/output/teradata/jdbc/TimestampFormat.java

@@ -1,37 +0,0 @@
-package org.embulk.output.jdbc;
-
-import java.sql.Timestamp;
-import java.text.FieldPosition;
-import java.text.SimpleDateFormat;
-import java.util.Date;
-
-
-public class TimestampFormat extends SimpleDateFormat
-{
-
-    private final int scale;
-
-    public TimestampFormat(String pattern, int scale)
-    {
-        super(pattern);
-
-        this.scale = scale;
-    }
-
-    @Override
-    public StringBuffer format(Date date, StringBuffer toAppendTo, FieldPosition pos)
-    {
-        StringBuffer buffer = super.format(date, toAppendTo, pos);
-        if (scale > 0) {
-            buffer.append('.');
-            String nanos = Integer.toString(((Timestamp)date).getNanos());
-            int zeros = Math.min(scale, 9 - nanos.length());
-            for (int i = 0; i < zeros; i++) {
-                buffer.append('0');
-            }
-            buffer.append(nanos.substring(0, scale - zeros));
-        }
-        return buffer;
-    }
-
-}

data/src/main/java/org/embulk/output/teradata/jdbc/ToString.java

@@ -1,54 +0,0 @@
-package org.embulk.output.jdbc;
-
-import com.google.common.base.Optional;
-import com.fasterxml.jackson.databind.JsonNode;
-import com.fasterxml.jackson.databind.JsonMappingException;
-import com.fasterxml.jackson.databind.node.NullNode;
-import com.fasterxml.jackson.annotation.JsonCreator;
-import com.fasterxml.jackson.annotation.JsonValue;
-
-public class ToString
-{
-    private final String string;
-
-    public ToString(String string)
-    {
-        this.string = string;
-    }
-
-    @JsonCreator
-    ToString(Optional<JsonNode> option) throws JsonMappingException
-    {
-        JsonNode node = option.or(NullNode.getInstance());
-        if (node.isTextual()) {
-            this.string = node.textValue();
-        } else if (node.isValueNode()) {
-            this.string = node.toString();
-        } else {
-            throw new JsonMappingException(String.format("Arrays and objects are invalid: '%s'", node));
-        }
-    }
-
-    @Override
-    public boolean equals(Object obj)
-    {
-        if (!(obj instanceof ToString)) {
-            return false;
-        }
-        ToString o = (ToString) obj;
-        return string.equals(o.string);
-    }
-
-    @Override
-    public int hashCode()
-    {
-        return string.hashCode();
-    }
-
-    @JsonValue
-    @Override
-    public String toString()
-    {
-        return string;
-    }
-}

data/src/main/java/org/embulk/output/teradata/jdbc/ToStringMap.java

@@ -1,34 +0,0 @@
-package org.embulk.output.jdbc;
-
-import java.util.Map;
-import java.util.HashMap;
-import java.util.Properties;
-import com.google.common.base.Function;
-import com.google.common.collect.Maps;
-import com.fasterxml.jackson.annotation.JsonCreator;
-
-public class ToStringMap
-        extends HashMap<String, String>
-{
-    @JsonCreator
-    ToStringMap(Map<String, ToString> map)
-    {
-        super(Maps.transformValues(map, new Function<ToString, String>() {
-            public String apply(ToString value)
-            {
-                if (value == null) {
-                    return "null";
-                } else {
-                    return value.toString();
-                }
-            }
-        }));
-    }
-
-    public Properties toProperties()
-    {
-        Properties props = new Properties();
-        props.putAll(this);
-        return props;
-    }
-}

data/src/main/java/org/embulk/output/teradata/jdbc/setter/BigDecimalColumnSetter.java

@@ -1,76 +0,0 @@
-package org.embulk.output.jdbc.setter;
-
-import java.math.BigDecimal;
-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;
-import org.msgpack.value.Value;
-
-public class BigDecimalColumnSetter
-        extends ColumnSetter
-{
-    private static final BigDecimal ZERO = BigDecimal.valueOf(0L);
-    private static final BigDecimal ONE = BigDecimal.valueOf(1L);
-
-    public BigDecimalColumnSetter(BatchInsert batch, JdbcColumn column,
-            DefaultValueSetter defaultValue)
-    {
-        super(batch, column, defaultValue);
-    }
-
-    @Override
-    public void nullValue() throws IOException, SQLException
-    {
-        defaultValue.setBigDecimal();
-    }
-
-    @Override
-    public void booleanValue(boolean v) throws IOException, SQLException
-    {
-        batch.setBigDecimal(v ? ONE : ZERO);
-    }
-
-    @Override
-    public void longValue(long v) throws IOException, SQLException
-    {
-        batch.setBigDecimal(BigDecimal.valueOf(v));
-    }
-
-    @Override
-    public void doubleValue(double v) throws IOException, SQLException
-    {
-        if (Double.isNaN(v) || Double.isInfinite(v)) {
-            defaultValue.setBigDecimal();
-        } else {
-            batch.setBigDecimal(BigDecimal.valueOf(v));
-        }
-    }
-
-    @Override
-    public void stringValue(String v) throws IOException, SQLException
-    {
-        BigDecimal dv;
-        try {
-            dv = new BigDecimal(v);
-        } catch (NumberFormatException ex) {
-            defaultValue.setBigDecimal();
-            return;
-        }
-        batch.setBigDecimal(dv);
-    }
-
-    @Override
-    public void timestampValue(Timestamp v) throws IOException, SQLException
-    {
-        defaultValue.setBigDecimal();
-    }
-
-    @Override
-    public void jsonValue(Value v) throws IOException, SQLException
-    {
-        defaultValue.setBigDecimal();
-    }
-}

data/src/main/java/org/embulk/output/teradata/jdbc/setter/BooleanColumnSetter.java

@@ -1,61 +0,0 @@
-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;
-import org.msgpack.value.Value;
-
-public class BooleanColumnSetter
-        extends ColumnSetter
-{
-    public BooleanColumnSetter(BatchInsert batch, JdbcColumn column,
-            DefaultValueSetter defaultValue)
-    {
-        super(batch, column, defaultValue);
-    }
-
-    @Override
-    public void nullValue() throws IOException, SQLException
-    {
-        defaultValue.setBoolean();
-    }
-
-    @Override
-    public void booleanValue(boolean v) throws IOException, SQLException
-    {
-        batch.setBoolean(v);
-    }
-
-    @Override
-    public void longValue(long v) throws IOException, SQLException
-    {
-        batch.setBoolean(v > 0);
-    }
-
-    @Override
-    public void doubleValue(double v) throws IOException, SQLException
-    {
-        batch.setBoolean(v > 0.0);
-    }
-
-    @Override
-    public void stringValue(String v) throws IOException, SQLException
-    {
-        defaultValue.setBoolean();
-    }
-
-    @Override
-    public void timestampValue(Timestamp v) throws IOException, SQLException
-    {
-        defaultValue.setBoolean();
-    }
-
-    @Override
-    public void jsonValue(Value v) throws IOException, SQLException
-    {
-        defaultValue.setBoolean();
-    }
-}

data/src/main/java/org/embulk/output/teradata/jdbc/setter/ByteColumnSetter.java

@@ -1,84 +0,0 @@
-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.time.Timestamp;
-import org.embulk.output.jdbc.JdbcColumn;
-import org.embulk.output.jdbc.BatchInsert;
-import org.msgpack.value.Value;
-
-public class ByteColumnSetter
-        extends ColumnSetter
-{
-    public ByteColumnSetter(BatchInsert batch, JdbcColumn column,
-            DefaultValueSetter defaultValue)
-    {
-        super(batch, column, defaultValue);
-    }
-
-    @Override
-    public void nullValue() throws IOException, SQLException
-    {
-        defaultValue.setByte();
-    }
-
-    @Override
-    public void booleanValue(boolean v) throws IOException, SQLException
-    {
-        batch.setByte(v ? (byte) 1 : (byte) 0);
-    }
-
-    @Override
-    public void longValue(long v) throws IOException, SQLException
-    {
-        if (v > Byte.MAX_VALUE || v < Byte.MIN_VALUE) {
-            defaultValue.setByte();
-        } else {
-            batch.setByte((byte) 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.setByte();
-            return;
-        }
-        longValue(lv);
-    }
-
-    @Override
-    public void stringValue(String v) throws IOException, SQLException
-    {
-        byte sv;
-        try {
-            sv = Byte.parseByte(v);
-        } catch (NumberFormatException e) {
-            defaultValue.setByte();
-            return;
-        }
-        batch.setByte(sv);
-    }
-
-    @Override
-    public void timestampValue(Timestamp v) throws IOException, SQLException
-    {
-        defaultValue.setByte();
-    }
-
-    @Override
-    public void jsonValue(Value v) throws IOException, SQLException
-    {
-        defaultValue.setByte();
-    }
-}

data/src/main/java/org/embulk/output/teradata/jdbc/setter/ColumnSetter.java

@@ -1,48 +0,0 @@
-package org.embulk.output.jdbc.setter;
-
-import java.io.IOException;
-import java.sql.SQLException;
-
-import org.embulk.output.jdbc.BatchInsert;
-import org.embulk.output.jdbc.JdbcColumn;
-import org.embulk.spi.time.Timestamp;
-import org.msgpack.value.Value;
-
-public abstract class ColumnSetter
-{
-    protected final BatchInsert batch;
-    protected final JdbcColumn column;
-    protected final DefaultValueSetter defaultValue;
-
-    public ColumnSetter(BatchInsert batch, JdbcColumn column,
-            DefaultValueSetter defaultValue)
-    {
-        this.batch = batch;
-        this.column = column;
-        this.defaultValue = defaultValue;
-    }
-
-    public JdbcColumn getColumn()
-    {
-        return column;
-    }
-
-    public int getSqlType()
-    {
-        return column.getSqlType();
-    }
-
-    public abstract void nullValue() throws IOException, SQLException;
-
-    public abstract void booleanValue(boolean v) throws IOException, SQLException;
-
-    public abstract void longValue(long v) throws IOException, SQLException;
-
-    public abstract void doubleValue(double v) throws IOException, SQLException;
-
-    public abstract void stringValue(String v) throws IOException, SQLException;
-
-    public abstract void timestampValue(Timestamp v) throws IOException, SQLException;
-
-    public abstract void jsonValue(Value v) throws IOException, SQLException;
-}

data/src/main/java/org/embulk/output/teradata/jdbc/setter/ColumnSetterFactory.java

@@ -1,199 +0,0 @@
-package org.embulk.output.jdbc.setter;
-
-import java.util.Calendar;
-import java.util.Locale;
-import java.sql.Types;
-import org.joda.time.DateTimeZone;
-import org.embulk.spi.time.TimestampFormatter;
-import org.embulk.output.jdbc.BatchInsert;
-import org.embulk.output.jdbc.JdbcColumn;
-import org.embulk.output.jdbc.JdbcColumnOption;
-import org.embulk.config.ConfigException;
-
-public class ColumnSetterFactory
-{
-    protected final BatchInsert batch;
-    protected final DateTimeZone defaultTimeZone;
-
-    public ColumnSetterFactory(BatchInsert batch, DateTimeZone defaultTimeZone)
-    {
-        this.batch = batch;
-        this.defaultTimeZone = defaultTimeZone;
-    }
-
-    public SkipColumnSetter newSkipColumnSetter()
-    {
-        return new SkipColumnSetter(batch);
-    }
-
-    public DefaultValueSetter newDefaultValueSetter(JdbcColumn column, JdbcColumnOption option)
-    {
-        return new NullDefaultValueSetter(batch, column);
-    }
-
-    public ColumnSetter newColumnSetter(JdbcColumn column, JdbcColumnOption option)
-    {
-        switch (option.getValueType()) {
-        case "coerce":
-            return newCoalesceColumnSetter(column, option);
-        case "byte":
-            return new ByteColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "short":
-            return new ShortColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "int":
-            return new IntColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "long":
-            return new LongColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "double":
-            return new DoubleColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "float":
-            return new FloatColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "boolean":
-            return new BooleanColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "string":
-            return new StringColumnSetter(batch, column, newDefaultValueSetter(column, option), newTimestampFormatter(option));
-        case "nstring":
-            return new NStringColumnSetter(batch, column, newDefaultValueSetter(column, option), newTimestampFormatter(option));
-        case "date":
-            return new SqlDateColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-        case "time":
-            return new SqlTimeColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-        case "timestamp":
-            return new SqlTimestampColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-        case "decimal":
-            return new BigDecimalColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "json":
-            return new JsonColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "null":
-            return new NullColumnSetter(batch, column, newDefaultValueSetter(column, option));
-        case "pass":
-            return new PassThroughColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-        default:
-            throw new ConfigException(String.format("Unknown value_type '%s' for column '%s'", option.getValueType(), column.getName()));
-        }
-    }
-
-    protected TimestampFormatter newTimestampFormatter(JdbcColumnOption option)
-    {
-        return new TimestampFormatter(
-                option.getJRuby(),
-                option.getTimestampFormat().getFormat(),
-                getTimeZone(option));
-    }
-
-    protected Calendar newCalendar(JdbcColumnOption option)
-    {
-        return Calendar.getInstance(getTimeZone(option).toTimeZone(), Locale.ENGLISH);
-    }
-
-    protected DateTimeZone getTimeZone(JdbcColumnOption option)
-    {
-        return option.getTimeZone().or(defaultTimeZone);
-    }
-
-    public ColumnSetter newCoalesceColumnSetter(JdbcColumn column, JdbcColumnOption option)
-    {
-        switch(column.getSqlType()) {
-        // setByte
-        case Types.TINYINT:
-            return new ByteColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setShort
-        case Types.SMALLINT:
-            return new ShortColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setInt
-        case Types.INTEGER:
-            return new IntColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setLong
-        case Types.BIGINT:
-            return new LongColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setDouble
-        case Types.DOUBLE:
-        case Types.FLOAT:
-            return new DoubleColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setFloat
-        case Types.REAL:
-            return new FloatColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setBool
-        case Types.BOOLEAN:
-        case Types.BIT:  // JDBC BIT is boolean, unlike SQL-92
-            return new BooleanColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // setString, Clob
-        case Types.CHAR:
-        case Types.VARCHAR:
-        case Types.LONGVARCHAR:
-        case Types.CLOB:
-            return new StringColumnSetter(batch, column, newDefaultValueSetter(column, option), newTimestampFormatter(option));
-
-        // setNString, NClob
-        case Types.NCHAR:
-        case Types.NVARCHAR:
-        case Types.LONGNVARCHAR:
-        case Types.NCLOB:
-            return new NStringColumnSetter(batch, column, newDefaultValueSetter(column, option), newTimestampFormatter(option));
-
-        // TODO
-        //// setBytes Blob
-        //case Types.BINARY:
-        //case Types.VARBINARY:
-        //case Types.LONGVARBINARY:
-        //case Types.BLOB:
-        //    return new BytesColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // Time
-        case Types.DATE:
-            return new SqlDateColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-        case Types.TIME:
-            return new SqlTimeColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-        case Types.TIMESTAMP:
-            return new SqlTimestampColumnSetter(batch, column, newDefaultValueSetter(column, option), newCalendar(option));
-
-        // Null
-        case Types.NULL:
-            return new NullColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // BigDecimal
-        case Types.NUMERIC:
-        case Types.DECIMAL:
-            return new BigDecimalColumnSetter(batch, column, newDefaultValueSetter(column, option));
-
-        // others
-        case Types.ARRAY:  // array
-        case Types.STRUCT: // map
-        case Types.REF:
-        case Types.DATALINK:
-        case Types.SQLXML: // XML
-        case Types.ROWID:
-        case Types.DISTINCT:
-        case Types.JAVA_OBJECT:
-        case Types.OTHER:
-        default:
-            throw unsupportedOperationException(column);
-        }
-    }
-
-    //private static String[] UNSUPPORTED = new String[] {
-    //    "ARRAY",
-    //    "STRUCT",
-    //    "REF",
-    //    "DATALINK",
-    //    "SQLXML",
-    //    "ROWID",
-    //    "DISTINCT",
-    //    "OTHER",
-    //};
-
-    private static UnsupportedOperationException unsupportedOperationException(JdbcColumn column)
-    {
-        throw new UnsupportedOperationException(
-                String.format("Unsupported type %s (sqlType=%d, size=%d, scale=%d)",
-                    column.getDeclaredType().or(column.getSimpleTypeName()),
-                    column.getSqlType(), column.getSizeTypeParameter(), column.getScaleTypeParameter()));
-    }
-}