embulk-output-cassandra 0.1.0

data/src/main/java/org/embulk/output/cassandra/setter/BigintColumnSetter.java

@@ -0,0 +1,41 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+
+public class BigintColumnSetter extends CassandraColumnSetter
+{
+    public BigintColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setBooleanValue(Boolean value, BoundStatement statement)
+    {
+        if (value) {
+            statement.setLong(cassandraColumn.getName(), 1);
+        }
+        else {
+            statement.setLong(cassandraColumn.getName(), 0);
+        }
+    }
+
+    @Override
+    public void setLongValue(Long value, BoundStatement statement)
+    {
+        statement.setLong(cassandraColumn.getName(), value);
+    }
+
+    @Override
+    public void setDoubleValue(Double value, BoundStatement statement)
+    {
+        statement.setLong(cassandraColumn.getName(), value.longValue());
+    }
+
+    @Override
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        statement.setLong(cassandraColumn.getName(), Long.parseLong(value));
+    }
+}

data/src/main/java/org/embulk/output/cassandra/setter/BooleanColumnSetter.java

@@ -0,0 +1,30 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+
+public class BooleanColumnSetter extends CassandraColumnSetter
+{
+    public BooleanColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setBooleanValue(Boolean value, BoundStatement statement)
+    {
+        statement.setBool(cassandraColumn.getName(), value);
+    }
+
+    @Override
+    public void setLongValue(Long value, BoundStatement statement)
+    {
+        statement.setBool(cassandraColumn.getName(), value != 0);
+    }
+
+    @Override
+    public void setDoubleValue(Double value, BoundStatement statement)
+    {
+        statement.setBool(cassandraColumn.getName(), value != 0);
+    }
+}

data/src/main/java/org/embulk/output/cassandra/setter/CassandraColumnSetter.java

@@ -0,0 +1,51 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+import org.embulk.spi.time.Timestamp;
+import org.msgpack.value.Value;
+
+public abstract class CassandraColumnSetter
+{
+    protected ColumnMetadata cassandraColumn;
+
+    public CassandraColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        this.cassandraColumn = cassandraColumn;
+    }
+
+    public void setBooleanValue(Boolean value, BoundStatement statement)
+    {
+        throw new RuntimeException("Unsupported type conversion: " + value.getClass().getName() + " -> " + cassandraColumn.getType().getName());
+    };
+
+    public void setLongValue(Long value, BoundStatement statement)
+    {
+        throw new RuntimeException("Unsupported type conversion: " + value.getClass().getName() + " -> " + cassandraColumn.getType().getName());
+    };
+
+    public void setDoubleValue(Double value, BoundStatement statement)
+    {
+        throw new RuntimeException("Unsupported type conversion: " + value.getClass().getName() + " -> " + cassandraColumn.getType().getName());
+    };
+
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        throw new RuntimeException("Unsupported type conversion: " + value.getClass().getName() + " -> " + cassandraColumn.getType().getName());
+    };
+
+    public void setTimestampValue(Timestamp value, BoundStatement statement)
+    {
+        throw new RuntimeException("Unsupported type conversion: " + value.getClass().getName() + " -> " + cassandraColumn.getType().getName());
+    };
+
+    public void setJsonValue(Value value, BoundStatement statement)
+    {
+        throw new RuntimeException("Unsupported type conversion: " + value.getClass().getName() + " -> " + cassandraColumn.getType().getName());
+    };
+
+    public void setNullValue(BoundStatement statement)
+    {
+        statement.setToNull(cassandraColumn.getName());
+    }
+}

data/src/main/java/org/embulk/output/cassandra/setter/CassandraColumnSetterFactory.java

@@ -0,0 +1,60 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.Cluster;
+import com.datastax.driver.core.ColumnMetadata;
+
+public class CassandraColumnSetterFactory
+{
+    private CassandraColumnSetterFactory()
+    {
+    }
+
+    public static CassandraColumnSetter createColumnSetter(ColumnMetadata cassandraColumn, Cluster cluster)
+    {
+        switch (cassandraColumn.getType().getName()) {
+            case TEXT:
+            case VARCHAR:
+            case ASCII:
+                return new TextColumnSetter(cassandraColumn);
+            case BOOLEAN:
+                return new BooleanColumnSetter(cassandraColumn);
+            case BIGINT:
+                return new BigintColumnSetter(cassandraColumn);
+            case INT:
+                return new IntColumnSetter(cassandraColumn);
+            case SMALLINT:
+                return new SmallintColumnSetter(cassandraColumn);
+            case TINYINT:
+                return new TinyintColumnSetter(cassandraColumn);
+            case VARINT:
+                return new VarintColumnSetter(cassandraColumn);
+            case DECIMAL:
+                return new DecimalColumnSetter(cassandraColumn);
+            case DOUBLE:
+                return new DoubleColumnSetter(cassandraColumn);
+            case FLOAT:
+                return new FloatColumnSetter(cassandraColumn);
+            case TIMESTAMP:
+                return new TimestampColumnSetter(cassandraColumn);
+            case DATE:
+                return new DateColumnSetter(cassandraColumn);
+            case TIME:
+                return new TimeColumnSetter(cassandraColumn);
+            case UUID:
+                return new UuidColumnSetter(cassandraColumn);
+            case TIMEUUID:
+                return new TimeuuidColumnSetter(cassandraColumn);
+            case LIST:
+                return new ListColumnSetter(cassandraColumn);
+            case MAP:
+                return new MapColumnSetter(cassandraColumn);
+            case SET:
+                return new SetColumnSetter(cassandraColumn);
+            case INET:
+                return new InetColumnSetter(cassandraColumn);
+            case TUPLE:
+                return new TupleColumnSetter(cassandraColumn, cluster);
+        }
+        throw new RuntimeException(cassandraColumn.getName() + ":" + cassandraColumn.getType().getName() + " is unsupported Type");
+    }
+}

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

@@ -0,0 +1,96 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import org.embulk.spi.Column;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.PageReader;
+
+public class ColumnSetterVisitor implements ColumnVisitor
+{
+    private final PageReader pageReader;
+    private final CassandraColumnSetter setter;
+
+    private BoundStatement statement;
+
+    public ColumnSetterVisitor(PageReader pageReader, CassandraColumnSetter setter)
+    {
+        this.pageReader = pageReader;
+        this.setter = setter;
+    }
+
+    public boolean hasSetter()
+    {
+        return this.setter != null;
+    }
+
+    public void setStatement(BoundStatement statement)
+    {
+        this.statement = statement;
+    }
+
+    @Override
+    public void booleanColumn(Column column)
+    {
+        if (pageReader.isNull(column)) {
+            setter.setNullValue(statement);
+        }
+        else {
+            setter.setBooleanValue(pageReader.getBoolean(column), statement);
+        }
+    }
+
+    @Override
+    public void longColumn(Column column)
+    {
+        if (pageReader.isNull(column)) {
+            setter.setNullValue(statement);
+        }
+        else {
+            setter.setLongValue(pageReader.getLong(column), statement);
+        }
+    }
+
+    @Override
+    public void doubleColumn(Column column)
+    {
+        if (pageReader.isNull(column)) {
+            setter.setNullValue(statement);
+        }
+        else {
+            setter.setDoubleValue(pageReader.getDouble(column), statement);
+        }
+    }
+
+    @Override
+    public void stringColumn(Column column)
+    {
+        if (pageReader.isNull(column)) {
+            setter.setNullValue(statement);
+        }
+        else {
+            setter.setStringValue(pageReader.getString(column), statement);
+        }
+    }
+
+    @Override
+    public void timestampColumn(Column column)
+    {
+        if (pageReader.isNull(column)) {
+            setter.setNullValue(statement);
+        }
+        else {
+            setter.setTimestampValue(pageReader.getTimestamp(column), statement);
+        }
+    }
+
+    @Override
+    public void jsonColumn(Column column)
+    {
+        if (pageReader.isNull(column)) {
+            setter.setNullValue(statement);
+        }
+        else {
+            setter.setJsonValue(pageReader.getJson(column), statement);
+        }
+    }
+}

data/src/main/java/org/embulk/output/cassandra/setter/DateColumnSetter.java

@@ -0,0 +1,29 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+import org.embulk.spi.time.Timestamp;
+
+import java.time.LocalDate;
+
+public class DateColumnSetter extends CassandraColumnSetter
+{
+    public DateColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        LocalDate date = LocalDate.parse(value);
+        statement.setDate(cassandraColumn.getName(), com.datastax.driver.core.LocalDate.fromYearMonthDay(date.getYear(), date.getMonthValue(), date.getDayOfMonth()));
+    }
+
+    @Override
+    public void setTimestampValue(Timestamp value, BoundStatement statement)
+    {
+        com.datastax.driver.core.LocalDate date = com.datastax.driver.core.LocalDate.fromMillisSinceEpoch(value.toEpochMilli());
+        statement.setDate(cassandraColumn.getName(), date);
+    }
+}

data/src/main/java/org/embulk/output/cassandra/setter/DecimalColumnSetter.java

@@ -0,0 +1,43 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+
+import java.math.BigDecimal;
+
+public class DecimalColumnSetter extends CassandraColumnSetter
+{
+    public DecimalColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setBooleanValue(Boolean value, BoundStatement statement)
+    {
+        if (value) {
+            statement.setDecimal(cassandraColumn.getName(), BigDecimal.valueOf(1));
+        }
+        else {
+            statement.setDecimal(cassandraColumn.getName(), BigDecimal.valueOf(0));
+        }
+    }
+
+    @Override
+    public void setLongValue(Long value, BoundStatement statement)
+    {
+        statement.setDecimal(cassandraColumn.getName(), BigDecimal.valueOf(value));
+    }
+
+    @Override
+    public void setDoubleValue(Double value, BoundStatement statement)
+    {
+        statement.setDecimal(cassandraColumn.getName(), BigDecimal.valueOf(value));
+    }
+
+    @Override
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        statement.setDecimal(cassandraColumn.getName(), new BigDecimal(value));
+    }
+}

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

@@ -0,0 +1,41 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+
+public class DoubleColumnSetter extends CassandraColumnSetter
+{
+    public DoubleColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setBooleanValue(Boolean value, BoundStatement statement)
+    {
+        if (value) {
+            statement.setDouble(cassandraColumn.getName(), 1);
+        }
+        else {
+            statement.setDouble(cassandraColumn.getName(), 0);
+        }
+    }
+
+    @Override
+    public void setLongValue(Long value, BoundStatement statement)
+    {
+        statement.setDouble(cassandraColumn.getName(), value.doubleValue());
+    }
+
+    @Override
+    public void setDoubleValue(Double value, BoundStatement statement)
+    {
+        statement.setDouble(cassandraColumn.getName(), value);
+    }
+
+    @Override
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        statement.setDouble(cassandraColumn.getName(), Double.parseDouble(value));
+    }
+}

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

@@ -0,0 +1,41 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+
+public class FloatColumnSetter extends CassandraColumnSetter
+{
+    public FloatColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setBooleanValue(Boolean value, BoundStatement statement)
+    {
+        if (value) {
+            statement.setFloat(cassandraColumn.getName(), 1);
+        }
+        else {
+            statement.setFloat(cassandraColumn.getName(), 0);
+        }
+    }
+
+    @Override
+    public void setLongValue(Long value, BoundStatement statement)
+    {
+        statement.setFloat(cassandraColumn.getName(), value.floatValue());
+    }
+
+    @Override
+    public void setDoubleValue(Double value, BoundStatement statement)
+    {
+        statement.setFloat(cassandraColumn.getName(), value.floatValue());
+    }
+
+    @Override
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        statement.setFloat(cassandraColumn.getName(), Float.parseFloat(value));
+    }
+}

data/src/main/java/org/embulk/output/cassandra/setter/InetColumnSetter.java

@@ -0,0 +1,26 @@
+package org.embulk.output.cassandra.setter;
+
+import com.datastax.driver.core.BoundStatement;
+import com.datastax.driver.core.ColumnMetadata;
+
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+
+public class InetColumnSetter extends CassandraColumnSetter
+{
+    public InetColumnSetter(ColumnMetadata cassandraColumn)
+    {
+        super(cassandraColumn);
+    }
+
+    @Override
+    public void setStringValue(String value, BoundStatement statement)
+    {
+        try {
+            statement.setInet(cassandraColumn.getName(), InetAddress.getByName(value));
+        }
+        catch (UnknownHostException e) {
+            throw new RuntimeException(e);
+        }
+    }
+}