如何在H2文件数据库已存在时禁用schema.sql加载？

我完全懂你的困扰——每次启动应用都跑一遍schema.sql实在没必要，尤其是数据库已经存在的时候。下面给你两个实用的Spring生态方案，精准实现「仅首次创建数据库时执行初始化脚本」的需求：

方案一：通过检查H2数据库文件是否存在控制 #

因为你用的是文件存储的H2，数据库创建后会生成对应的物理文件（比如JDBC URL是jdbc:h2:file:./my-db，就会生成my-db.mv.db文件）。我们可以利用这个特性，通过Spring条件注解决定是否执行初始化：

1. 自定义条件类，判断H2数据库文件是否存在：

import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.core.type.AnnotatedTypeMetadata;
import java.io.File;

public class H2DatabaseNotExistsCondition implements Condition {
    @Override
    public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        // 从环境变量获取H2的JDBC URL
        String jdbcUrl = context.getEnvironment().getProperty("spring.datasource.url");
        if (jdbcUrl == null || !jdbcUrl.startsWith("jdbc:h2:file:")) {
            return false;
        }
        // 提取文件路径，补全H2默认的.mv.db后缀
        String filePath = jdbcUrl.replace("jdbc:h2:file:", "") + ".mv.db";
        File dbFile = new File(filePath);
        // 文件不存在=首次创建数据库，返回true执行初始化
        return !dbFile.exists();
    }
}

2. 配置DataSourceInitializer，仅满足条件时生效：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Conditional;
import org.springframework.core.io.ClassPathResource;
import org.springframework.jdbc.datasource.init.DataSourceInitializer;
import org.springframework.jdbc.datasource.init.DatabasePopulator;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
import javax.sql.DataSource;

@Configuration
public class DatabaseConfig {

    @Bean
    @Conditional(H2DatabaseNotExistsCondition.class)
    public DataSourceInitializer dataSourceInitializer(DataSource dataSource) {
        DataSourceInitializer initializer = new DataSourceInitializer();
        initializer.setDataSource(dataSource);
        initializer.setDatabasePopulator(databasePopulator());
        return initializer;
    }

    private DatabasePopulator databasePopulator() {
        ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
        // 加载你的schema.sql脚本
        populator.addScript(new ClassPathResource("schema.sql"));
        // 如有需要，可添加data.sql初始化数据
        // populator.addScript(new ClassPathResource("data.sql"));
        return populator;
    }
}

这个方案的优势是直接通过物理文件判断，无需连接数据库，性能更快；缺点是依赖H2的文件命名规则，如果JDBC URL有特殊配置（如加密、自定义路径），需要调整路径提取逻辑。

方案二：通过数据库表存在性判断（更通用） #

如果担心文件路径判断不够稳妥，或者未来可能切换其他文件型数据库，可以通过连接数据库后检查业务表（或系统表）是否存在来决定是否执行脚本：

1. 自定义初始化类，实现InitializingBean做检查：

import org.springframework.beans.factory.InitializingBean;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
import org.springframework.core.io.ClassPathResource;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

@Component
public class DatabaseInitializer implements InitializingBean {

    private final DataSource dataSource;
    private final JdbcTemplate jdbcTemplate;

    public DatabaseInitializer(DataSource dataSource, JdbcTemplate jdbcTemplate) {
        this.dataSource = dataSource;
        this.jdbcTemplate = jdbcTemplate;
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        // 检查目标业务表是否存在（替换成你的表名，比如"USER"）
        String checkTableSql = "SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = 'USER'";
        try {
            Integer tableCount = jdbcTemplate.queryForObject(checkTableSql, Integer.class);
            if (tableCount == null || tableCount == 0) {
                // 表不存在，执行初始化脚本
                executeSchemaScript();
            }
        } catch (Exception e) {
            // 查询失败（比如首次创建空库），同样执行脚本
            executeSchemaScript();
        }
    }

    private void executeSchemaScript() throws SQLException {
        ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
        populator.addScript(new ClassPathResource("schema.sql"));
        try (Connection connection = dataSource.getConnection()) {
            populator.populate(connection);
        }
    }
}

这个方案更通用，不依赖H2的文件系统细节，支持所有标准SQL数据库；缺点是需要先建立数据库连接，但对于文件型H2来说，这个开销几乎可以忽略。

额外提示 #

• 如果你用Spring Boot，默认的spring.datasource.initialization-mode属性中，always会每次执行，embedded仅针对内存数据库，所以还是需要上述自定义逻辑实现文件H2的首次执行。
• 如果你的项目有长期版本迭代需求，推荐使用Flyway或Liquibase这类专业数据库迁移工具，它们能更优雅地管理脚本版本和执行时机。

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