Spring配置多个数据源并实现动态切换示例

1.配置两个不同的数据源，如下（由于项目使用的是druid数据库连接，配置可以会复杂点比较）：

2.定义一个类继承AbstractRoutingDataSource实现determineCurrentLookupKey方法，该方法可以实现数据库的动态切换，如下：

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

3.定义一个可以设置当前线程的变量的工具类，用于设置对应的数据源名称：

public class DataSourceContextHolder { 
  private static final ThreadLocal contextHolder = new ThreadLocal(); 
  /** 
   * @Description: 设置数据源类型 
   * @param dataSourceType 数据库类型 
   * @return void 
   * @throws 
   */ 
  public static void setDataSourceType(String dataSourceType) { 
    contextHolder.set(dataSourceType); 
  } 
   
  /** 
   * @Description: 获取数据源类型 
   * @param 
   * @return String 
   * @throws 
   */ 
  public static String getDataSourceType() { 
    return contextHolder.get(); 
  } 
   
  /** 
   * @Description: 清除数据源类型 
   * @param 
   * @return void 
   * @throws 
   */ 
  public static void clearDataSourceType() { 
    contextHolder.remove(); 
  } 
} 

然后在spring中配置，如下：

这样配置两个数据源对应的key分别为testDataSource1和testDataSource2，默认数据库是testDataSource。

4.完成以上步骤后，如果没有数据库的事务管理，已经可以实现数据库的动态切换了。但是如果涉及到数据库的事务管理，需要在数据库事务开启切换数据库，

否则数据库的切换只能在下次数据库操作时才生效。可以定义一个aop处理类在数据库事务开启之前切换数据库，如下：

public class DataSourceAspect implements MethodBeforeAdvice,AfterReturningAdvice  
{ 
 
  @Override 
  public void afterReturning(Object returnValue, Method method, 
      Object[] args, Object target) throws Throwable { 
    // TODO Auto-generated method stub 
    DataSourceContextHolder.clearDataSourceType(); 
  } 
 
  @Override 
  public void before(Method method, Object[] args, Object target) 
      throws Throwable { 
   
    if (method.isAnnotationPresent(DataSource.class))  
    { 
      DataSource datasource = method.getAnnotation(DataSource.class); 
      DataSourceContextHolder.setDataSourceType(datasource.name()); 
    } 
    else 
    { 
      DataSourceContextHolder.setDataSourceType(SinoConstant.DataSourceType.unityDataSource.toString()); 
    } 
     
  } 
} 

5.设置数据库事务切面和切换数据库切面执行的顺序，如下：

利用aop的order属性设置执行的顺序，这样实现了带事务管理的spring数据库动态切换。

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