AbstractRoutingDataSource实现动态数据源切换

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

• 博客分类： 
• spring

spring 

最近要为公司多个游戏做类似的统计功能，考虑到模块的复用性，决定做个动态数据源，根据不同的游戏参数切换不同的datasource。

网上查了下，spring2.0以后增加了AbstractRoutingDataSource这个东西。下面是实现方法

 

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

Java代码  

1. public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {  
2.   
3. }  

 

既然是AbstractDataSource，当然就是javax.sql.DataSource的子类，于是我们自然地回去看它的getConnection方法：

Java代码  

1. public Connection getConnection() throws SQLException {  
2.     return determineTargetDataSource().getConnection();  
3. }  

 原来奥妙就在determineTargetDataSource()里：

Java代码  

1. /** 
2.  * Retrieve the current target DataSource. Determines the 
3.  * {@link #determineCurrentLookupKey() current lookup key}, performs 
4.  * a lookup in the {@link #setTargetDataSources targetDataSources} map, 
5.  * falls back to the specified 
6.  * {@link #setDefaultTargetDataSource default target DataSource} if necessary. 
7.  * @see #determineCurrentLookupKey() 
8.  */  
9. protected DataSource determineTargetDataSource() {  
10.     Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");  
11.     Object lookupKey = determineCurrentLookupKey();  
12.     DataSource dataSource = (DataSource) this.resolvedDataSources.get(lookupKey);  
13.     if (dataSource == null) {  
14.         dataSource = this.resolvedDefaultDataSource;  
15.     }  
16.     if (dataSource == null) {  
17.         throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");  
18.     }  
19.     return dataSource;  
20. }  

这里用到了我们需要进行实现的抽象方法determineCurrentLookupKey()，该方法返回需要使用的DataSource的key值，然后根据这个key从resolvedDataSources这个map里取出对应的DataSource，如果找不到，则用默认的resolvedDefaultDataSource

 

Java代码  

1. <bean id="onlineDynamicDataSource" class="com.xx.stat.base.dynamic.DynamicDataSource">  
2.    <property name="targetDataSources">     
3.       <map key-type="java.lang.String">     
4.          <entry key="xx" value-ref="dataSourceXX"/>     
5.          <entry key="yy" value-ref="dataSourceYY"/>     
6.       </map>     
7.    </property>     
8.    <property name="defaultTargetDataSource" ref="dataSource"/>    
9. </bean>  

 

观察上面的配置文件，发现我们配置的是targetDataSources和defaultTargetDataSource

 

Java代码  

1. public void afterPropertiesSet() {  
2.     if (this.targetDataSources == null) {  
3.         throw new IllegalArgumentException("targetDataSources is required");  
4.     }  
5.     this.resolvedDataSources = new HashMap(this.targetDataSources.size());  
6.     for (Iterator it = this.targetDataSources.entrySet().iterator(); it.hasNext();) {  
7.         Map.Entry entry = (Map.Entry) it.next();  
8.         Object lookupKey = resolveSpecifiedLookupKey(entry.getKey());  
9.         DataSource dataSource = resolveSpecifiedDataSource(entry.getValue());  
10.         this.resolvedDataSources.put(lookupKey, dataSource);  
11.     }  
12.     if (this.defaultTargetDataSource != null) {  
13.         this.resolvedDefaultDataSource = resolveSpecifiedDataSource(this.defaultTargetDataSource);  
14.     }  
15. }  

 

下面就是我们自己实现的子类DynamicDataSource

Java代码  

1. 1. import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;  
   2. public class DynamicDataSource extends AbstractRoutingDataSource {  
   3.     @Override  
   4.     protected Object determineCurrentLookupKey() {  
   5.         return CustomerContextHolder.getCustomerType();  
   6.     }  
   7. }  

 

CustomerContextHolder 

Java代码  

1. 1. public class CustomerContextHolder {  
   2.     public static final String DATA_SOURCE_A = "dataSource";  
   3.     public static final String DATA_SOURCE_B = "dataSource2";  
   4.     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
   5.     public static void setCustomerType(String customerType) {  
   6.         contextHolder.set(customerType);  
   7.     }  
   8.     public static String getCustomerType() {  
   9.         return contextHolder.get();  
   10.     }  
   11.     public static void clearCustomerType() {  
   12.         contextHolder.remove();  
   13.     }  
   14. }  

在需要操作数据库的时候加入如下代码即可       

 DataSourceContextHolder.setDbType(“指定数据源”);