diff --git a/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/DynamicDataSource.java b/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/DynamicDataSource.java index 3c6e573..12f2e31 100644 --- a/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/DynamicDataSource.java +++ b/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/DynamicDataSource.java @@ -18,11 +18,9 @@ import java.util.concurrent.atomic.AtomicInteger; public class DynamicDataSource extends AbstractRoutingDataSource { private AtomicInteger count = new AtomicInteger(); - private DataSource defaultTargetDataSource; @Getter private Map targetDataSources; - public DynamicDataSource(DataSource defaultTargetDataSource, Map targetDataSources) { - this.defaultTargetDataSource = defaultTargetDataSource; + public DynamicDataSource(Map targetDataSources) { this.targetDataSources = targetDataSources; } @@ -35,7 +33,6 @@ public class DynamicDataSource extends AbstractRoutingDataSource { @Override public void afterPropertiesSet() { - super.setDefaultTargetDataSource(defaultTargetDataSource); super.setTargetDataSources(new HashMap<>(targetDataSources)); super.afterPropertiesSet(); } diff --git a/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/SpringDataSourceConfig.java b/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/SpringDataSourceConfig.java index 038abf4..f2aabac 100644 --- a/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/SpringDataSourceConfig.java +++ b/druid-spring-boot-example/druid-spring-boot-mybatis-example/src/main/java/com/github/trang/druid/example/config/SpringDataSourceConfig.java @@ -31,7 +31,7 @@ public class SpringDataSourceConfig { @Primary public DynamicDataSource dataSource(Map druidDataSourceMap) { Map dataSourceMap = new HashMap<>(druidDataSourceMap); - return new DynamicDataSource(dataSourceMap.get("master"), dataSourceMap); + return new DynamicDataSource(dataSourceMap); } } \ No newline at end of file