Contents

package org.embulk.output.postgresql;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import org.postgresql.copy.CopyManager;
import org.postgresql.core.BaseConnection;
import org.embulk.spi.Exec;
import org.embulk.output.jdbc.JdbcOutputConnection;
import org.embulk.output.jdbc.JdbcColumn;
import org.embulk.output.jdbc.JdbcSchema;

public class PostgreSQLOutputConnection
        extends JdbcOutputConnection
{
    public PostgreSQLOutputConnection(Connection connection, String schemaName, boolean autoCommit)
            throws SQLException
    {
        super(connection, schemaName);
        connection.setAutoCommit(autoCommit);
    }

    public String buildCopySql(String toTable, JdbcSchema toTableSchema)
    {
        StringBuilder sb = new StringBuilder();

        sb.append("COPY ");
        quoteIdentifierString(sb, toTable);
        sb.append(" (");
        for (int i=0; i < toTableSchema.getCount(); i++) {
            if(i != 0) { sb.append(", "); }
            quoteIdentifierString(sb, toTableSchema.getColumnName(i));
        }
        sb.append(") ");
        sb.append("FROM STDIN");

        return sb.toString();
    }

    public CopyManager newCopyManager() throws SQLException
    {
        return new CopyManager((BaseConnection) connection);
    }

    @Override
    protected String convertTypeName(String typeName)
    {
        switch(typeName) {
        case "CLOB":
            return "TEXT";
        case "BLOB":
            return "BYTEA";
        default:
            return typeName;
        }
    }
}

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
embulk-output-postgresql-0.2.2	src/main/java/org/embulk/output/postgresql/PostgreSQLOutputConnection.java
embulk-output-postgresql-0.2.1	src/main/java/org/embulk/output/postgresql/PostgreSQLOutputConnection.java
embulk-output-postgresql-0.2.0	src/main/java/org/embulk/output/postgresql/PostgreSQLOutputConnection.java
embulk-output-postgresql-0.1.2	src/main/java/org/embulk/output/postgresql/PostgreSQLOutputConnection.java
embulk-output-postgresql-0.1.1	src/main/java/org/embulk/output/postgresql/PostgreSQLOutputConnection.java
embulk-output-postgresql-0.1.0	src/main/java/org/embulk/output/postgresql/PostgreSQLOutputConnection.java