Spring Boot生产环境下如何自动创建数据库

你遇到的核心问题是自定义DataSource Bean覆盖了Spring Boot的自动配置，导致配置参数失效，同时需要在数据源初始化前执行数据库创建逻辑。下面是两种无需修改默认数据源的可行方案：

方案一：监听ApplicationStartingEvent事件（最早期执行） #

这个事件是Spring Boot启动流程中最早触发的事件之一，此时Spring上下文尚未初始化，数据源还未开始配置，完全不会干扰自动配置逻辑。我们可以直接用原生JDBC连接数据库服务器的默认库（比如MySQL的mysql库、PostgreSQL的postgres库）创建目标数据库。

import org.springframework.boot.context.event.ApplicationStartingEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.stereotype.Component;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;

@Component
public class DatabasePreCreator implements ApplicationListener<ApplicationStartingEvent> {

    @Override
    public void onApplicationEvent(ApplicationStartingEvent event) {
        // 从环境中读取配置参数
        String dbUrl = event.getEnvironment().getProperty("spring.datasource.url");
        String username = event.getEnvironment().getProperty("spring.datasource.username");
        String password = event.getEnvironment().getProperty("spring.datasource.password");

        // 拆分出数据库服务器地址和目标库名称
        int lastSlashIndex = dbUrl.lastIndexOf('/');
        String baseServerUrl = dbUrl.substring(0, lastSlashIndex);
        String targetDbName = dbUrl.substring(lastSlashIndex + 1);

        // 根据数据库类型调整创建语句，这里以MySQL为例
        String createDbSql = String.format("CREATE DATABASE IF NOT EXISTS `%s`", targetDbName);

        try (Connection conn = DriverManager.getConnection(baseServerUrl, username, password);
             Statement stmt = conn.createStatement()) {
            stmt.executeUpdate(createDbSql);
        } catch (SQLException e) {
            throw new RuntimeException("创建目标数据库失败", e);
        }
    }
}

方案二：用@AutoConfigureBefore提前执行数据库创建逻辑 #

通过@AutoConfigureBefore注解，让我们的配置类优先于Spring Boot的数据源自动配置类执行，同时利用DataSourceProperties读取配置参数，避免硬编码。

import jakarta.annotation.PostConstruct;
import org.springframework.boot.autoconfigure.AutoConfigureBefore;
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;

@Configuration
@AutoConfigureBefore(DataSourceAutoConfiguration.class)
@EnableConfigurationProperties(DataSourceProperties.class)
public class DatabaseCreationConfig {

    @Bean
    public DatabaseInitializer databaseInitializer(DataSourceProperties dataSourceProperties) {
        return new DatabaseInitializer(dataSourceProperties);
    }

    public static class DatabaseInitializer {
        private final String baseServerUrl;
        private final String targetDbName;
        private final String username;
        private final String password;

        public DatabaseInitializer(DataSourceProperties properties) {
            String dbUrl = properties.getUrl();
            int lastSlashIndex = dbUrl.lastIndexOf('/');
            this.baseServerUrl = dbUrl.substring(0, lastSlashIndex);
            this.targetDbName = dbUrl.substring(lastSlashIndex + 1);
            this.username = properties.getUsername();
            this.password = properties.getPassword();
        }

        @PostConstruct
        public void createTargetDatabase() {
            // 示例为MySQL语句，PostgreSQL/Oracle等需调整
            String createSql = String.format("CREATE DATABASE IF NOT EXISTS `%s`", targetDbName);

            try (Connection conn = DriverManager.getConnection(baseServerUrl, username, password);
                 Statement stmt = conn.createStatement()) {
                stmt.executeUpdate(createSql);
            } catch (SQLException e) {
                throw new RuntimeException("初始化目标数据库失败", e);
            }
        }
    }
}

注意事项 #

• 确保执行创建操作的数据库用户拥有CREATE DATABASE权限；
• 不同数据库的创建语句有差异，比如PostgreSQL需用CREATE DATABASE IF NOT EXISTS "%s";，Oracle需创建表空间和用户，需根据实际数据库调整；
• 方案一和方案二都不会修改或覆盖Spring Boot自动配置的DataSource，完全保留原有配置逻辑。

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