通过JDBC从AWS Redshift检索插入的标识值

时间:2013-10-17 14:01:07

标签: java postgresql jdbc insert amazon-redshift

使用AWS Redshift时,我注意从带有标识列的表中获取最后插入的id 无法使用以下任一方法完成JDBC驱动程序:

RETURNING key word

Statement.RETURN_GENERATED_KEYS 

如Stack Overflow条目中所述:

How to get a value from the last inserted row?

上述方法不可用,因为Redshift(截至2013年10月17日)是基于PostgreSQL版本8.0.2构建的。请参阅以下内容 以下链接中的文档:

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

如果您打算将Redshift用作RDBMS,那么阅读以下内容也是值得的:

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

问题:

在Redshift中检索自动增量/序列/标识列上的最后一个插入ID的最佳策略是什么? PostgreSQL JDBC驱动程序?

1 个答案:

答案 0 :(得分:4)

鉴于Redshift引擎是基于PostgreSQL 8.0.2构建的,上面的RETURNING和Statement.RETURN_GENERATED_KEYS选项不是 可用AND Redshift不支持CREATE SEQUENCE使用CURRVAL / NEXTVAL功能套件,一个选项是 将两个SQL语句组合在一起,在JDBC事务中将INSERT和SELECT MAX([标识列])组合在一起。

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中返回。