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

2024-07-13 10:04:58

字体：大中小

来源：转载

供稿：网友

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

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

研究成果如下：

1、首先配置多个datasource

<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">     <property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver">     </property>     <property name="url" value="jdbc:jtds:sqlserver://10.82.81.51:1433;databaseName=standards">     </property>     <property name="username" value="youguess"></property>     <property name="password" value="youguess"></property>   </bean>   <bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource">     <property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver">     </property>     <property name="url" value="jdbc:jtds:sqlserver://10.82.81.52:1433;databaseName=standards">     </property>     <property name="username" value="youguess"></property>     <property name="password" value="youguess"></property> </bean>

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

package com.standard.core.util; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class DynamicDataSource extends AbstractRoutingDataSource {   @Override   protected Object determineCurrentLookupKey() {     return CustomerContextHolder.getCustomerType();   } }

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

package com.standard.core.util; public class CustomerContextHolder {   public static final String DATA_SOURCE_A = "dataSource";   public static final String DATA_SOURCE_B = "dataSource2";   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();   } }

4、数据源配置

<bean id="dynamicDataSource" class="com.standard.core.util.DynamicDataSource" >     <property name="targetDataSources">       <map key-type="java.lang.String">         <entry value-ref="dataSource" key="dataSource"></entry>         <entry value-ref="dataSource2" key="dataSource2"></entry>       </map>     </property>     <property name="defaultTargetDataSource" ref="dataSource" >     </property>   </bean>

5、在DAOImpl中切换数据源

复制代码代码如下:

CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_B);

搞定！

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持VeVb武林网。

注：相关教程知识阅读请移步到JAVA教程频道。

上一篇：java 算法之归并排序详解及实现代码

下一篇：java 使用异常的好处总结