事务内插入多条记录后提交前获取H2DB自增主键ID值

嘿，这个需求我之前在项目里刚好碰到过，H2DB提供了几种简单有效的方式，能让你在事务提交前就拿到每条插入记录的自增ID，分场景给你拆解下：

1. 用RETURNING子句（最简洁的H2专属方式） #

这是H2原生支持的语法，插入语句末尾加上RETURNING 你的主键字段名，执行后直接返回所有生成的ID，不管是单条还是批量插入都适用，非常直观。

举个SQL例子：

-- 假设你的表叫prompt，主键是ID，还有content字段
INSERT INTO prompt (content) 
VALUES ('第一条提示内容'), ('第二条提示内容'), ('第三条提示内容')
RETURNING ID;

执行这条语句后，结果集里会直接列出三条记录对应的自增ID，事务提交前就能拿到。如果用JDBC的话，记得用executeQuery()来执行，然后遍历ResultSet提取ID就行。

2. 用IDENTITY()函数（适合循环单条插入的场景） #

如果你是循环插入单条记录，每次插入完成后立刻执行SELECT IDENTITY();，就能拿到刚插入那条记录的ID。比如：

INSERT INTO prompt (content) VALUES ('单条插入的提示');
SELECT IDENTITY(); -- 这条语句返回刚才生成的ID

注意这个函数只会返回最后一次插入生成的ID，所以循环插的时候要在每次插入后马上调用，事务提交前完全可以正常获取。

3. JDBC标准的getGeneratedKeys()（跨数据库兼容） #

如果你的代码需要兼顾其他数据库，不想用H2专属语法，那JDBC的标准方法就很合适。在创建PreparedStatement的时候，指定Statement.RETURN_GENERATED_KEYS参数，批量插入后调用getGeneratedKeys()就能拿到所有生成的ID。

给个Java代码示例：

// 假设conn是已经开启事务的数据库连接
String insertSql = "INSERT INTO prompt (content) VALUES (?)";
try (PreparedStatement pstmt = conn.prepareStatement(insertSql, Statement.RETURN_GENERATED_KEYS)) {
    // 添加第一条记录
    pstmt.setString(1, "JDBC批量插入第一条");
    pstmt.addBatch();
    // 添加第二条记录
    pstmt.setString(1, "JDBC批量插入第二条");
    pstmt.addBatch();
    
    // 执行批量插入
    pstmt.executeBatch();
    
    // 获取所有生成的自增ID
    try (ResultSet rs = pstmt.getGeneratedKeys()) {
        while (rs.next()) {
            long generatedId = rs.getLong(1);
            // 这里可以把ID存起来用，事务提交前就能拿到
            System.out.println("生成的ID: " + generatedId);
        }
    }
} catch (SQLException e) {
    // 异常处理
    e.printStackTrace();
}

额外提醒 #

• 不管用哪种方式，自增ID都是在插入执行成功时就生成的，和事务提交与否无关，所以完全可以在事务提交前获取并使用这些ID。
• 确保你的主键字段确实是自增类型，比如表定义的时候写ID BIGINT AUTO_INCREMENT PRIMARY KEY或者H2特有的ID INT IDENTITY。

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