批量插入多行时如何获取所有生成ID？现有代码仅返回最后一个ID

Hey，刚好之前处理过类似的场景！你现在的代码只能拿到最后一个生成ID，主要是两个小问题：一是没用到JDBC的批量插入API，二是循环内的变量声明覆盖了外部变量，导致没法保存多个ID。下面给你整理出批量插入时获取所有生成ID的正确写法：

首先修正原代码的核心问题，然后给出完整示例：

// 1. 编写带占位符的插入SQL（替换成你的表和字段）
String sql = "INSERT INTO your_table (column1, column2) VALUES (?, ?)";

// 2. 初始化PreparedStatement时，指定RETURN_GENERATED_KEYS
PreparedStatement ps = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);

// 3. 批量添加插入参数
// 第一行数据
ps.setString(1, "sample_value_1");
ps.setString(2, "sample_value_2");
ps.addBatch(); // 将这组参数加入批量队列

// 第二行数据
ps.setString(1, "sample_value_3");
ps.setString(2, "sample_value_4");
ps.addBatch();

// 可以继续添加更多行...

// 4. 执行批量插入
int[] affectedRowCounts = ps.executeBatch();
// 这个数组里的每个元素对应每一批插入操作的受影响行数，可以用来验证插入是否成功

// 5. 获取所有生成的主键ID
ResultSet rs = ps.getGeneratedKeys();
List<Integer> allGeneratedIds = new ArrayList<>();

while (rs.next()) {
    // 注意：不要在这里重新声明int generatedKey，否则会覆盖外部变量
    int generatedId = rs.getInt(1);
    allGeneratedIds.add(generatedId);
    System.out.println("插入记录的ID: " + generatedId);
}

// 后续可以直接使用allGeneratedIds集合处理所有生成的ID
System.out.println("所有生成的ID列表: " + allGeneratedIds);

关键要点说明 #

• 使用批量插入API：必须用addBatch()添加参数，再用executeBatch()执行批量操作，而不是原代码的execute()，后者只能执行单条插入
• 用集合存储多个ID：单个变量只能保存最后一个ID，用List集合可以轻松存储所有生成的主键
• 避免变量覆盖：原代码中while循环里重新声明了int generatedKey，导致外部的变量始终是初始值0，这是个容易踩的小坑
• 驱动兼容性：主流数据库驱动（比如MySQL Connector/J、PostgreSQL驱动）都支持这种方式返回批量插入的所有生成ID，如果你用的是Oracle等特殊数据库，可能需要调整主键获取的语法，但大部分场景下上面的代码都能正常工作

内容的提问来源于stack exchange，提问作者Dama Ramesh