|
@@ -1,19 +1,17 @@
|
|
|
package com.baomidou.mybatisplus.test.h2.config;
|
|
|
|
|
|
-import java.sql.SQLException;
|
|
|
-
|
|
|
import javax.sql.DataSource;
|
|
|
-
|
|
|
import org.h2.Driver;
|
|
|
import org.springframework.context.annotation.Bean;
|
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
-import org.springframework.core.io.ClassPathResource;
|
|
|
+import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
|
|
|
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
|
|
|
import org.springframework.jdbc.datasource.SimpleDriverDataSource;
|
|
|
import org.springframework.jdbc.datasource.init.DataSourceInitializer;
|
|
|
import org.springframework.jdbc.datasource.init.DatabasePopulator;
|
|
|
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
|
|
|
import org.springframework.transaction.annotation.EnableTransactionManagement;
|
|
|
+import java.io.IOException;
|
|
|
|
|
|
/**
|
|
|
* <p>
|
|
@@ -43,7 +41,7 @@ public class DBConfig {
|
|
|
}
|
|
|
|
|
|
@Bean
|
|
|
- public DataSourceInitializer dataSourceInitializer(DataSource dataSource) {
|
|
|
+ public DataSourceInitializer dataSourceInitializer(DataSource dataSource) throws IOException {
|
|
|
final DataSourceInitializer initializer = new DataSourceInitializer();
|
|
|
initializer.setDataSource(dataSource);
|
|
|
initializer.setDatabasePopulator(databasePopulator());
|
|
@@ -51,15 +49,11 @@ public class DBConfig {
|
|
|
return initializer;
|
|
|
}
|
|
|
|
|
|
- private DatabasePopulator databasePopulator() {
|
|
|
+ private DatabasePopulator databasePopulator() throws IOException {
|
|
|
ResourceDatabasePopulator resourceDatabasePopulator = new ResourceDatabasePopulator();
|
|
|
resourceDatabasePopulator.setContinueOnError(false);
|
|
|
resourceDatabasePopulator.addScripts(
|
|
|
- //todo 这里由你们来进化一下.
|
|
|
- new ClassPathResource("/h2/student.ddl.sql"),
|
|
|
- new ClassPathResource("/h2/student.insert.sql"),
|
|
|
- new ClassPathResource("/h2/user.ddl.sql"),
|
|
|
- new ClassPathResource("/h2/user.insert.sql")
|
|
|
+ new PathMatchingResourcePatternResolver().getResources("classpath:/h2/*.sql")
|
|
|
);
|
|
|
return resourceDatabasePopulator;
|
|
|
}
|