在不使用XMLconfiguration的情况下初始化数据库,但使用@Configuration

我想知道如何初始化一个数据库,而不必创build一个XML文件。

我已经使用这种初始化工作正常,但在我目前的情况下,我不想创build一个XML:

<jdbc:initialize-database data-source="dataSource"> <jdbc:script location="classpath:com/foo/sql/db-schema.sql"/> <jdbc:script location="classpath:com/foo/sql/db-test-data.sql"/> </jdbc:initialize-database> 

我知道我可以创build一个embedded式数据库:

 EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder(); EmbeddedDatabase db = builder.setType(H2).addScript("my-schema.sql").addScript("my-test-data.sql").build(); 

就我而言,数据库和模式是使用Liquibase创build的。

我只想用Spring和我的自定义数据集进行初始化,而不必每次都创build一个新的XML文件。

可能吗?

您的@Configuration类中的以下代码行可能工作。

 @Value("classpath:com/foo/sql/db-schema.sql") private Resource schemaScript; @Value("classpath:com/foo/sql/db-test-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() { final ResourceDatabasePopulator populator = new ResourceDatabasePopulator(); populator.addScript(schemaScript); populator.addScript(dataScript); return populator; } 

您必须创build自己的schema.sql并将其放在src/main/resources -folder中。

 import javax.sql.DataSource; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.core.io.ClassPathResource; import org.springframework.jdbc.datasource.SimpleDriverDataSource; import org.springframework.jdbc.datasource.init.DatabasePopulator; import org.springframework.jdbc.datasource.init.DatabasePopulatorUtils; import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator; @Configuration public class DataSourceInitializer { @Bean(name = "dataSource") public DataSource getDataSource(){ DataSource dataSource = createDataSource(); DatabasePopulatorUtils.execute(createDatabasePopulator(), dataSource); return dataSource; } private DatabasePopulator createDatabasePopulator() { ResourceDatabasePopulator databasePopulator = new ResourceDatabasePopulator(); databasePopulator.setContinueOnError(true); databasePopulator.addScript(new ClassPathResource("schema.sql")); return databasePopulator; } private SimpleDriverDataSource createDataSource() { SimpleDriverDataSource simpleDriverDataSource = new SimpleDriverDataSource(); simpleDriverDataSource.setDriverClass(org.h2.Driver.class); simpleDriverDataSource.setUrl("jdbc:h2:target/database/example;AUTO_RECONNECT=TRUE"); simpleDriverDataSource.setUsername(""); simpleDriverDataSource.setPassword(""); return simpleDriverDataSource; } } 

在查看与EmbeddedDatabaseBuilder相关的Spring类之后,我发现DatabaseBuilder正在使用如下代码:

 ResourceDatabasePopulator populator = new ResourceDatabasePopulator(); for (String sqlScript: sqlInitializationScripts ) { Resource sqlScriptResource = RESOURCE_LOADER.getResource(sqlScript); populator.addScript(sqlScriptResource); } DatabasePopulatorUtils.execute(populator, dataSource); 

这对我来说很好,即使它是在@BeforeTest方法而不是在Springconfiguration。

这当然是可能的。

如果你已经有了一个由Spring的ApplicationContext加载的@Configuration类,那么你只需创build一个新的@Bean方法,它将包含你已经存在的代码(当然还有一个额外的return语句)。

EmbeddedDatabase实现了DataSource接口,所以它可以很容易地和JdbcTemplate一起使用。

 @Bean public DataSource db() { EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder(); builder.setType(H2).addScript("my-schema.sql").addScript("my-test-data.sql"); return builder.build(); }