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开发这么好干呢。而且从这个问题能发现,应该是所有属性都可以自定义实现的!真是学到了。
By:努力向前的菜B
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· 单线程的Redis速度为什么快?