利用AbstractRoutingDataSource实现动态数据源切换

╰半橙微兮° 2022-08-10 03:34 6阅读 0赞

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

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

研究成果如下：

1、首先配置多个datasource

**\[html\]**  [view plain][] [copy][view plain]

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方法

**\[java\]**  [view plain][] [copy][view plain]

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][view plain]

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][view plain]

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][view plain]

1.  CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA\_SOURCE\_B);

From: http://blog.csdn.net/alaahong/article/details/8707915

另外参考文章：

Spring中容易被忽视但有用的AbstractRoutingDataSource （根据语言确定数据源）

http://jackyrong.iteye.com/blog/2001003

//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

## **利用AbstractRoutingDataSource实现动态数据源切换** ##

在Spring 2.0.1中引入了AbstractRoutingDataSource, 该类充当了DataSource的路由中介, 能有在运行时, 根据某种key值来动态切换到真正的DataSource上。

Spring动态配置多数据源，即在大型应用中对数据进行切分，并且采用多个数据库实例进行管理，这样可以有效提高系统的水平伸缩性。而这样的方案就会不同于常见的单一数据实例的方案，这就要程序在运行时根据当时的请求及系统状态来动态的决定将数据存储在哪个数据库实例中，以及从哪个数据库提取数据。

Spring对于多数据源，以数据库表为参照，大体上可以分成两大类情况：   
一是，表级上的跨数据库。即，对于不同的数据库却有相同的表（表名和表结构完全相同）。   
二是，非表级上的跨数据库。即，多个数据源不存在相同的表。   
Spring2.x的版本中采用Proxy模式，就是我们在方案中实现一个虚拟的数据源，并且用它来封装数据源选择逻辑，这样就可以有效地将数据源选择逻辑从Client中分离出来。Client提供选择所需的上下文（因为这是Client所知道的），由虚拟的DataSource根据Client提供的上下文来实现数据源的选择。   
具体的实现就是，虚拟的DataSource仅需继承AbstractRoutingDataSource实现determineCurrentLookupKey（）在其中封装数据源的选择逻辑。

一、原理

首先看下AbstractRoutingDataSource类结构，继承了AbstractDataSource