From 1404c6048e41e71e35577bea5f4906570042f506 Mon Sep 17 00:00:00 2001 From: lsy <18332763730@163.com> Date: Wed, 27 Dec 2017 00:47:07 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B0=81=E5=AD=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../browser/config/BrowserSecurityConfig.java | 1 - boot-starter/boot-starter-ssm/pom.xml | 4 --- ...ion.java => SlaveDataSourceCondition.java} | 8 ++--- .../common/config/DruidDataSourceConfig.java | 19 +++++++---- .../config/DynamicDataSourceConfig.java | 33 +++++++++++++++---- .../config/SqlSessionFactoryConfig.java | 1 - .../common/dao/DataSourceContextHolder.java | 6 +++- .../fast/common/dao/DynamicDataSource.java | 4 +-- ms/cms/src/main/resources/bootstrap.yml | 2 +- ms/gateway/src/main/resources/bootstrap.yml | 2 +- .../src/main/resources/bootstrap.yml | 2 +- .../src/main/resources/bootstrap.yml | 2 +- .../fast/ucenter/UcenterApplication.java | 4 --- ms/ucenter/src/main/resources/bootstrap.yml | 2 +- ms/upms/src/main/resources/bootstrap.yml | 2 +- 15 files changed, 55 insertions(+), 37 deletions(-) rename boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/{SalveDataSourceCondition.java => SlaveDataSourceCondition.java} (78%) diff --git a/boot-starter/boot-starter-security/boot-starter-security-browser/src/main/java/com/micro/fast/security/browser/config/BrowserSecurityConfig.java b/boot-starter/boot-starter-security/boot-starter-security-browser/src/main/java/com/micro/fast/security/browser/config/BrowserSecurityConfig.java index 6fd4f0c..990b0ef 100644 --- a/boot-starter/boot-starter-security/boot-starter-security-browser/src/main/java/com/micro/fast/security/browser/config/BrowserSecurityConfig.java +++ b/boot-starter/boot-starter-security/boot-starter-security-browser/src/main/java/com/micro/fast/security/browser/config/BrowserSecurityConfig.java @@ -65,7 +65,6 @@ public class BrowserSecurityConfig extends WebSecurityConfigurerAdapter { * 需要注入数据源 */ @Autowired - @Qualifier("masterDataSource") private DataSource dataSource; @Autowired diff --git a/boot-starter/boot-starter-ssm/pom.xml b/boot-starter/boot-starter-ssm/pom.xml index 05badc3..14aa8d1 100644 --- a/boot-starter/boot-starter-ssm/pom.xml +++ b/boot-starter/boot-starter-ssm/pom.xml @@ -61,10 +61,6 @@ mysql-connector-java - - com.alibaba - druid - com.alibaba druid-spring-boot-starter diff --git a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/SalveDataSourceCondition.java b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/SlaveDataSourceCondition.java similarity index 78% rename from boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/SalveDataSourceCondition.java rename to boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/SlaveDataSourceCondition.java index 8266020..8433a04 100644 --- a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/SalveDataSourceCondition.java +++ b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/condition/SlaveDataSourceCondition.java @@ -13,13 +13,13 @@ import org.springframework.core.type.AnnotatedTypeMetadata; * @version 1.0 * @since 1.0 */ -public class SalveDataSourceCondition implements Condition { - private static final Logger logger = LoggerFactory.getLogger(SalveDataSourceCondition.class); +public class SlaveDataSourceCondition implements Condition { + private static final Logger logger = LoggerFactory.getLogger(SlaveDataSourceCondition.class); @Override public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) { Environment environment = conditionContext.getEnvironment(); //判断环境中是否有从数据库的配置 - logger.info("是否包含从数据库: {}",environment.containsProperty("druid.two.username")); - return environment.containsProperty("druid.two.username"); + logger.info("是否包含从数据库: {}",environment.containsProperty("spring.datasource.druid.slave.username")); + return environment.containsProperty("spring.datasource.druid.slave.username"); } } diff --git a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DruidDataSourceConfig.java b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DruidDataSourceConfig.java index 2361a8f..da794ea 100644 --- a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DruidDataSourceConfig.java +++ b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DruidDataSourceConfig.java @@ -1,6 +1,7 @@ package com.micro.fast.common.config; import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; +import com.micro.fast.common.condition.SlaveDataSourceCondition; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Conditional; @@ -9,19 +10,23 @@ import org.springframework.context.annotation.Primary; import javax.sql.DataSource; +/** + * @author lsy + */ @Configuration public class DruidDataSourceConfig { + @Primary - @Bean(name = "masterDataSource") - @ConfigurationProperties("druid.one") - public DataSource dataSourceOne(){ + @Bean(name = DynamicDataSourceConfig.MASTER_DATA_SOURCE) + @ConfigurationProperties("spring.datasource.druid.master") + public DataSource masterDataSource(){ return DruidDataSourceBuilder.create().build(); } - @Bean(name = "slaveDataSource") - @ConfigurationProperties("druid.two") - @Conditional(com.micro.fast.common.condition.SalveDataSourceCondition.class) - public DataSource dataSourceTwo(){ + @Bean(name = DynamicDataSourceConfig.SLAVE_DATA_SOURCE) + @ConfigurationProperties("spring.datasource.druid.slave") + @Conditional(SlaveDataSourceCondition.class) + public DataSource slaveDataSource(){ return DruidDataSourceBuilder.create().build(); } } diff --git a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DynamicDataSourceConfig.java b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DynamicDataSourceConfig.java index c806075..dfaf130 100644 --- a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DynamicDataSourceConfig.java +++ b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/DynamicDataSourceConfig.java @@ -1,5 +1,6 @@ package com.micro.fast.common.config; +import com.micro.fast.common.dao.DataSourceContextHolder; import com.micro.fast.common.dao.DynamicDataSource; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; @@ -22,20 +23,31 @@ import java.util.Map; @Configuration @AutoConfigureBefore({SqlSessionFactoryConfig.class}) public class DynamicDataSourceConfig { + public static final String SLAVE = "slave"; + public static final String MASTER = "master"; + public static final String SLAVE_DATA_SOURCE = "slaveDataSource"; + public static final String MASTER_DATA_SOURCE = "masterDataSource"; + /** - * 注入主数据源 + * 当ataSource存在时 */ @Autowired - @Qualifier("masterDataSource") + @Qualifier(MASTER_DATA_SOURCE) private DataSource masterDataSource; /** * 从数据源 */ @Autowired(required = false) - @Qualifier("slaveDataSource") + @Qualifier(SLAVE_DATA_SOURCE) private DataSource slaveDataSource; + /** + * 收集所有的datasource + */ + @Autowired + private Map dataSourceMap; + @Bean(name = "dynamicDataSource") public DataSource dynamicDataSource(){ //动态数据源对象 @@ -44,12 +56,19 @@ public class DynamicDataSourceConfig { dynamicDataSource.setDefaultTargetDataSource(masterDataSource); //配置多数据源 Map dataSourceMap = new HashMap<>(); - //填入主数据源 - dataSourceMap.put("master",masterDataSource); - //如果从数据源存在，填入从数据源 + //根据数据源存在情况调整动态数据源配置 + dataSourceMap.put(MASTER,masterDataSource); if (slaveDataSource!=null){ - dataSourceMap.put("slave",slaveDataSource); + dataSourceMap.put(SLAVE,slaveDataSource); } + this.dataSourceMap.forEach((key,value) -> { + if (!SLAVE_DATA_SOURCE.equals(key) + && !MASTER_DATA_SOURCE.equals(key) + && !MASTER.equals(key) + && !SLAVE.equals(key)){ + dataSourceMap.put(key,value); + } + }); dynamicDataSource.setTargetDataSources(dataSourceMap); return dynamicDataSource; } diff --git a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/SqlSessionFactoryConfig.java b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/SqlSessionFactoryConfig.java index 550a332..861aebc 100644 --- a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/SqlSessionFactoryConfig.java +++ b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/config/SqlSessionFactoryConfig.java @@ -38,7 +38,6 @@ public class SqlSessionFactoryConfig implements TransactionManagementConfigurer{ /** * 数据源对象 - * @see com.micro.fast.common.config.DataSourceMasterConfig */ @Autowired @Qualifier("dynamicDataSource") diff --git a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DataSourceContextHolder.java b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DataSourceContextHolder.java index bc9560d..f59faf3 100644 --- a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DataSourceContextHolder.java +++ b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DataSourceContextHolder.java @@ -1,5 +1,6 @@ package com.micro.fast.common.dao; +import com.micro.fast.common.config.DynamicDataSourceConfig; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -25,7 +26,7 @@ public class DataSourceContextHolder { /** * 默认的数据源的名称 */ - public static final String DEFAULT_DATDASOURCE_NAME = "master"; + public static final String DEFAULT_DATDASOURCE_NAME = DynamicDataSourceConfig.MASTER; /** * 设置数据源名 @@ -41,6 +42,9 @@ public class DataSourceContextHolder { * @return */ public static String getDatdasourceName(){ + if (contextHolder.get()==null){ + return DEFAULT_DATDASOURCE_NAME; + } return contextHolder.get(); } diff --git a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DynamicDataSource.java b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DynamicDataSource.java index 420bd41..439833b 100644 --- a/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DynamicDataSource.java +++ b/boot-starter/boot-starter-ssm/src/main/java/com/micro/fast/common/dao/DynamicDataSource.java @@ -14,11 +14,11 @@ public class DynamicDataSource extends AbstractRoutingDataSource{ /** * 记录日志 */ - private static final Logger logger = LoggerFactory.getLogger(DynamicDataSource.class); + private static final Logger log = LoggerFactory.getLogger(DynamicDataSource.class); @Override public Object determineCurrentLookupKey() { - logger.debug("数据源{}",DataSourceContextHolder.getDatdasourceName()); + log.debug("数据源{}",DataSourceContextHolder.getDatdasourceName()); return DataSourceContextHolder.getDatdasourceName(); } } diff --git a/ms/cms/src/main/resources/bootstrap.yml b/ms/cms/src/main/resources/bootstrap.yml index 64b3ea4..8ab47dc 100644 --- a/ms/cms/src/main/resources/bootstrap.yml +++ b/ms/cms/src/main/resources/bootstrap.yml @@ -3,7 +3,7 @@ spring: cloud: config: profile: dev - label: master + label: druid-springboot name: cms discovery: enabled: true diff --git a/ms/gateway/src/main/resources/bootstrap.yml b/ms/gateway/src/main/resources/bootstrap.yml index 5888fe6..a34061d 100644 --- a/ms/gateway/src/main/resources/bootstrap.yml +++ b/ms/gateway/src/main/resources/bootstrap.yml @@ -3,7 +3,7 @@ spring: cloud: config: profile: dev - label: master + label: druid-springboot name: gateway discovery: enabled: true diff --git a/ms/register-center1/src/main/resources/bootstrap.yml b/ms/register-center1/src/main/resources/bootstrap.yml index 61e2980..77e0220 100644 --- a/ms/register-center1/src/main/resources/bootstrap.yml +++ b/ms/register-center1/src/main/resources/bootstrap.yml @@ -4,7 +4,7 @@ spring: config: uri: http://config:server@www.config.com:10000/ profile: dev - label: master + label: druid-springboot name: eureka-server eureka: client: diff --git a/ms/register-center2/src/main/resources/bootstrap.yml b/ms/register-center2/src/main/resources/bootstrap.yml index da23d1a..1a0d8bd 100644 --- a/ms/register-center2/src/main/resources/bootstrap.yml +++ b/ms/register-center2/src/main/resources/bootstrap.yml @@ -3,7 +3,7 @@ spring: config: name: eureka-server2 profile: dev - label: master + label: druid-springboot uri: http://config:server@www.config.com:10000/ eureka: client: diff --git a/ms/ucenter/src/main/java/com/micro/fast/ucenter/UcenterApplication.java b/ms/ucenter/src/main/java/com/micro/fast/ucenter/UcenterApplication.java index ae5362c..7a85b24 100644 --- a/ms/ucenter/src/main/java/com/micro/fast/ucenter/UcenterApplication.java +++ b/ms/ucenter/src/main/java/com/micro/fast/ucenter/UcenterApplication.java @@ -2,10 +2,6 @@ package com.micro.fast.ucenter; import org.mybatis.spring.annotation.MapperScan; import org.springframework.boot.SpringApplication; -import org.springframework.boot.autoconfigure.EnableAutoConfiguration; -import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration; -import org.springframework.boot.autoconfigure.jdbc.JdbcTemplateAutoConfiguration; -import org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaAutoConfiguration; import org.springframework.cloud.client.SpringCloudApplication; import org.springframework.cloud.netflix.feign.EnableFeignClients; diff --git a/ms/ucenter/src/main/resources/bootstrap.yml b/ms/ucenter/src/main/resources/bootstrap.yml index bea055e..a295237 100644 --- a/ms/ucenter/src/main/resources/bootstrap.yml +++ b/ms/ucenter/src/main/resources/bootstrap.yml @@ -3,7 +3,7 @@ spring: cloud: config: profile: dev - label: master + label: druid-springboot name: ucenter discovery: service-id: config-server diff --git a/ms/upms/src/main/resources/bootstrap.yml b/ms/upms/src/main/resources/bootstrap.yml index e791fcb..28f13fa 100644 --- a/ms/upms/src/main/resources/bootstrap.yml +++ b/ms/upms/src/main/resources/bootstrap.yml @@ -5,7 +5,7 @@ spring: cloud: config: profile: dev - label: master + label: druid-springboot name: upms discovery: service-id: config-server