Alawn/druid-spring-boot
1
0
Fork 0
mirror of https://github.com/drtrang/druid-spring-boot.git synced 2026-11-01 22:43:46 +00:00
Code Issues Releases Wiki Activity

initializer

Browse Source
This commit is contained in:
trang
2018-02-02 17:52:12 +08:00
parent abd4b4ae7e
commit c5fcba3fc1
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
druid-spring-boot/druid-spring-boot-autoconfigure/src/main/java/com/github/trang/druid/autoconfigure/datasource/init/DruidDataSourceInitializer.java
+9 -7
View File
@@ -1,5 +1,6 @@
package com.github.trang.druid.autoconfigure.datasource.init;
import com.alibaba.druid.pool.DruidDataSource;
import com.github.trang.druid.autoconfigure.DruidDataSourceInitializerAutoConfiguration;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
@@ -37,7 +38,7 @@ public class DruidDataSourceInitializer {
private final ApplicationContext applicationContext;
private List<DataSource> dataSources = new ArrayList<>();
private List<DruidDataSource> dataSources = new ArrayList<>();
public DruidDataSourceInitializer(DataSourceProperties properties, ApplicationContext applicationContext) {
this.properties = properties;
@@ -50,20 +51,20 @@ public class DruidDataSourceInitializer {
log.debug("Initialization disabled (not running DDL scripts)");
return;
}
if (this.applicationContext.getBeanNamesForType(DataSource.class, false, false).length > 0) {
Map<String, DataSource> beans = this.applicationContext.getBeansOfType(DataSource.class, false, false);
if (this.applicationContext.getBeanNamesForType(DruidDataSource.class, false, false).length > 0) {
Map<String, DruidDataSource> beans = this.applicationContext.getBeansOfType(DruidDataSource.class, false, false);
dataSources.addAll(beans.values());
}
if (this.dataSources == null || this.dataSources.isEmpty()) {
log.debug("No DataSource found so not initializing");
return;
}
for (DataSource dataSource : dataSources) {
for (DruidDataSource dataSource : dataSources) {
runSchemaScripts(dataSource);
}
}
private void runSchemaScripts(DataSource dataSource) {
private void runSchemaScripts(DruidDataSource dataSource) {
List<Resource> scripts = getScripts("spring.datasource.schema", this.properties.getSchema(), "schema");
if (!scripts.isEmpty()) {
String username = this.properties.getSchemaUsername();
@@ -73,7 +74,7 @@ public class DruidDataSourceInitializer {
}
}
private void runDataScripts(DataSource dataSource) {
private void runDataScripts(DruidDataSource dataSource) {
List<Resource> scripts = getScripts("spring.datasource.data", this.properties.getData(), "data");
if (!scripts.isEmpty()) {
String username = this.properties.getDataUsername();
@@ -97,7 +98,8 @@ public class DruidDataSourceInitializer {
.driverClassName(this.properties.determineDriverClassName())
.url(this.properties.determineUrl())
.username(username)
.password(password).build();
.password(password)
.build();
}
DatabasePopulatorUtils.execute(populator, dataSource);
}