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中配置,如下:

testDataSource1">

这样配置两个数据源对应的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.设置数据库事务切面和切换数据库切面执行的顺序,如下:

advice-ref="txAdvice" order="2" />

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

欢迎工作一到五年的Java工程师朋友们加入Java架构开发:855801563 获取更多免费视频教程。

合理利用自己每一分每一秒的时间来学习提升自己,不要再用"没有时间“来掩饰自己思想上的懒惰!趁年轻,使劲拼,给未来的自己一个交代

你可能感兴趣的:(Spring 配置多个数据源,并实现动态切换)