方法一:配置application.yml文件
纯配置。最方便简洁,是最优选择。
yml文件:
spring:
datasource:
schema: classpath:schema.sql
data: classpath:data.sql
initialization-mode: always
再在resource目录下添加schema.sql和data.sql的文件。schema中主要用于创建表的语句,data中主要存放插入数据及更新操作。
源码中是先执行schema,在执行data,并且schema中不能为空文件,所以可以在schema.sql文件中随便写一行sql,不产生沉余数据就行,比如:show tables;
需要注意的:
springboot 2.0以上的版本initialization-mode属性必须要有。
启动类中的 DataSourceAutoConfiguration.class注解会让配置失效,或者druid的防火墙也会让此方法失效。
方法二:自定义方法
@Component
public class CustomizeDataSourceInitializer implements InitializingBean{
@Value("classpath:sql/data.sql")
private Resource dataScript;
@Bean
public DataSourceInitializer dataSourceInitializer(final DataSource dataSource) {
final DataSourceInitializer initializer = new DataSourceInitializer();
// 设置数据源
initializer.setDataSource(dataSource);
initializer.setDatabasePopulator(databasePopulator());
return initializer;
}
private DatabasePopulator databasePopulator() {
System.out.println("==================sql脚本正在执行==================");
final ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
populator.addScripts(dataScript);
System.out.println("==================sql脚本初始化完成==================");
return populator;
}
@Override
public void afterPropertiesSet() throws Exception {
}
然后在启动时执行方法就行,这里是实现InitializingBean接口来执行的。
springboot启动执行方法的方式有很多,选一种就行。