JDBC中获取插入行的自增主键：getGeneratedKeys() 方法详解

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class GetGeneratedKeysExample {

    private static final String DB_URL = "jdbc:postgresql://localhost:5432/your_database";
    private static final String USER = "your_user";
    private static final String PASSWORD = "your_password";

    public static void main(String[] args) {
        Connection connection = null;
        PreparedStatement pstmt = null;
        ResultSet keys = null;

        try {
            // 1. 建立数据库连接
            connection = DriverManager.getConnection(DB_URL, USER, PASSWORD);
            connection.setAutoCommit(false); // 开启事务

            // 2. 准备插入语句，并指定返回生成的键
            // 假设表名为 'products'，主键为 'product_id' (SERIAL 类型)
            // 假设 'product_name' 是另一个列
            String sql = "INSERT INTO products(product_name) VALUES (?)";

            // 方法一：指定列名
            // pstmt = connection.prepareStatement(sql, new String[]{"product_id"});

            // 方法二：使用常量 (更通用)
            pstmt = connection.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);

            // 3. 设置参数并执行插入
            pstmt.setString(1, "Sample Product A");
            int numRowsAffected = pstmt.executeUpdate();

            System.out.println("受影响的行数: " + numRowsAffected);

            // 4. 获取生成的键
            keys = pstmt.getGeneratedKeys();
            int newProductId = -1; // 默认值，表示未获取到

            if (keys.next()) { // 检查是否有生成的键
                newProductId = keys.getInt(1); // 获取第一个（通常是唯一的）生成的键
                System.out.println("新插入的产品ID: " + newProductId);
            } else {
                System.out.println("未获取到生成的键。");
            }

            connection.commit(); // 提交事务

        } catch (SQLException e) {
            e.printStackTrace();
            if (connection != null) {
                try {
                    System.err.println("事务回滚...");
                    connection.rollback(); // 发生异常时回滚事务
                } catch (SQLException ex) {
                    ex.printStackTrace();
                }
            }
        } finally {
            // 5. 关闭资源
            try {
                if (keys != null) keys.close();
                if (pstmt != null) pstmt.close();
                if (connection != null) connection.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

CREATE TABLE products (
    product_id SERIAL PRIMARY KEY,
    product_name VARCHAR(255) NOT NULL
);

// ... (之前的连接和PreparedStatement准备) ...

// 假设是批量插入
// pstmt.addBatch(); 添加多个批次
// pstmt.executeBatch();

// 或者是一次性插入多行（如果数据库支持）
// String multiRowSql = "INSERT INTO products(product_name) VALUES (?), (?), (?)";
// pstmt = connection.prepareStatement(multiRowSql, PreparedStatement.RETURN_GENERATED_KEYS);
// pstmt.setString(1, "Product X");
// pstmt.setString(2, "Product Y");
// pstmt.setString(3, "Product Z");
// pstmt.executeUpdate();

ResultSet keys = pstmt.getGeneratedKeys();
System.out.println("获取到的所有生成键:");
while (keys.next()) {
    int generatedId = keys.getInt(1); // 获取当前行的生成键
    System.out.println(" - ID: " + generatedId);
}

// ... (资源关闭) ...