如何在JDBC中获取插入ID？

Question

我想用Java中的JDBC在数据库中插入一条记录（在我的例子中是Microsoft SQL Server）。同时，我想获得insert ID。如何使用JDBC API实现这一点？

Answer 1

我正在从基于单线程JDBC的应用程序中访问Microsoft SQL Server 2008 R2，并在不使用RETURN_GENERATED_KEYS属性或任何PreparedStatement的情况下提取最后一个ID。看起来像这样：

private int insertQueryReturnInt(String SQLQy) {
    ResultSet generatedKeys = null;
    int generatedKey = -1;

    try {
        Statement statement = conn.createStatement();
        statement.execute(SQLQy);
    } catch (Exception e) {
        errorDescription = "Failed to insert SQL query: " + SQLQy + "( " + e.toString() + ")";
        return -1;
    }

    try {
        generatedKey = Integer.parseInt(readOneValue("SELECT @@IDENTITY"));
    } catch (Exception e) {
        errorDescription = "Failed to get ID of just-inserted SQL query: " + SQLQy + "( " + e.toString() + ")";
        return -1;
    }

    return generatedKey;
} 

这篇博文很好地隔离了三个主要的SQL Server“last ID”选项：http://msjawahar.wordpress.com/2008/01/25/how-to-find-the-last-identity-value-inserted-in-the-sql-server/-还不需要另外两个。

Answer 2

>

创建生成的列

String generatedColumns[] = { "ID" };

把这个基因专栏交给你的陈述

PreparedStatement stmtInsert = conn.prepareStatement(insertSQL, generatedColumns);

使用ResultSet对象获取GeneratedKeys on语句

ResultSet rs = stmtInsert.getGeneratedKeys();

if (rs.next()) {
    long id = rs.getLong(1);
    System.out.println("Inserted ID -" + id); // display inserted record
}

Answer 3

如果它是自动生成的密钥，那么可以使用语句#getGeneratedKeys（）进行此操作。您需要在与用于插入的语句相同的语句上调用它。首先需要使用语句创建语句。RETURN_生成的_密钥通知JDBC驱动程序返回密钥。

以下是一个基本示例：

public void create(User user) throws SQLException {
    try (
        Connection connection = dataSource.getConnection();
        PreparedStatement statement = connection.prepareStatement(SQL_INSERT,
                                      Statement.RETURN_GENERATED_KEYS);
    ) {
        statement.setString(1, user.getName());
        statement.setString(2, user.getPassword());
        statement.setString(3, user.getEmail());
        // ...

        int affectedRows = statement.executeUpdate();

        if (affectedRows == 0) {
            throw new SQLException("Creating user failed, no rows affected.");
        }

        try (ResultSet generatedKeys = statement.getGeneratedKeys()) {
            if (generatedKeys.next()) {
                user.setId(generatedKeys.getLong(1));
            }
            else {
                throw new SQLException("Creating user failed, no ID obtained.");
            }
        }
    }
}

请注意，JDBC驱动程序是否工作取决于它。目前，大多数最新版本都可以工作，但如果我没有记错的话，Oracle JDBC驱动程序仍然有点麻烦。MySQL和DB2已经支持它很多年了。PostgreSQL不久前开始支持它。我无法评论MSSQL，因为我从未使用过它。

对于Oracle，您可以在同一事务中的INSERT之后直接调用CallableStatement，使用RETURNING子句或SELECT CURRVAL（sequencename）（或任何特定于数据库的语法）来获取最后生成的密钥。请参见此答案。