Spring Boot JPA：在应用启动时优雅地创建和管理数据库视图

package com.example.app.bootstrap;

public interface DataLoader {
    /**
     * 加载环境特定数据并创建数据库视图。
     * 该方法应在Spring应用启动后，JPA表结构创建完成后执行。
     */
    void loadEnvironmentSpecificData();
}

package com.example.app.bootstrap;

import com.example.app.repository.PriceRepository; // 假设您有一个JPA仓库
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;

// 生产环境数据加载器
@Component
@Profile("prod") // 仅在Spring Profile为"prod"时激活
public class ProductionDataLoader implements DataLoader {

    private final PriceRepository priceRepository; // 示例：如果您需要基于实体数据创建视图
    private final JdbcTemplate jdbcTemplate; // 用于执行SQL创建视图

    @Autowired
    public ProductionDataLoader(PriceRepository priceRepository, JdbcTemplate jdbcTemplate) {
        this.priceRepository = priceRepository;
        this.jdbcTemplate = jdbcTemplate;
    }

    @Override
    public void loadEnvironmentSpecificData() {
        System.out.println("生产环境数据加载器：开始创建数据库视图...");

        // 示例：创建一个名为 'active_products_view' 的视图
        // 假设您有一个 'Product' 表，并且它已经被JPA自动创建
        String createProductViewSql = """
            CREATE OR REPLACE VIEW active_products_view AS
            SELECT p.id, p.name, p.price, p.status
            FROM Product p
            WHERE p.status = 'ACTIVE'
            """;
        try {
            jdbcTemplate.execute(createProductViewSql);
            System.out.println("视图 'active_products_view' 创建成功。");
        } catch (Exception e) {
            System.err.println("创建视图 'active_products_view' 失败: " + e.getMessage());
            // 生产环境应记录更详细的日志并考虑合适的错误处理策略
        }

        // 可以在这里执行其他生产环境特有的初始化数据加载逻辑
        // 例如：加载一些默认配置数据
        // priceRepository.save(new Price(...));
    }
}

package com.example.app.bootstrap;

import com.example.app.repository.PriceRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;

// 开发环境数据加载器
@Component
@Profile("dev") // 仅在Spring Profile为"dev"时激活
public class DevelopmentDataLoader implements DataLoader {

    private final PriceRepository priceRepository;
    private final JdbcTemplate jdbcTemplate;

    @Autowired
    public DevelopmentDataLoader(PriceRepository priceRepository, JdbcTemplate jdbcTemplate) {
        this.priceRepository = priceRepository;
        this.jdbcTemplate = jdbcTemplate;
    }

    @Override
    public void loadEnvironmentSpecificData() {
        System.out.println("开发环境数据加载器：开始创建数据库视图和加载测试数据...");

        // 示例：创建一个名为 'test_products_summary_view' 的视图
        String createTestViewSql = """
            CREATE OR REPLACE VIEW test_products_summary_view AS
            SELECT p.id, p.name, p.status
            FROM Product p
            LIMIT 10
            """; // 开发环境可能只需要部分数据或简化视图
        try {
            jdbcTemplate.execute(createTestViewSql);
            System.out.println("视图 'test_products_summary_view' 创建成功。");
        } catch (Exception e) {
            System.err.println("创建视图 'test_products_summary_view' 失败: " + e.getMessage());
        }

        // 加载测试数据
        // priceRepository.save(new Price(1L, "Test Product A", BigDecimal.TEN));
        // priceRepository.save(new Price(2L, "Test Product B", BigDecimal.valueOf(20)));
    }
}

package com.example.app.bootstrap;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component;

import java.util.List;

/**
 * 应用启动协调器，负责在应用完全就绪后触发数据加载器。
 */
@Component
public class DataBootstrapRunner implements ApplicationRunner {

    // Spring会自动注入当前激活的所有DataLoader实现
    // 由于我们使用了@Profile，通常只有一个DataLoader会被激活并注入到列表中
    private final List<DataLoader> dataLoaders;

    @Autowired
    public DataBootstrapRunner(List<DataLoader> dataLoaders) {
        this.dataLoaders = dataLoaders;
    }

    @Override
    public void run(ApplicationArguments args) throws Exception {
        System.out.println("数据引导程序启动：准备执行环境特定的数据加载和视图创建...");
        if (dataLoaders.isEmpty()) {
            System.out.println("未找到激活的数据加载器。请检查Spring Profile配置。");
        } else {
            // 遍历并执行所有激活的DataLoader
            dataLoaders.forEach(DataLoader::loadEnvironmentSpecificData);
            System.out.println("数据引导程序执行完毕。");
        }
    }
}

spring.profiles.active=dev
# 或者
spring.profiles.active=prod