springboot项目启动时初始化数据库(只执行一次)
先说结论。
cloud项目中spring.data.sql属性可以做到项目每次启动时,都执行相应的sql,但这不够个性化。
通过DataSourceInitializer自定义来实现,具体代码:
@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
浙公网安备 33010602011771号