Examples with BatchUpdateException java.sql.BatchUpdateException used on opensource projects

Example 46 with BatchUpdateException

use of java.sql.BatchUpdateException in project requery by requery.

the class BatchUpdateOperation method evaluate.

@Override
public int[] evaluate(QueryElement<int[]> query) {
    int[] result = batchInStatement ? null : new int[length];
    try (Connection connection = configuration.getConnection()) {
        DefaultOutput generator = new DefaultOutput(configuration, query);
        String sql = generator.toSql();
        StatementListener listener = configuration.getStatementListener();
        try (PreparedStatement statement = prepare(sql, connection)) {
            for (int i = 0; i < length; i++) {
                E element = elements[i];
                parameterBinder.bindParameters(statement, element, null);
                if (batchInStatement) {
                    statement.addBatch();
                } else {
                    listener.beforeExecuteBatchUpdate(statement, sql);
                    result[i] = statement.executeUpdate();
                    listener.afterExecuteBatchUpdate(statement, result);
                    readGeneratedKeys(i, statement);
                }
            }
            if (batchInStatement) {
                listener.beforeExecuteBatchUpdate(statement, sql);
                result = statement.executeBatch();
                listener.afterExecuteBatchUpdate(statement, result);
                readGeneratedKeys(0, statement);
            }
        }
    } catch (BatchUpdateException e) {
        result = e.getUpdateCounts();
        if (result == null) {
            throw new PersistenceException(e);
        }
    } catch (SQLException e) {
        throw new PersistenceException(e);
    }
    return result;
}

Example 47 with BatchUpdateException

use of java.sql.BatchUpdateException in project spring-framework by spring-projects.

the class SQLErrorCodeSQLExceptionTranslatorTests method batchExceptionTranslation.

@Test
public void batchExceptionTranslation() {
    SQLExceptionTranslator sext = new SQLErrorCodeSQLExceptionTranslator(ERROR_CODES);
    SQLException badSqlEx = new SQLException("", "", 1);
    BatchUpdateException batchUpdateEx = new BatchUpdateException();
    batchUpdateEx.setNextException(badSqlEx);
    BadSqlGrammarException bsgex = (BadSqlGrammarException) sext.translate("task", "SQL", batchUpdateEx);
    assertThat(bsgex.getSql()).isEqualTo("SQL");
    assertThat((Object) bsgex.getSQLException()).isEqualTo(badSqlEx);
}

Example 48 with BatchUpdateException

use of java.sql.BatchUpdateException in project ignite by apache.

the class JdbcThinErrorsSelfTest method testBatchUpdateException.

/**
 * Test error code for the case when error is caused on batch execution.
 * @throws SQLException if failed.
 */
@Test
public void testBatchUpdateException() throws SQLException {
    try (final Connection conn = getConnection()) {
        try (Statement stmt = conn.createStatement()) {
            stmt.executeUpdate("CREATE TABLE test (id int primary key, val varchar)");
            stmt.addBatch("insert into test (id, val) values (1, 'val1')");
            stmt.addBatch("insert into test (id, val) values (2, 'val2')");
            stmt.addBatch("insert into test (id1, val1) values (3, 'val3')");
            stmt.executeBatch();
            fail("BatchUpdateException is expected");
        } catch (BatchUpdateException e) {
            assertEquals(3, e.getUpdateCounts().length);
            assertArrayEquals("", new int[] { 1, 1, Statement.EXECUTE_FAILED }, e.getUpdateCounts());
            assertEquals("42000", e.getSQLState());
            assertTrue("Unexpected error message: " + e.getMessage(), e.getMessage() != null && e.getMessage().contains("Failed to parse query. Column \"ID1\" not found"));
        }
    }
}

Example 49 with BatchUpdateException

use of java.sql.BatchUpdateException in project ignite by apache.

the class JdbcRequestHandler method executeBatchedQuery.

/**
 * Executes query and updates result counters.
 *
 * @param qry Query.
 * @param updCntsAcc Per query rows updates counter.
 * @param firstErr First error data - code and message.
 * @param cancel Hook for query cancellation.
 * @throws QueryCancelledException If query was cancelled during execution.
 */
@SuppressWarnings({ "ForLoopReplaceableByForEach" })
private void executeBatchedQuery(SqlFieldsQueryEx qry, List<Integer> updCntsAcc, IgniteBiTuple<Integer, String> firstErr, GridQueryCancel cancel) throws QueryCancelledException {
    try {
        if (cliCtx.isStream()) {
            List<Long> cnt = connCtx.kernalContext().query().streamBatchedUpdateQuery(qry.getSchema(), cliCtx, qry.getSql(), qry.batchedArguments(), connCtx.clientDescriptor());
            for (int i = 0; i < cnt.size(); i++) updCntsAcc.add(cnt.get(i).intValue());
            return;
        }
        List<FieldsQueryCursor<List<?>>> qryRes = connCtx.kernalContext().query().querySqlFields(null, qry, cliCtx, true, true, cancel);
        for (FieldsQueryCursor<List<?>> cur : qryRes) {
            if (cur instanceof BulkLoadContextCursor)
                throw new IgniteSQLException("COPY command cannot be executed in batch mode.");
            assert !((QueryCursorImpl) cur).isQuery();
            Iterator<List<?>> it = cur.iterator();
            if (it.hasNext()) {
                int val = ((Long) it.next().get(0)).intValue();
                updCntsAcc.add(val);
            }
        }
    } catch (Exception e) {
        int code;
        String msg;
        if (X.cause(e, QueryCancelledException.class) != null)
            throw new QueryCancelledException();
        else if (e instanceof IgniteSQLException) {
            BatchUpdateException batchCause = X.cause(e, BatchUpdateException.class);
            if (batchCause != null) {
                int[] updCntsOnErr = batchCause.getUpdateCounts();
                for (int i = 0; i < updCntsOnErr.length; i++) updCntsAcc.add(updCntsOnErr[i]);
                msg = batchCause.getMessage();
                code = batchCause.getErrorCode();
            } else {
                for (int i = 0; i < qry.batchedArguments().size(); i++) updCntsAcc.add(Statement.EXECUTE_FAILED);
                msg = e.getMessage();
                code = ((IgniteSQLException) e).statusCode();
            }
        } else {
            for (int i = 0; i < qry.batchedArguments().size(); i++) updCntsAcc.add(Statement.EXECUTE_FAILED);
            msg = e.getMessage();
            code = IgniteQueryErrorCode.UNKNOWN;
        }
        if (firstErr.isEmpty())
            firstErr.set(code, msg);
        else
            U.error(log, "Failed to execute batch query [qry=" + qry + ']', e);
    }
}

Example 50 with BatchUpdateException

use of java.sql.BatchUpdateException in project ignite by apache.

the class JdbcThinStatement method executeBatch.

/**
 * {@inheritDoc}
 */
@Override
public int[] executeBatch() throws SQLException {
    ensureNotClosed();
    closeResults();
    checkStatementBatchEmpty();
    if (conn.isStream()) {
        int[] res = new int[batchSize];
        batchSize = 0;
        return res;
    }
    if (F.isEmpty(batch))
        return new int[0];
    JdbcBatchExecuteRequest req = new JdbcBatchExecuteRequest(conn.getSchema(), batch, conn.getAutoCommit(), false);
    try {
        JdbcBatchExecuteResult res = conn.sendRequest(req, this, null).response();
        if (res.errorCode() != ClientListenerResponse.STATUS_SUCCESS) {
            throw new BatchUpdateException(res.errorMessage(), IgniteQueryErrorCode.codeToSqlState(res.errorCode()), res.errorCode(), res.updateCounts());
        }
        return res.updateCounts();
    } finally {
        batchSize = 0;
        batch = null;
    }
}