Hibernate 6.1中如何获取新创建对象的ID？

问题原因 #

Hibernate 5.x的save()方法会隐式触发数据库插入操作，因此能立即拿到生成的主键；而遵循JPA规范的persist()方法默认延迟执行插入（直到事务提交或手动触发flush），这就导致调用persist()后直接读取id会是空值。

解决方案 #

以下是几种可靠的实现方式：

• 手动触发Flush
  调用persist()后，主动调用EntityManager.flush()强制Hibernate执行数据库插入，此时主键会自动赋值到实体的id字段中：

  // 创建实体实例
  MyEntity entity = new MyEntity();
  entity.setName("测试数据"); // 设置其他业务字段
  
  // 持久化实体
  entityManager.persist(entity);
  // 手动触发flush，执行插入并生成主键
  entityManager.flush();
  
  // 现在可以获取到生成的主键
  int generatedId = entity.getId();
  

• 利用事务提交自动触发Flush
  如果代码运行在事务上下文（比如标注@Transactional注解），事务提交时Hibernate会自动执行flush操作，此时直接读取id即可：

  @Transactional
  public int saveEntity() {
      MyEntity entity = new MyEntity();
      entity.setName("测试数据");
      entityManager.persist(entity);
      // 事务提交时自动flush，直接返回主键
      return entity.getId();
  }
  

• 使用merge方法（仅适合特定场景）
  merge()方法会返回包含主键的实体实例，但它更适配分离状态的实体，对于新创建的实体，效率不如persist()+flush：

  MyEntity entity = new MyEntity();
  entity.setName("测试数据");
  // merge执行插入并返回带主键的实体
  MyEntity persistedEntity = entityManager.merge(entity);
  int generatedId = persistedEntity.getId();
  

注意事项 #

IDENTITY策略依赖数据库自增主键，必须等插入操作执行后才能获取主键值，所以所有方案的核心都是确保插入操作被触发执行。

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