Spring Data JPA+H2测试环境下无法重置实体主键序列的问题求助

我太懂这种测试时主键序列乱跳的烦躁了！你在Spring Data JPA+H2的测试环境里想每次测试前清表、让主键从1开始，但那个用了@GeneratedValue(strategy = GenerationType.AUTO)的Long类型ID实体，怎么都搞不定，试了两种方法都没达到预期，对吧？

先帮你梳理下问题根源：当你用GenerationType.AUTO时，H2默认会用**独立的序列（Sequence）**来生成主键，而不是IDENTITY自增列——这就导致你用TRUNCATE TABLE ... RESTART IDENTITY只能重置表的自增标识，却碰不到独立的序列；而你之前试的ALTER SEQUENCE可能找错了序列名，或者操作顺序不对。

先说说你之前踩的坑：

• 第一种删主键再重建的方法完全没必要，反而会破坏表结构，自然起不到重置ID的作用
• 第二种ALTER SEQUENCE TABLE_NAME_SEQ RESTART WITH 1得到高ID，大概率是因为你找错了序列名！Spring Data JPA（准确说是Hibernate）在H2里用GenerationType.AUTO时，默认会用全局的HIBERNATE_SEQUENCE，而不是你以为的TABLE_NAME_SEQ，所以你改的是个根本没在使用的序列，当然没用。

接下来给你几个亲测有效的解决方案，按推荐程度排序：

方案一：找对序列名，先清表再重置序列 #

1. 先确认实际在用的序列名：可以打开H2控制台（Spring Boot里配了spring.h2.console.enabled=true就能访问），查看数据库里的序列列表，或者开启SQL日志（spring.jpa.show-sql=true），看插入数据时Hibernate调用的是哪个序列的nextval()
2. 然后在测试前的初始化方法里执行两步操作：

   // 第一步：清空表，同时重置表的自增标识
   entityManager.createNativeQuery("TRUNCATE TABLE table_name CASCADE RESTART IDENTITY").executeUpdate();
   // 第二步：重置对应的序列，如果是全局序列就用HIBERNATE_SEQUENCE
   entityManager.createNativeQuery("ALTER SEQUENCE HIBERNATE_SEQUENCE RESTART WITH 1").executeUpdate();
   

   注意顺序要对，先清表再重置序列，这样第一条插入的数据ID就会是1。

方案二：测试环境切换主键生成策略 #

如果不想折腾序列，可以在测试环境里把这个实体的主键策略改成IDENTITY，这样H2会用自增列而不是序列，TRUNCATE TABLE ... RESTART IDENTITY就能直接生效：
你可以用Spring的Profile来区分测试和生产环境：

import org.springframework.context.annotation.Profile;
import jakarta.persistence.GeneratedValue;
import jakarta.persistence.GenerationType;
import jakarta.persistence.Id;

@Entity
public class YourEntity {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Profile("test")
    @GeneratedValue(strategy = GenerationType.AUTO)
    @Profile("!test")
    private Long id;

    // 其他字段...
}

然后在测试类上加上@ActiveProfiles("test")，这样测试时就会用IDENTITY策略，清表后ID自动从1开始。

方案三：用Spring Boot的DDL自动重建（适合快速测试） #

如果你的测试对启动速度要求不高，可以在application-test.properties里配置：

spring.jpa.hibernate.ddl-auto=create-drop

这样每次测试上下文启动时都会重新创建表结构，序列自然也会重置为初始值1，不过这个方法会增加测试启动时间，适合小型项目用。

最后再给你个小提示：如果你的测试是用@Transactional注解的，测试结束后会自动回滚数据，但序列值不会回滚——这也是为什么你会遇到ID越来越高的情况，所以必须手动重置序列或者用上面的方法处理。

内容来源于stack exchange