首页 > 代码库 > 利用Spring的AbstractRoutingDataSource解决多数据源的问题

利用Spring的AbstractRoutingDataSource解决多数据源的问题

2024-10-17 00:02:41   215人阅读

多数据源问题很常见，例如读写分离数据库配置。

原来的项目出现了新需求，局方要求新增某服务器用以提供某代码，涉及到多数据源的问题。

研究成果如下：

1、首先配置多个datasource

 

[html] view plain copy
 
  1. <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">  
  2.         <property name="driverClassName" value=http://www.mamicode.com/"net.sourceforge.jtds.jdbc.Driver">  
  3.         </property>  
  4.         <property name="url" value=http://www.mamicode.com/"jdbc:jtds:sqlserver://10.82.81.51:1433;databaseName=standards">  
  5.         </property>  
  6.         <property name="username" value=http://www.mamicode.com/"youguess"></property>  
  7.         <property name="password" value=http://www.mamicode.com/"youguess"></property>  
  8.     </bean>  
  9.     <bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource">  
  10.         <property name="driverClassName" value=http://www.mamicode.com/"net.sourceforge.jtds.jdbc.Driver">  
  11.         </property>  
  12.         <property name="url" value=http://www.mamicode.com/"jdbc:jtds:sqlserver://10.82.81.52:1433;databaseName=standards">  
  13.         </property>  
  14.         <property name="username" value=http://www.mamicode.com/"youguess"></property>  
  15.         <property name="password" value=http://www.mamicode.com/"youguess"></property>  
  16. </bean>  



 

2、写一个DynamicDataSource类继承AbstractRoutingDataSource，并实现determineCurrentLookupKey方法

 

[java] view plain copy
 
  1. package com.standard.core.util;  
  2. import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;  
  3. public class DynamicDataSource extends AbstractRoutingDataSource {  
  4.     @Override  
  5.     protected Object determineCurrentLookupKey() {  
  6.         return CustomerContextHolder.getCustomerType();  
  7.     }  
  8. }  



 

 

3、利用ThreadLocal解决线程安全问题

 

[java] view plain copy
 
  1. package com.standard.core.util;  
  2. public class CustomerContextHolder {  
  3.     public static final String DATA_SOURCE_A = "dataSource";  
  4.     public static final String DATA_SOURCE_B = "dataSource2";  
  5.     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
  6.     public static void setCustomerType(String customerType) {  
  7.         contextHolder.set(customerType);  
  8.     }  
  9.     public static String getCustomerType() {  
  10.         return contextHolder.get();  
  11.     }  
  12.     public static void clearCustomerType() {  
  13.         contextHolder.remove();  
  14.     }  
  15. }  


4、数据源配置

 

 

[html] view plain copy
 
  1. <bean id="dynamicDataSource" class="com.standard.core.util.DynamicDataSource" >  
  2.         <property name="targetDataSources">  
  3.             <map key-type="java.lang.String">  
  4.                 <entry value-ref="dataSource" key="dataSource"></entry>  
  5.                 <entry value-ref="dataSource2" key="dataSource2"></entry>  
  6.             </map>  
  7.         </property>  
  8.         <property name="defaultTargetDataSource" ref="dataSource" >  
  9.         </property>  
  10.     </bean>   


5、在DAOImpl中切换数据源

 

 

[java] view plain copy
 
  1. CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_B);   


搞定！

利用Spring的AbstractRoutingDataSource解决多数据源的问题


联系
我们