转:
利用Spring的AbstractRoutingDataSource解决多数据源的问题
多数据源问题很常见,例如读写分离数据库配置。
原来的项目出现了新需求,局方要求新增某服务器用以提供某代码,涉及到多数据源的问题。
解决方法如下:
1、首先配置多个datasource
1 <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource"> 2 <property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver"> 3 </property> 4 <property name="url" value="jdbc:jtds:sqlserver://10.82.81.51:1433;databaseName=standards"> 5 </property> 6 <property name="username" value="youguess"></property> 7 <property name="password" value="youguess"></property> 8 </bean> 9 <bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource"> 10 <property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver"> 11 </property> 12 <property name="url" value="jdbc:jtds:sqlserver://10.82.81.52:1433;databaseName=standards"> 13 </property> 14 <property name="username" value="youguess"></property> 15 <property name="password" value="youguess"></property> 16 </bean>
2、写一个DynamicDataSource类继承AbstractRoutingDataSource,并实现determineCurrentLookupKey方法
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解决线程安全问题
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、数据源配置
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中切换数据源
1 CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_B);