如何在JDBC中获取插入ID？

Z时代
2024-01-10
分类：问答

如何在JDBC中获取插入ID？

回答：

如果它是自动生成的密钥，那么你可以使用Statement#getGeneratedKeys()它。你需要Statement使用与用于相同的名称进行调用INSERT。首先，你需要创建Statement.RETURN_GENERATED_KEYS用于通知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.");
            }
        }
    }
}