org.skife.jdbi.v2.exceptions.UnableToExecuteStatementException Java Examples

The following examples show how to use org.skife.jdbi.v2.exceptions.UnableToExecuteStatementException. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source File: TestShardDao.java    From presto with Apache License 2.0 6 votes vote down vote up 
@Test
public void testExternalBatches()
{
    assertFalse(dao.externalBatchExists("foo"));
    assertFalse(dao.externalBatchExists("bar"));

    dao.insertExternalBatch("foo");

    assertTrue(dao.externalBatchExists("foo"));
    assertFalse(dao.externalBatchExists("bar"));

    try {
        dao.insertExternalBatch("foo");
        fail("expected exception");
    }
    catch (UnableToExecuteStatementException e) {
        assertInstanceOf(e.getCause(), SQLException.class);
        assertTrue(((SQLException) e.getCause()).getSQLState().startsWith("23"));
    }
}
Example #2
Source File: BasicDatabaseStoreManager.java    From digdag with Apache License 2.0 6 votes vote down vote up 
public <T> T catchConflict(NewResourceAction<T> function,
        String messageFormat, Object... messageParameters)
        throws ResourceConflictException
{
    try {
        return function.call();
    }
    catch (UnableToExecuteStatementException ex) {
        if (ex.getCause() instanceof SQLException) {
            SQLException sqlEx = (SQLException) ex.getCause();
            if (isConflictException(sqlEx)) {
                throw new ResourceConflictException("Resource already exists: " + String.format(messageFormat, messageParameters));
            }
        }
        throw ex;
    }
}
Example #3
Source File: BasicDatabaseStoreManager.java    From digdag with Apache License 2.0 6 votes vote down vote up 
public <T> T catchForeignKeyNotFound(NewResourceAction<T> function,
        String messageFormat, Object... messageParameters)
        throws ResourceNotFoundException, ResourceConflictException
{
    try {
        return function.call();
    }
    catch (UnableToExecuteStatementException ex) {
        if (ex.getCause() instanceof SQLException) {
            SQLException sqlEx = (SQLException) ex.getCause();
            if (isForeignKeyException(sqlEx)) {
                throw new ResourceNotFoundException("Resource not found: " + String.format(messageFormat, messageParameters));
            }
        }
        throw ex;
    }
}
Example #4
Source File: PostgresStorage.java    From cassandra-reaper with Apache License 2.0 5 votes vote down vote up 
private boolean takeLead(UUID leaderId, Duration ttl) {
  if (null != jdbi) {
    try (Handle h = jdbi.open()) {
      try {
        int rowsInserted = getPostgresStorage(h).insertLeaderEntry(
            leaderId,
            reaperInstanceId,
            AppContext.REAPER_INSTANCE_ADDRESS
        );
        if (rowsInserted == 1) {  // insert should modify exactly 1 row
          return true;
        }
      } catch (UnableToExecuteStatementException e) {
        if (JdbiExceptionUtil.isDuplicateKeyError(e)) {
          // if it's a duplicate key error, then try to update it
          int rowsUpdated = getPostgresStorage(h).updateLeaderEntry(
              leaderId,
              reaperInstanceId,
              AppContext.REAPER_INSTANCE_ADDRESS,
              getExpirationTime(ttl)
          );
          if (rowsUpdated == 1) {  // if row updated, took ownership from an expired leader
            LOG.debug("Took lead from expired entry for segment {}", leaderId);
            return true;
          }
        }
        return false;
      }
    }
  }
  LOG.warn("Unknown error occurred while taking lead on segment {}", leaderId);
  return false;
}
Example #5
Source File: SqlExtractorTest.java    From pocket-etl with Apache License 2.0 4 votes vote down vote up 
@Test(expected = UnableToExecuteStatementException.class)
public void openThrowsUnableToExecuteStatementExceptionIfQueryThrowsSqlException() throws Exception {
    when(mockPreparedStatement.execute()).thenThrow(new SQLException());
    getBasicDTOSqlExtractor();
}
Example #6
Source File: JdbiExceptionUtil.java    From cassandra-reaper with Apache License 2.0 4 votes vote down vote up 
public static boolean isDuplicateKeyError(UnableToExecuteStatementException exc) {
  if (exc.getCause() instanceof PSQLException) {
    return ((PSQLException) exc.getCause()).getSQLState().equals(DUPLICATE_KEY_CODE);
  }
  return false;
}