动态数据源配置

分手后的思念是犯贱 2022-07-12 07:30 205阅读 0赞

在使用dubbo作为分布式服务治理框架时，遇到了一个问题，像优惠码这样的大批量数据查询和导出接口，是不适合提供dubbo服务的。![Center][]

所以要讲优惠码查询接口迁移到cms里来。这样的话需要配置两个数据源，具体做法如下：

一、配置多数据源：

<bean id="komectbackend" class="org.apache.commons.dbcp2.BasicDataSource">
            <property name="url" value="${jdbc.komectbackend.url}" />
            <property name="driverClassName" value="${jdbc.driver}" />
            <property name="username" value="${jdbc.komectbackend.username}" />
            <property name="password" value="${jdbc.komectbackend.password}" />
    
            
            <property name="maxIdle" value="20"/>
            
            <property name="minIdle" value="5"/>
            <property name="initialSize" value="5"/>
    
            
            <property name="testOnBorrow" value="false"/>
            
            <property name="testOnReturn" value="false"/>
            
            <property name="testWhileIdle" value="true"/>
            
            <property name="timeBetweenEvictionRunsMillis" value="60000"/>
            
            <property name="validationQuery" value="select user()"/>
        </bean>
         
        <bean id="komecthealth" class="org.apache.commons.dbcp2.BasicDataSource">
            <property name="url" value="${jdbc.komecthealth.url}" />
            <property name="driverClassName" value="${jdbc.driver}" />
            <property name="username" value="${jdbc.komecthealth.username}" />
            <property name="password" value="${jdbc.komecthealth.password}" />
    
            
            <property name="maxIdle" value="20"/>
            
            <property name="minIdle" value="5"/>
            <property name="initialSize" value="5"/>
    
            
            <property name="testOnBorrow" value="false"/>
            
            <property name="testOnReturn" value="false"/>
            
            <property name="testWhileIdle" value="true"/>
            
            <property name="timeBetweenEvictionRunsMillis" value="60000"/>
            
            <property name="validationQuery" value="select user()"/>
        </bean>
        <bean id="dynamicDataSource" class="com.cmcc.komectcms.common.util.DynamicDataSource" >  
    	      
    	    <property name="targetDataSources">  
    	        <map>  
    	            <entry value-ref="komectbackend" key="komectbackend"></entry>
    	            <entry value-ref="komecthealth" key="komecthealth"></entry>
    	        </map>  
    	    </property>  
    	    <property name="defaultTargetDataSource" ref="komectbackend" />
    	</bean>  
        
    	<bean id="sessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
            <property name="dataSource" ref="dynamicDataSource" />
            <property name="configLocation" value="classpath:mybatis.xml" />
            <property name="mapperLocations" value="classpath*:sqlmap/**/*.xml" />
    		<property name="plugins">
    			<bean
    				class="com.cmcc.akso.plugin.database.page.PaginationInterceptor">
    				<property name="properties">
    					<props>
    						<prop key="dialect">${jdbc.type}</prop>
            			</props>
    				</property>
    			</bean>
    		</property>
        </bean>
         
        <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
            <property name="basePackage" value="com.cmcc.komectcms.dao" />
            <property name="sqlSessionFactoryBeanName" value="sessionFactory"/>
        </bean>
        
        
        <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
            <property name="dataSource" ref="dynamicDataSource" />
        </bean>
        <aop:config>
    		<aop:aspect id="dsAspect" ref="dataSourceInterceptor">
    			<aop:pointcut id="dskommecthealth" expression="execution(* com.cmcc.komectcms.dao.coupon..*(..))" />
    			<aop:before method="setKomecthealth" pointcut-ref="dskommecthealth"/>
    		</aop:aspect>
    	</aop:config>
        
        <tx:annotation-driven transaction-manager="transactionManager" />

二、因为cms配置了komectend默认为数据源，这里只需将komecthealth作为备用的数据源，拦截优惠券相关接口，使用该数据源，配置拦截器如上

三、配置好sqlSessionFactory和事务管理器

四、添加相应的拦截器DataSourceInterceptor，并声明为spring组件

@Component
    public class DataSourceInterceptor {
    
    	public void setKomectbackend(JoinPoint jp) {
    		DatabaseContextHolder.setCustomerType("komectbackend");
    	}
    	
    	public void setKomecthealth(JoinPoint jp) {
    		DatabaseContextHolder.setCustomerType("komecthealth");
    	}
    }

拦截器的方法名要与配置中的一致，在dao.coupon包下的文件执行之前加入切面dsAspect，设置切点为dskomecthealth，即数据源为komecthealth。

五、添加一个线程池：

public class DatabaseContextHolder {
    
    	private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
    
    	public static void setCustomerType(String customerType) {
    		contextHolder.set(customerType);
    	}
    
    	public static String getCustomerType() {
    		return contextHolder.get();
    	}
    
    	public static void clearCustomerType() {
    		contextHolder.remove();
    	}
    }

public class DynamicDataSource extends AbstractRoutingDataSource{
    
    	@Override
    	protected Object determineCurrentLookupKey() {
    		return DatabaseContextHolder.getCustomerType(); 
    	}
    
    }

该功能提交一段时间后，居然诡异的在网站上出现了错误代码，说是调不到komecthealth.t\_sys\_users等表，经排查将aop切点修改为如下就好了，可能是这样更精确点吧

<aop:pointcut id="dskommecthealth" expression="execution(* com.cmcc.komectcms.dao.coupon.CouponCmmsDao.*(..))" />

[Center]: /images/20220711/329c4eb8c50a4a84af69b7eaa618da75.png