web-dev-qa-db-ja.com

XML設定なしでデータベースを初期化するが、@ Configurationを使用する

XMLファイルを作成せずにデータベースを初期化する方法を知りたいです。

私はすでにこの種の初期化をうまく使用していますが、現在のケースでは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>

私は埋め込みデータベースを作成できることを知っています：

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

私の場合、データベースとスキーマはLiquibaseを使用して作成されます。

そのために毎回新しいXMLファイルを作成することなく、Springとカスタマイズされたデータセットで初期化したいだけです。

出来ますか？

javaspringh2
43
Sebastien Lorber

EmbeddedDatabaseBuilderに関連するSpringクラスを見て、DatabaseBuilderが次のようなコードを使用していることがわかりました。

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

これは、Spring構成ではなく@BeforeTestメソッドで実行される場合でも、うまく機能します。

6
Sebastien Lorber

@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;
}
34
Sascha Krüger

独自のschema.sqlを作成し、src/main/resourcesフォルダーに配置する必要があります。

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;      
    }
}
25
Jan-Terje Sørensen

確かに可能です。

SpringのApplicationContextによってロードされる@Configurationクラスが既にある場合は、既にあるコードを含む新しい@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();
}
6
nicholas.hauschild