如何在每个Spring Data JPA集成测试后回滚多线程产生的变更？

针对你遇到的多线程独立事务无法通过主线程回滚、且需保留TestContainer容器及Liquibase初始化表结构的场景，以下几种简便的测试后数据清理方案供你选择：

1. 自定义JDBC清理逻辑（最直接） #

利用JdbcTemplate在每个测试方法结束后执行批量清理SQL，处理外键约束后截断所有业务表（跳过Liquibase系统表）：

@Autowired
private JdbcTemplate jdbcTemplate;

@AfterEach
void cleanUpData() {
    // 禁用外键约束（以MySQL为例，其他数据库需调整语法）
    jdbcTemplate.execute("SET FOREIGN_KEY_CHECKS = 0");
    
    // 查询当前库下所有业务表
    List<String> tableNames = jdbcTemplate.queryForList(
        "SELECT table_name FROM information_schema.tables WHERE table_schema = DATABASE()",
        String.class
    );
    
    // 排除Liquibase系统表
    tableNames.removeIf(table -> table.startsWith("databasechangelog"));
    
    // 批量截断表（TRUNCATE比DELETE效率更高，且重置自增主键）
    for (String table : tableNames) {
        jdbcTemplate.execute("TRUNCATE TABLE " + table);
    }
    
    // 恢复外键约束
    jdbcTemplate.execute("SET FOREIGN_KEY_CHECKS = 1");
}

不同数据库的外键禁用语法：

• PostgreSQL: ALTER TABLE table_name DISABLE TRIGGER ALL;
• SQL Server: ALTER TABLE table_name NOCHECK CONSTRAINT ALL;

2. 用@Sql注解执行清理脚本 #

将清理逻辑写入独立SQL文件，通过Spring Test的@Sql注解指定在测试方法结束后执行：

@Sql(scripts = "/cleanup.sql", executionPhase = Sql.ExecutionPhase.AFTER_TEST_METHOD)
public class MultiThreadedIntegrationTest {
    // 测试代码
}

src/test/resources/cleanup.sql示例（MySQL）：

SET FOREIGN_KEY_CHECKS = 0;

TRUNCATE TABLE user;
TRUNCATE TABLE order_info;
-- 其他业务表...

SET FOREIGN_KEY_CHECKS = 1;

3. 基于Liquibase的清空方案 #

复用现有Liquibase流程，创建专门的清理changelog，在测试后执行：

@Autowired
private Liquibase liquibase;

@AfterEach
void cleanUpWithLiquibase() throws LiquibaseException {
    // 执行指定的清理changelog
    liquibase.update("cleanup");
}

src/test/resources/cleanup-changelog.xml示例：

<databaseChangeLog xmlns="http://www.liquibase.org/xml/ns/dbchangelog"
                   xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
                   xsi:schemaLocation="http://www.liquibase.org/xml/ns/dbchangelog
                   http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-4.9.xsd">

    <changeSet id="cleanup-data" author="test">
        <sql>SET FOREIGN_KEY_CHECKS = 0;</sql>
        <truncateTable tableName="user"/>
        <truncateTable tableName="order_info"/>
        <!-- 其他业务表 -->
        <sql>SET FOREIGN_KEY_CHECKS = 1;</sql>
    </changeSet>
</databaseChangeLog>

4. TestContainer生命周期回调 #

如果希望在整个测试类结束后统一清理，可给TestContainer添加生命周期回调：

static MySQLContainer<?> mysqlContainer = new MySQLContainer<>("mysql:8.0")
    .withInitScript("init.sql");

static {
    mysqlContainer.start();
    mysqlContainer.addLifecycleCallback(new TestcontainersLifecycleAdapter() {
        @Override
        public void afterTest(TestDescription description) {
            try (Connection conn = mysqlContainer.createConnection("")) {
                Statement stmt = conn.createStatement();
                stmt.execute("SET FOREIGN_KEY_CHECKS = 0");
                stmt.execute("TRUNCATE TABLE user");
                stmt.execute("TRUNCATE TABLE order_info");
                // 其他业务表
                stmt.execute("SET FOREIGN_KEY_CHECKS = 1");
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
    });
}

注意事项 #

• 优先使用TRUNCATE而非DELETE，效率更高且重置自增主键；
• 必须处理外键约束，否则会因关联数据存在导致清理失败；
• 若测试数据量较大，可将清理粒度从@AfterEach改为@AfterAll，但需确保测试类间数据隔离。

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