springboot项目启动时初始化数据库(只执行一次)

  先说结论。

  cloud项目中spring.data.sql属性可以做到项目每次启动时,都执行相应的sql,但这不够个性化。

  通过DataSourceInitializer自定义来实现,具体代码:

  

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
@Configuration
@Slf4j
public class DataSourceInitializer {
 
 
    @javax.annotation.Resource(name = "dataSourceFirst")
    private DataSource dataSource;
 
    @Value("classpath:db/xx.sql")
    private Resource ddl;
    @Value("classpath:db/data.sql")
    private Resource dml;
 
    @Bean
    public DataSourceInitializer dataSourceInitializer(final DataSource dataSource) {
        final DataSourceInitializer initializer = new DataSourceInitializer();
        // 设置数据源
        initializer.setDataSource(dataSource);
        initializer.setDatabasePopulator(databasePopulator());
        return initializer;
    }
 
    private DatabasePopulator databasePopulator() {
        final ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
 
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
 
        try {
            Map map = jdbcTemplate.queryForMap("select count(1) from xx");
        } catch (DataAccessException e) {
            log.warn("初始化脚本报错:" + e.getMessage());
            // 报错,表不存在,初次导入,执行相关脚本
            populator.addScripts(ddl);
//            populator.addScripts(dml);
        } catch (Exception e) {
            log.warn("初始化脚本报错:" + e.getMessage());
        }
 
 
        return populator;
    }
 
 
 
 
 
 
}

  大体就是这样,然后DML的sql,放到服务器上会有新的报错,我就用了比较简单的方式,放到Java里执行,CommandLineRunner,通过这个类,一样的判断,这样就可以初始化相关的只执行一次,不用每次都覆盖了。

  这个我断断续续找了几天都没找到,可能是因为这个需求比较少见吧。写这个文章前我突然想到,sql里似乎也可以做到,create if not exsit就可以……怪不得前两天没发现别人这么写,

  但这个写法多少也有点用吧,linux执行有的会乱码,导致执行失败,很怪。

  最后,springcloud真牛啊,大部分功能都约定了,怪不得Java开发这么好干呢。而且从这个问题能发现,应该是所有属性都可以自定义实现的!真是学到了。

  

 

posted on   唯心、tt  阅读(3170)  评论(0编辑  收藏  举报

相关博文:
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· 单线程的Redis速度为什么快?

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示