最近项目需要切换数据源，查询了相关资料后做了一个简单的数据源的切换。
一、基本的简单配置
首先是我认为比较好的切换是在service层进行切换，基本思想是利用spring的AbstractRoutingDataSource类进行datasource的选择，就像map的key-value,AbstractRoutingDataSource是有个determineCurrentLookupKey选择datasource的bean进行切换。
下面是java类的配置
代码语言：javascript复制import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContextHolder.getDataSource();
}
}/**数据源的管理类**/
代码语言：javascript复制public class DataSourceContextHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
public static final String DEFAULT_DATA_SOURCE="defaultDataSource"; //默认数据源
public static final String READ_DATA_SOURCE="readDataSource"; //第二个数据源，因为这个数据源是只提供读取，所以叫read...
/**
* 设置数据源类型
* @param dataSource 数据源名称
*/
public static void setDataSource(String dataSource) {
contextHolder.set(dataSource);
}
/**
* 获取数据源名称
*/
public static String getDataSource() {
return contextHolder.get();
}
/**
* 清除数据源名称
*/
public static void clearDataSource() {
contextHolder.remove();
}
}接下来要在xml文件中配置动态的数据源
部分配置文件如下：
代码语言：javascript复制<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dynamicDataSource"/>
<property name="configLocation" value="classpath:mybatis-config.xml"/>
<property name="mapperLocations" value="classpath*:/com/test/core/domain/mapper/**/*.xml"/>
</bean>
<bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">
<property name="locations">
<list>
<value>classpath:/properties/app-${env}.properties</value>
</list>
</property>
</bean>
<!-- 数据源1-->
<bean id="defaultDataSource" class="com.zaxxer.hikari.HikariDataSource" destroy-method="close">
<property name="driverClassName" value="${datasource.classname}"/>
<property name="jdbcUrl" value="${jdbc.url}"/>
<property name="username" value="${username}"/>
<property name="password" value="${password}"/>
<property name="connectionTimeout" value="${connection.timeout}"/>
<property name="maximumPoolSize" value="${max.pool.size}"/>
<property name="leakDetectionThreshold" value="${leak.detection.threshold}"/>
</bean>
<!-- 数据源2-->
<bean id="readDataSource" class="com.zaxxer.hikari.HikariDataSource" destroy-method="close">
<property name="driverClassName" value="${read.datasource.classname}"/>
<property name="jdbcUrl" value="${read.jdbc.url}"/>
<property name="username" value="${read.username}"/>
<property name="password" value="${read.password}"/>
<property name="connectionTimeout" value="${read.connection.timeout}"/>
<property name="maximumPoolSize" value="${read.max.pool.size}"/>
<property name="leakDetectionThreshold" value="${read.leak.detection.threshold}"/>
</bean>
<!-- 动态数据源-->
<bean id="dynamicDataSource" class="com.test.datasource.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry value-ref="defaultDataSource" key="defaultDataSource"/>
<entry value-ref="readDataSource" key="readDataSource"/>
</map>
</property>
<!-- 默认数据源为defaultDataSource-->
<property name="defaultTargetDataSource" ref="defaultDataSource"/>
</bean>
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="dynamicDataSource"/>
</bean>
<tx:annotation-driven mode="proxy" transaction-manager="transactionManager"/>
<bean id="mapperScannerConfigurer" class="org.mybatis.spring.mapper.MapperScannerConfigurer">
<property name="basePackage" value="com.test.core.domain.mapper"/>
<property name="annotationClass" value="org.springframework.stereotype.Repository"/>
</bean>做好了这些工作就可以在dao层进行数据源的切换了，
如：DataSourceContextHolder.setDataSource(DataSourceContextHolder.READ_DATA_SOURCE);
二、利用spring的aop实现在service切换数据源
然而我们需要在service层或者controller层进行切换，而在进入service时已经开启了一个事务，这时候直接切换数据源是无效的，所以要在service前就切换数据源。
利用spring 的aop来实现我们的想法吧
切换数据源的java类：
/**切面**/
代码语言：javascript复制import lombok.extern.slf4j.Slf4j;
import org.springframework.aop.AfterReturningAdvice;
import org.springframework.aop.MethodBeforeAdvice;
import java.lang.reflect.Method;
@Slf4j
public class DataSourceExchange implements MethodBeforeAdvice,AfterReturningAdvice{
@Override
public void afterReturning(Object returnValue, Method method, Object[] args, Object target) throws Throwable {
DataSourceContextHolder.clearDataSource();
log.info("del datasource");
}
@Override
public void before(Method method, Object[] args, Object target) throws Throwable {
if (method.isAnnotationPresent(DataSource.class))
{
DataSource datasource = method.getAnnotation(DataSource.class);
DataSourceContextHolder.setDataSource(datasource.name());
}
else
{
DataSourceContextHolder.setDataSource(DataSourceContextHolder.DEFAULT_DATA_SOURCE);
}
log.info("now datasource:" DataSourceContextHolder.getDataSource());
}
}/**为了更好的体验，直接用注解来定义数据源**/
代码语言：javascript复制@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
String name() default "defaultDataSource";
}xml的相关配置
代码语言：javascript复制<!--1.配置Spring框架自身提供的切面类-->
<tx:advice id="userTxAdvice" transaction-manager="transactionManager">
<tx:attributes>
<tx:method name="delete*" propagation="REQUIRED" read-only="false"
rollback-for="java.lang.Exception" no-rollback-for="java.lang.RuntimeException" />
<tx:method name="insert*" propagation="REQUIRED" read-only="false"
rollback-for="java.lang.Exception" />
<tx:method name="update*" propagation="REQUIRED" read-only="false"
rollback-for="java.lang.Exception" />
<tx:method name="find*" propagation="SUPPORTS" />
<tx:method name="get*" propagation="SUPPORTS" />
<tx:method name="select*" propagation="SUPPORTS" />
</tx:attributes>
</tx:advice>
<!-- 2.配置自己定义的切面类 -->
<bean id="dataSourceExchange" class="com.chejianer.admin.datasource.DataSourceExchange"/>
<!-- 3.根据切面动态切换数据源-->
<aop:config>
<aop:pointcut id="service" expression="execution(* com.test.service.*.*(..))" />
<aop:advisor pointcut-ref="service" advice-ref="userTxAdvice" order="2" />
<aop:advisor pointcut-ref="service" advice-ref="dataSourceExchange" order="1" />
</aop:config>配置完成后就可以进行测试了，在serviceImpl的的方法上面加上@DataSource(name=你要切换的数据源)就ok了，这里就不贴代码了。
三、直接在controller进行切换，继续探索中
因为项目需要切数据源的地方特别多，在service层切换很费时间，所以准备在controller层进行切换，而尝试了下spring的切面在这里直接切不了，具体原因还没找到，所以暂时只能采用笨的方法，在controller的方法开头加上：
DataSourceContextHolder.setDataSource(DataSourceContextHolder.READ_DATA_SOURCE);
方法结尾加上：
DataSourceContextHolder.clearDataSource();
2017-03-25
发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/100317.html原文链接：https://javaforall.cn
