SpringBoot+Hibernate测试ID负数：如何编程刷新SessionFactory/SpringContext？

问题根源 #

你的推测完全正确：Hibernate使用SEQUENCE策略时，会预分配一批ID（对应序列的增量值50）并缓存起来。当你通过Flyway重置数据库序列到初始值1后，Hibernate的缓存里还保留着之前预取的ID范围。此时如果Hibernate发现这些预取的ID已经被数据库视为已使用（或者序列当前值远低于缓存的ID），就会生成负数ID作为 fallback 策略。

不需要刷新SpringContext或重启EntityManagerFactory #

刷新整个SpringContext或重启EntityManagerFactory代价极高，会导致应用重启、测试速度骤降，完全没必要——我们只需要针对性重置Hibernate的序列生成器缓存即可。

具体实现方案 #

方案1：重置所有序列生成器的预分配缓存 #

在你的CleanDatabaseController中添加逻辑，直接重置Hibernate内部的序列生成器缓存：

@Autowired
private EntityManager entityManager;

// 在Flyway迁移、缓存清除后调用此方法
private void resetHibernateSequenceCaches() {
    Session session = entityManager.unwrap(Session.class);
    SessionFactory sessionFactory = session.getSessionFactory();
    Metamodel metamodel = sessionFactory.getMetamodel();

    // 遍历所有实体，处理使用SEQUENCE策略的ID生成器
    for (EntityType<?> entityType : metamodel.getEntities()) {
        IdentifierGenerator generator = sessionFactory.getIdentifierGenerator(entityType.getName());
        if (generator instanceof SequenceStyleGenerator) {
            SequenceStyleGenerator sequenceGenerator = (SequenceStyleGenerator) generator;
            // 针对PooledOptimizer（Hibernate默认的预分配优化器）重置缓存
            if (sequenceGenerator.getOptimizer() instanceof PooledOptimizer) {
                PooledOptimizer optimizer = (PooledOptimizer) sequenceGenerator.getOptimizer();
                try {
                    // 通过反射重置预分配缓存的起始值
                    Field lastValueField = PooledOptimizer.class.getDeclaredField("lastValue");
                    lastValueField.setAccessible(true);
                    lastValueField.set(optimizer, 0L);

                    Field hiValueField = PooledOptimizer.class.getDeclaredField("hiValue");
                    hiValueField.setAccessible(true);
                    hiValueField.set(optimizer, 0L);
                } catch (NoSuchFieldException | IllegalAccessException e) {
                    throw new RuntimeException("Failed to reset sequence optimizer cache", e);
                }
            }
        }
    }
}

方案2：同步Hibernate缓存与数据库序列当前值 #

如果需要更精准的控制，可以先查询数据库序列的当前值，再同步到Hibernate的缓存中：

private void syncHibernateSequenceWithDb() {
    Session session = entityManager.unwrap(Session.class);
    SessionFactory sessionFactory = session.getSessionFactory();
    Metamodel metamodel = sessionFactory.getMetamodel();

    for (EntityType<?> entityType : metamodel.getEntities()) {
        IdentifierGenerator generator = sessionFactory.getIdentifierGenerator(entityType.getName());
        if (generator instanceof SequenceStyleGenerator) {
            SequenceStyleGenerator sequenceGenerator = (SequenceStyleGenerator) generator;
            // 获取实体对应的序列名
            String sequenceName = sequenceGenerator.getDatabaseStructure().getName();
            
            // 查询数据库序列的当前值（nextval后减1，因为nextval会让序列自增一次）
            Long currentSeqValue = session.createNativeQuery(
                    String.format("SELECT nextval('%s') - 1", sequenceName), Long.class)
                .uniqueResult();

            // 重置缓存值
            if (sequenceGenerator.getOptimizer() instanceof PooledOptimizer) {
                PooledOptimizer optimizer = (PooledOptimizer) sequenceGenerator.getOptimizer();
                try {
                    Field lastValueField = PooledOptimizer.class.getDeclaredField("lastValue");
                    lastValueField.setAccessible(true);
                    lastValueField.set(optimizer, currentSeqValue);

                    Field hiValueField = PooledOptimizer.class.getDeclaredField("hiValue");
                    hiValueField.setAccessible(true);
                    hiValueField.set(optimizer, currentSeqValue);
                } catch (Exception e) {
                    throw new RuntimeException("Failed to sync sequence cache with database", e);
                }
            }
        }
    }
}

注意事项 #

• 反射操作依赖Hibernate的内部实现，不同Hibernate版本可能需要调整字段名（比如部分版本中hiValue可能叫upperBound），建议在测试环境验证后使用。
• 如果你的实体使用了自定义序列名，确保sequenceName的获取逻辑正确（SequenceStyleGenerator.getDatabaseStructure().getName()可以正确拿到数据库中的序列名）。
• 两种方案都需要在Flyway完成数据库迁移、缓存清除后立即调用，确保Hibernate的序列缓存与数据库状态一致。

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