Spring Boot JPA本地测试：如何自动删数据并重建单个指定表

方法一：自定义SQL初始化脚本（推荐，简单直接） #

这种方式借助Spring Boot的数据源初始化能力，仅在本地环境启动时执行指定DDL脚本，完成目标表的删除与重建。

1. 创建本地专属SQL脚本
   在src/main/resources下新建schema-local.sql，写入目标表的删除、重建语句：

-- 删除目标表（存在则删除）
DROP TABLE IF EXISTS target_table;
-- 重建目标表，根据实体结构定义字段
CREATE TABLE target_table (
    id BIGINT PRIMARY KEY AUTO_INCREMENT COMMENT '主键ID',
    username VARCHAR(50) NOT NULL COMMENT '用户名',
    create_time DATETIME DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间'
    -- 按需添加其他字段
);

2. 配置本地环境初始化参数
   在application-local.properties（本地环境专属配置文件）中添加以下配置：

# 设置每次启动都执行初始化脚本
spring.sql.init.mode=always
# 指定要执行的DDL脚本路径
spring.sql.init.schema-locations=classpath:schema-local.sql
# 如需插入测试数据，可额外配置数据脚本
# spring.sql.init.data-locations=classpath:data-local.sql

注意：生产环境需将spring.sql.init.mode设为never或embedded，防止误操作。

方法二：通过Hibernate API动态生成DDL #

如果希望基于实体类自动生成DDL（避免手写SQL的维护成本），可使用Hibernate的SchemaExport工具，在应用启动时针对指定实体对应的表执行删除、重建操作。

1. 编写启动监听器
   创建Spring组件，监听应用启动完成事件，仅在本地环境执行DDL：

import org.hibernate.boot.Metadata;
import org.hibernate.boot.MetadataSources;
import org.hibernate.boot.registry.StandardServiceRegistry;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.tool.hbm2ddl.SchemaExport;
import org.hibernate.tool.schema.TargetType;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.util.EnumSet;

@Component
public class TargetTableInitializer implements ApplicationListener<ContextRefreshedEvent> {

    private final DataSource dataSource;
    private final Environment environment;

    public TargetTableInitializer(DataSource dataSource, Environment environment) {
        this.dataSource = dataSource;
        this.environment = environment;
    }

    @Override
    public void onApplicationEvent(ContextRefreshedEvent event) {
        // 仅在本地环境执行
        boolean isLocalEnv = false;
        for (String profile : environment.getActiveProfiles()) {
            if ("local".equals(profile)) {
                isLocalEnv = true;
                break;
            }
        }
        if (!isLocalEnv) return;

        // 构建Hibernate服务注册器
        StandardServiceRegistry registry = new StandardServiceRegistryBuilder()
                .applySetting("hibernate.connection.datasource", dataSource)
                .applySetting("hibernate.dialect", "org.hibernate.dialect.MySQL8Dialect") // 根据数据库调整方言
                .build();

        // 指定目标实体类（对应要重建的表）
        Metadata metadata = new MetadataSources(registry)
                .addAnnotatedClass(TargetEntity.class) // 替换为你的实体类
                .buildMetadata();

        // 执行DROP + CREATE操作
        SchemaExport export = new SchemaExport();
        export.setHaltOnError(true);
        export.setFormat(true);
        export.execute(EnumSet.of(TargetType.DATABASE), SchemaExport.Action.DROP_CREATE, metadata);

        registry.close();
    }
}

注意：替换TargetEntity为对应目标表的实体类，并根据数据库类型调整Hibernate方言。

方法三：测试用例专属脚本执行 #

如果是在单元/集成测试中需要每次测试前重建指定表，可使用Spring Test的@Sql注解：

import org.springframework.test.context.jdbc.Sql;
import org.springframework.test.context.junit.jupiter.SpringBootTest;

@SpringBootTest
// 每次测试方法执行前执行脚本
@Sql(scripts = "/drop-and-create-table.sql", executionPhase = Sql.ExecutionPhase.BEFORE_TEST_METHOD)
public class YourTestClass {
    // 测试逻辑
}

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