Извлечь введенное значение идентификатора из AWS Redshift через JDBC

Во время работы с AWS Redshift мне стало полезно получить последний вставленный идентификатор из таблицы с столбцом идентификатора через Драйвер JDBC нельзя выполнить одним из следующих способов:

RETURNING key word

или

Statement.RETURN_GENERATED_KEYS 

как указано в записи:

Как получить значение из последней вставленной строки?

Вышеуказанные методы недоступны, так как Redshift (по состоянию на 10/17/2013) построен на PostgreSQL версии 8.0.2. См. Следующие документацию по следующей ссылке:

http://docs.aws.amazon.com/redshift/latest/dg/c_high_level_system_architecture.html

Если вы намерены использовать Redshift как СУБД, стоит также прочитать следующее:

http://docs.aws.amazon.com/redshift/latest/dg/c_redshift-and-postgres-sql.html

Вопрос:

Какова наилучшая стратегия для получения последнего вставленного идентификатора в столбце автоинкремент/серийный/идентификационный номер в Redshift через Драйвер PostgreSQL JDBC?

Ответ 1

Учитывая, что механизм Redshift построен на PostgreSQL 8.0.2, а вышеперечисленные опции RETURNING и Statement.RETURN_GENERATED_KEYS не являются доступно AND Redshift не поддерживает CREATE SEQUENCE для использования набора функций CURRVAL/NEXTVAL, один из вариантов - группировать два оператора SQL вместе, INSERT и SELECT MAX ([столбец идентификации]) в транзакции JDBC.

try {

    // create the JDBC connection
    Class.forName(JDBC_DRIVER);
    Connection conn = DriverManager.getConnection(DB_URL, USER, PASSWORD);

    // start the transaction
    conn.setAutoCommit(false);  
    // create the prepared statement for insert
    PreparedStatement prpd = conn.prepareStatement(SQL_INSERT_STATEMENT);  

    // set input/output parameters as needed...

    // execute the SQL prepared statement
    int j = prpd.executeUpdate();

    // create a statement for select max()
    Statement stmt = conn.createStatement(); 
    // execute the statement to return a result set           
    ResultSet key = stmt.executeQuery(SQL_SELECT_MAX_STATEMENT);

    // initialize and retrieve the incremented identity value, in this case it is a long (bigint data type in Redshift)
    long id = 0;
    if (key.next()) {
        id = key.getLong(1);
    }

    // commit the entire transaction        
    conn.commit();

} catch (SQLException se) {
    // if an SQL exception occurs, rollback the whole deal
    try {
        if (conn!=null && !conn.isClosed()) {
            conn.rollback();
        }

    } catch (Exception e) {

    }
} catch (Exception e) {
    // roll back if something other than an SQLException occurs
    try {
        if (conn!=null && !conn.isClosed()) {
            conn.rollback();
        }
    } catch (Exception e) {

    }
} finally {
    // do whatever you want to return a value, shut down resources
    // close out JDBC resources
    try {
        if (conn!=null && !conn.isClosed()) {
            conn.setAutoCommit(true);
        }                
    } catch (SQLException se) {

    }

    try {                
        if (prpd!=null && !prpd.isClosed()) {
            prpd.close();
        }                                
    } catch (SQLException se) {

    }

    try {                
        if (stmt!=null && !stmt.isClosed()) {
            stmt.close();
        }                                
    } catch (SQLException se) {

    }

    try {
        if (conn!=null && !conn.isClosed()) {
            conn.close();
        }
    } catch (SQLException se) {

    }
}

Вышеуказанное будет работать, если SQL_INSERT_STATEMENT пишет/блокирует одну таблицу. Множественные блокировки таблиц потребуют синхронизации ключевое слово для защиты от тупика. Выбор в заблокированной таблице позволит увеличить значение идентификатора возвращается в ResultSet.