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

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

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

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

 

一、原理

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

 public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean

 

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

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

     public Connection getConnection(String username, String password) throws SQLException {
         return determineTargetDataSource().getConnection(username, password);
     }

 

 原来关键就在determineTargetDataSource()里：

 /**
      * Retrieve the current target DataSource. Determines the
      * {@link #determineCurrentLookupKey() current lookup key}, performs
      * a lookup in the {@link #setTargetDataSources targetDataSources} map,
      * falls back to the specified
      * {@link #setDefaultTargetDataSource default target DataSource} if necessary.
      * @see #determineCurrentLookupKey()
      */
     protected DataSource determineTargetDataSource() {
         Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
         Object lookupKey = determineCurrentLookupKey();
         DataSource dataSource = this.resolvedDataSources.get(lookupKey);
         if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
             dataSource = this.resolvedDefaultDataSource;
         }
         if (dataSource == null) {
             throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
         }
         return dataSource;
     }

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

         public void afterPropertiesSet() {
         if (this.targetDataSources == null) {
             throw new IllegalArgumentException("Property 'targetDataSources' is required");
         }
         this.resolvedDataSources = new HashMap(this.targetDataSources.size());
         for (Map.Entry entry : this.targetDataSources.entrySet()) {
             Object lookupKey = resolveSpecifiedLookupKey(entry.getKey());
             DataSource dataSource = resolveSpecifiedDataSource(entry.getValue());
             this.resolvedDataSources.put(lookupKey, dataSource);
         }
         if (this.defaultTargetDataSource != null) {
             this.resolvedDefaultDataSource = resolveSpecifiedDataSource(this.defaultTargetDataSource);
         }
     }

 

二、Spring配置多数据源的方式和具体使用过程

1、建立一个获得和设置上下文环境的类，主要负责改变上下文数据源的名称

 public class DynamicDataSourceHolder {
 // 线程局部变量（多线程并发设计，为了线程安全）
 private static final ThreadLocal<String> contextHolder = new ThreadLocal();

 // 设置数据源类型
 public static void setDataSourceType(String dataSourceType) {
     Assert.notNull(dataSourceType, "DataSourceType cannot be null");
     contextHolder.set(dataSourceType);
 }

 // 获取数据源类型
 public static String getDataSourceType() {
     return (String) contextHolder.get();
 }

 // 清除数据源类型
 public static void clearDataSourceType() {
     contextHolder.remove();
 }

 

2、建立动态数据源类，注意，这个类必须继承AbstractRoutingDataSource，且实现方法 determineCurrentLookupKey，该方法返回一个Object，一般是返回字符串

  public class DynamicDataSource extends AbstractRoutingDataSource {
     @Override
     protected Object determineCurrentLookupKey() {
         return DynamicDataSourceHolder.getDataSourceType();
     }

 }

3、编写spring的配置文件配置多个数据源

  

    
    
        
        
        
        

        
        
        
        
        
        
        
        
        
        
    

    
    
        
        
        
        

        
        
        
        
        
        
        
        
        
        
    
    
    
    
        
        
        
        

        
        
        
        
        
        
        
        
        
        
    

    
        
        
            
        
    

   
    

    
    
        
        
        
            
        
        
            
                ${hibernate.dialect}
                ${hibernate.show_sql}
                ${hibernate.format_sql}
                ${hibernate.hbm2ddl.auto}
                ${hibernate.generate_statistics}


                ${hibernate.cache.use_second_level_cache}
                ${hibernate.cache.use_query_cache}
                ${hibernate.cache.region.factory_class}
                ${net.sf.ehcache.configurationResourceName}
                0
                
                ${hibernate.search.default.indexBase}
            
        
        
            
                com.xxxx.modules.*.entity
            
        
    

4.数据源注解定义

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {

    String value();
}

5、配置注解切换数据源

   
    
    
    
    
        
        
        
        
        
        
            
                com.xxx.modules.cms.web.*Controller.*(..)
                com.xxx.modules.md.web.*Controller.*(..)
            
        
    

6.注解的方法拦截器代码

/**
 * 多数据源动态配置拦截器
 *
 */
public class DataSourceMethodInterceptor implements MethodInterceptor, InitializingBean {

    private Logger logger = LoggerFactory.getLogger(DataSourceMethodInterceptor.class);

    @Override
    public Object invoke(MethodInvocation invocation) throws Throwable {
        Class clazz = invocation.getThis().getClass();
        String className = clazz.getName();
        if (ClassUtils.isAssignable(clazz, Proxy.class)) {
            className = invocation.getMethod().getDeclaringClass().getName();
        }
        String methodName = invocation.getMethod().getName();
        Object[] arguments = invocation.getArguments();
        logger.trace("execute {}.{}({})", className, methodName, arguments);

        invocation.getMethod();
        DataSource classDataSource = ReflectionUtils.getAnnotation(invocation.getThis(), DataSource.class);
        DataSource methodDataSource = ReflectionUtils.getAnnotation(invocation.getMethod(), DataSource.class);
        if(methodDataSource != null){
            DataSourceContextHolder.setDataSourceType(methodDataSource.value());
        }else if(classDataSource != null){
            DataSourceContextHolder.setDataSourceType(classDataSource.value());
        }else {
            DataSourceContextHolder.clearDataSourceType();
        }

        Object result = invocation.proceed();
        return result;
    }

    @Override
    public void afterPropertiesSet() throws Exception {
    }
}

7.使用

    @RequestMapping(value = {"datagrid"})
    @ResponseBody
    @DataSource("DWDataSource")//切换到DATASOURCE_DW数据源
    public Datagrid

完结。。。