在 Java 中使用 JDBC 获取插入 ID：深入指南

java

2024-03-12 14:13:36

使用 JDBC 获取插入 ID

简介

在使用 Java 中的 JDBC 向数据库插入记录时，获取新插入记录的 ID 至关重要。这对于跟踪记录和管理数据库中的数据非常有用。本文将深入探讨如何使用 JDBC API 检索插入 ID。

JDBC 插入 ID

当插入新记录到数据库时，数据库会自动生成一个唯一的 ID。这个 ID 通常称为“插入 ID”，它唯一标识新插入的记录。

JDBC 提供了 getGeneratedKeys() 方法，允许你检索由最近的 INSERT 语句生成的键。

获取插入 ID 的步骤

1. 创建 PreparedStatement

使用 PreparedStatement 来执行插入操作，因为它允许设置参数并检索插入 ID。

PreparedStatement statement = connection.prepareStatement("INSERT INTO table_name (column1, column2) VALUES (?, ?)", Statement.RETURN_GENERATED_KEYS);

2. 设置参数

使用 set 方法设置插入语句中的参数。

statement.setString(1, "value1");
statement.setInt(2, 10);

3. 执行更新

使用 executeUpdate() 方法执行插入语句。

int rowCount = statement.executeUpdate();

4. 获取插入 ID

使用 getGeneratedKeys() 方法获取插入 ID。

ResultSet generatedKeys = statement.getGeneratedKeys();
if (generatedKeys.next()) {
    long insertId = generatedKeys.getLong(1);
}

5. 关闭资源

使用完后关闭 Statement 和 ResultSet 对象。

statement.close();
generatedKeys.close();

示例代码

import java.sql.*;

public class GetInsertIdExample {

    public static void main(String[] args) {
        // Database connection details
        String url = "jdbc:sqlserver://localhost:1433;databaseName=testdb";
        String username = "sa";
        String password = "password";

        try (Connection connection = DriverManager.getConnection(url, username, password)) {

            // Create a PreparedStatement for inserting a record
            PreparedStatement statement = connection.prepareStatement("INSERT INTO users (name, email) VALUES (?, ?)", Statement.RETURN_GENERATED_KEYS);

            // Set the parameters
            statement.setString(1, "John Doe");
            statement.setString(2, "john.doe@example.com");

            // Execute the insert statement
            int rowCount = statement.executeUpdate();

            // Check if the insert was successful
            if (rowCount > 0) {

                // Get the generated key (insert ID)
                ResultSet generatedKeys = statement.getGeneratedKeys();
                if (generatedKeys.next()) {
                    long insertId = generatedKeys.getLong(1);
                    System.out.println("Inserted record with ID: " + insertId);
                }
            }

        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}