embulk-output-oracle 0.2.4 → 0.3.0

data/src/main/java/org/embulk/output/oracle/OracleOutputConnection.java

@@ -1,134 +1,165 @@
-package org.embulk.output.oracle;
-
-
-import java.nio.charset.Charset;
-import java.sql.Connection;
-import java.sql.PreparedStatement;
-import java.sql.ResultSet;
-import java.sql.SQLException;
-import java.sql.Statement;
-import java.util.HashMap;
-import java.util.Map;
-
-import org.embulk.output.jdbc.JdbcOutputConnection;
-import org.embulk.output.jdbc.JdbcSchema;
-
-public class OracleOutputConnection
-        extends JdbcOutputConnection
-{
-    private static final Map<String, String> CHARSET_NAMES = new HashMap<String, String>();
-    static {
-        CHARSET_NAMES.put("JA16SJIS", "Shift_JIS");
-        CHARSET_NAMES.put("JA16SJISTILDE", "Shift_JIS");
-        CHARSET_NAMES.put("JA16EUC", "EUC-JP");
-        CHARSET_NAMES.put("JA16EUCTILDE", "EUC-JP");
-        CHARSET_NAMES.put("AL32UTF8", "UTF-8");
-        CHARSET_NAMES.put("UTF8", "UTF-8");
-        CHARSET_NAMES.put("AL16UTF16", "UTF-16");
-    }
-
-    private final boolean direct;
-
-
-    public OracleOutputConnection(Connection connection, boolean autoCommit, boolean direct)
-            throws SQLException
-    {
-        super(connection, getSchema(connection));
-        connection.setAutoCommit(autoCommit);
-
-        this.direct = direct;
-    }
-
-    @Override
-    protected String convertTypeName(String typeName)
-    {
-        switch(typeName) {
-        case "BIGINT":
-            return "NUMBER(19,0)";
-        default:
-            return typeName;
-        }
-    }
-
-    @Override
-    protected void setSearchPath(String schema) throws SQLException {
-        // NOP
-    }
-
-
-    @Override
-    public void dropTableIfExists(String tableName) throws SQLException
-    {
-        if (tableExists(tableName)) {
-            dropTable(tableName);
-        }
-    }
-
-    @Override
-    protected void dropTableIfExists(Statement stmt, String tableName) throws SQLException {
-        if (tableExists(tableName)) {
-            dropTable(stmt, tableName);
-        }
-    }
-
-    @Override
-    public void createTableIfNotExists(String tableName, JdbcSchema schema) throws SQLException
-    {
-        if (!tableExists(tableName)) {
-            createTable(tableName, schema);
-        }
-    }
-
-    private static String getSchema(Connection connection) throws SQLException
-    {
-        // Because old Oracle JDBC drivers don't support Connection#getSchema method.
-        String sql = "SELECT SYS_CONTEXT('USERENV', 'CURRENT_SCHEMA') FROM DUAL";
-        try (Statement statement = connection.createStatement()) {
-            try (ResultSet resultSet = statement.executeQuery(sql)) {
-                if (resultSet.next()) {
-                    return resultSet.getString(1);
-                }
-                throw new SQLException(String.format("Cannot get schema becase \"%s\" didn't return any value.", sql));
-            }
-        }
-    }
-
-    @Override
-    protected String buildPrepareInsertSql(String toTable, JdbcSchema toTableSchema) throws SQLException
-    {
-        String sql = super.buildPrepareInsertSql(toTable, toTableSchema);
-        if (direct) {
-            sql = sql.replaceAll("^INSERT ", "INSERT /*+ APPEND_VALUES */ ");
-        }
-        return sql;
-    }
-
-    public OracleCharset getCharset() throws SQLException
-    {
-        String charsetName = "UTF8";
-        try (Statement statement = connection.createStatement()) {
-            try (ResultSet resultSet = statement.executeQuery("SELECT VALUE FROM NLS_DATABASE_PARAMETERS WHERE PARAMETER='NLS_CHARACTERSET'")) {
-                if (resultSet.next()) {
-                    String nlsCharacterSet = resultSet.getString(1);
-                    if (CHARSET_NAMES.containsKey(nlsCharacterSet)) {
-                        charsetName = nlsCharacterSet;
-                    }
-                }
-            }
-        }
-
-        try (PreparedStatement statement = connection.prepareStatement("SELECT NLS_CHARSET_ID(?) FROM DUAL")) {
-            statement.setString(1, charsetName);
-            try (ResultSet resultSet = statement.executeQuery()) {
-                if (!resultSet.next()) {
-                    throw new SQLException("Unknown NLS_CHARACTERSET : " + charsetName);
-                }
-
-                return new OracleCharset(charsetName,
-                        resultSet.getShort(1),
-                        Charset.forName(CHARSET_NAMES.get(charsetName)));
-            }
-        }
-    }
-
-}
+package org.embulk.output.oracle;
+
+import java.nio.charset.Charset;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.HashMap;
+import java.util.Map;
+
+import org.embulk.output.jdbc.JdbcOutputConnection;
+import org.embulk.output.jdbc.JdbcColumn;
+import org.embulk.output.jdbc.JdbcSchema;
+
+public class OracleOutputConnection
+        extends JdbcOutputConnection
+{
+    private static final Map<String, String> CHARSET_NAMES = new HashMap<String, String>();
+    static {
+        CHARSET_NAMES.put("JA16SJIS", "Shift_JIS");
+        CHARSET_NAMES.put("JA16SJISTILDE", "Shift_JIS");
+        CHARSET_NAMES.put("JA16EUC", "EUC-JP");
+        CHARSET_NAMES.put("JA16EUCTILDE", "EUC-JP");
+        CHARSET_NAMES.put("AL32UTF8", "UTF-8");
+        CHARSET_NAMES.put("UTF8", "UTF-8");
+        CHARSET_NAMES.put("AL16UTF16", "UTF-16");
+    }
+
+    private final boolean direct;
+    private OracleCharset charset;
+
+    public OracleOutputConnection(Connection connection, boolean autoCommit, boolean direct)
+            throws SQLException
+    {
+        super(connection, getSchema(connection));
+        connection.setAutoCommit(autoCommit);
+
+        this.direct = direct;
+    }
+
+    @Override
+    protected String buildColumnTypeName(JdbcColumn c)
+    {
+        switch(c.getSimpleTypeName()) {
+        case "BIGINT":
+            return "NUMBER(19,0)";
+        default:
+            return super.buildColumnTypeName(c);
+        }
+    }
+
+    @Override
+    protected void setSearchPath(String schema) throws SQLException {
+        // NOP
+    }
+
+    @Override
+    public void dropTableIfExists(String tableName) throws SQLException
+    {
+        if (tableExists(tableName)) {
+            dropTable(tableName);
+        }
+    }
+
+    @Override
+    protected void dropTableIfExists(Statement stmt, String tableName) throws SQLException {
+        if (tableExists(tableName)) {
+            dropTable(stmt, tableName);
+        }
+    }
+
+    @Override
+    public void createTableIfNotExists(String tableName, JdbcSchema schema) throws SQLException
+    {
+        if (!tableExists(tableName)) {
+            createTable(tableName, schema);
+        }
+    }
+
+    public void createTable(String tableName, JdbcSchema schema) throws SQLException
+    {
+        Statement stmt = connection.createStatement();
+        try {
+            String sql = buildCreateTableSql(tableName, schema);
+            executeUpdate(stmt, sql);
+            commitIfNecessary(connection);
+        } catch (SQLException ex) {
+            throw safeRollback(connection, ex);
+        } finally {
+            stmt.close();
+        }
+    }
+
+    protected String buildCreateTableSql(String name, JdbcSchema schema)
+    {
+        StringBuilder sb = new StringBuilder();
+
+        sb.append("CREATE TABLE ");
+        quoteIdentifierString(sb, name);
+        sb.append(buildCreateTableSchemaSql(schema));
+        return sb.toString();
+    }
+
+    private static String getSchema(Connection connection) throws SQLException
+    {
+        // Because old Oracle JDBC drivers don't support Connection#getSchema method.
+        String sql = "SELECT SYS_CONTEXT('USERENV', 'CURRENT_SCHEMA') FROM DUAL";
+        try (Statement statement = connection.createStatement()) {
+            try (ResultSet resultSet = statement.executeQuery(sql)) {
+                if (resultSet.next()) {
+                    return resultSet.getString(1);
+                }
+                throw new SQLException(String.format("Cannot get schema becase \"%s\" didn't return any value.", sql));
+            }
+        }
+    }
+
+    @Override
+    protected String buildPreparedInsertSql(String toTable, JdbcSchema toTableSchema) throws SQLException
+    {
+        String sql = super.buildPreparedInsertSql(toTable, toTableSchema);
+        if (direct) {
+            sql = sql.replaceAll("^INSERT ", "INSERT /*+ APPEND_VALUES */ ");
+        }
+        return sql;
+    }
+
+    @Override
+    public Charset getTableNameCharset() throws SQLException
+    {
+        return getOracleCharset().getJavaCharset();
+    }
+
+    public synchronized OracleCharset getOracleCharset() throws SQLException
+    {
+        if (charset == null) {
+            String charsetName = "UTF8";
+            try (Statement statement = connection.createStatement()) {
+                try (ResultSet resultSet = statement.executeQuery("SELECT VALUE FROM NLS_DATABASE_PARAMETERS WHERE PARAMETER='NLS_CHARACTERSET'")) {
+                    if (resultSet.next()) {
+                        String nlsCharacterSet = resultSet.getString(1);
+                        if (CHARSET_NAMES.containsKey(nlsCharacterSet)) {
+                            charsetName = nlsCharacterSet;
+                        }
+                    }
+                }
+            }
+
+            try (PreparedStatement statement = connection.prepareStatement("SELECT NLS_CHARSET_ID(?) FROM DUAL")) {
+                statement.setString(1, charsetName);
+                try (ResultSet resultSet = statement.executeQuery()) {
+                    if (!resultSet.next()) {
+                        throw new SQLException("Unknown NLS_CHARACTERSET : " + charsetName);
+                    }
+
+                    charset = new OracleCharset(charsetName,
+                            resultSet.getShort(1),
+                            Charset.forName(CHARSET_NAMES.get(charsetName)));
+                }
+            }
+        }
+        return charset;
+    }
+}

data/src/main/java/org/embulk/output/oracle/OracleOutputConnector.java

@@ -1,49 +1,49 @@
-package org.embulk.output.oracle;
-
-import java.sql.Connection;
-import java.sql.DriverManager;
-import java.sql.SQLException;
-import java.util.Properties;
-
-import org.embulk.output.jdbc.JdbcOutputConnector;
-
-public class OracleOutputConnector
-        implements JdbcOutputConnector
-{
-    private final String url;
-    private final Properties properties;
-    private final boolean direct;
-
-    public OracleOutputConnector(String url, Properties properties, boolean direct)
-    {
-        try {
-            Class.forName("oracle.jdbc.OracleDriver");
-        } catch (Exception ex) {
-            throw new RuntimeException(ex);
-        }
-        this.url = url;
-        this.properties = properties;
-        this.direct = direct;
-    }
-
-    @Override
-    public OracleOutputConnection connect(boolean autoCommit) throws SQLException
-    {
-        Connection c = DriverManager.getConnection(url, properties);
-        if (c == null) {
-            // driver.connect returns null when url is "jdbc:mysql://...".
-            throw new SQLException("Invalid url : " + url);
-        }
-
-        try {
-            OracleOutputConnection con = new OracleOutputConnection(c, autoCommit, direct);
-            c = null;
-            return con;
-
-        } finally {
-            if (c != null) {
-                c.close();
-            }
-        }
-    }
-}
+package org.embulk.output.oracle;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+import java.util.Properties;
+
+import org.embulk.output.jdbc.JdbcOutputConnector;
+
+public class OracleOutputConnector
+        implements JdbcOutputConnector
+{
+    private final String url;
+    private final Properties properties;
+    private final boolean direct;
+
+    public OracleOutputConnector(String url, Properties properties, boolean direct)
+    {
+        try {
+            Class.forName("oracle.jdbc.OracleDriver");
+        } catch (Exception ex) {
+            throw new RuntimeException(ex);
+        }
+        this.url = url;
+        this.properties = properties;
+        this.direct = direct;
+    }
+
+    @Override
+    public OracleOutputConnection connect(boolean autoCommit) throws SQLException
+    {
+        Connection c = DriverManager.getConnection(url, properties);
+        if (c == null) {
+            // driver.connect returns null when url is "jdbc:mysql://...".
+            throw new SQLException("Invalid url : " + url);
+        }
+
+        try {
+            OracleOutputConnection con = new OracleOutputConnection(c, autoCommit, direct);
+            c = null;
+            return con;
+
+        } finally {
+            if (c != null) {
+                c.close();
+            }
+        }
+    }
+}

data/src/main/java/org/embulk/output/oracle/TimestampFormat.java

@@ -1,37 +1,37 @@
-package org.embulk.output.oracle;
-
-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;
-    }
-
-}
+package org.embulk.output.oracle;
+
+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/oracle/oci/ColumnDefinition.java

@@ -1,26 +1,26 @@
-package org.embulk.output.oracle.oci;
-
-public class ColumnDefinition
-{
-    public static int SQLT_CHR = 1;
-    public static int SQLT_INT = 3;
-
-    public final String columnName;
-    public final int columnType;
-    public final int columnSize;
-    public final String columnDateFormat;
-
-
-    public ColumnDefinition(String columnName, int columnType, int columnSize, String columnDateFormat)
-    {
-        this.columnName = columnName;
-        this.columnType = columnType;
-        this.columnSize = columnSize;
-        this.columnDateFormat = columnDateFormat;
-    }
-
-    public ColumnDefinition(String columnName, int columnType, int columnSize)
-    {
-        this(columnName, columnType, columnSize, null);
-    }
-}
+package org.embulk.output.oracle.oci;
+
+public class ColumnDefinition
+{
+    public static int SQLT_CHR = 1;
+    public static int SQLT_INT = 3;
+
+    public final String columnName;
+    public final int columnType;
+    public final int columnSize;
+    public final String columnDateFormat;
+
+
+    public ColumnDefinition(String columnName, int columnType, int columnSize, String columnDateFormat)
+    {
+        this.columnName = columnName;
+        this.columnType = columnType;
+        this.columnSize = columnSize;
+        this.columnDateFormat = columnDateFormat;
+    }
+
+    public ColumnDefinition(String columnName, int columnType, int columnSize)
+    {
+        this(columnName, columnType, columnSize, null);
+    }
+}