1.首先需要一个 数据源类型的类。
public enum DBType {
dataSource, frontDataSource;
}
2.需要一个本地线程变量对象保存 数据源类型,这里使用的 是ThreadLocal,因为ThreadLocal 可以为每个线程单独创立一个新的副本变量。
public class ContextHolder {
private static final ThreadLocal<Object> holder = new ThreadLocal<Object>();
public static void setDbType(DBType dbType) {
holder.set(dbType);
}
public static DBType getDbType() {
return (DBType) holder.get();
}
public static void clearDbType() {
holder.remove();
}
}
3.需要一个动态数据源类用于代替之前的 dataSource bean,这个类需要继承AbstractRoutingDataSource,并且重写 determineCurrentLookupKey 方法,返回本地线程变量中保存的 数据源类型key。
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
DBType key = ContextHolder.getDbType();//获得当前数据源标识符
System.out.println("当前数据源 :" + key);
return key;
}
}
4.需要一个拦截器,该拦截器可以根据实际情况进行配置,我这里是拦截 service根据方法名称去切换数据源,这里切记:
a.@Order(1) 这个值必须有,因为数据源的获取也是通过拦截器中获取的,这个地方拦截器优先级必须高过数据源获取的拦截器。
b.@Aspect 注解 需要在application.xml 配置启用
<context:component-scan base-package="com">
<context:exclude-filter type="annotation" expression="org.springframework.stereotype.Controller" />
</context:component-scan>
<!-- -开启 aspect 注解使用 -->
<aop:aspectj-autoproxy />
import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
/**
* aop实现动态切换数据源
*
*/
@Aspect
@Component
@Order(1)
public class DataSourceAopProxy {
@Before("dataSourcePoint()")
public void setDataSource(JoinPoint joinPoint){
Signature signature = joinPoint.getSignature();
MethodSignature methodSignature = (MethodSignature) signature;
Method method = methodSignature.getMethod();
//此处进行切换数据源 根据业务条件 获取数据源id
if(method.getName().contains("Front")){
ContextHolder.setDbType(DBType.frontDataSource);
System.out.println("设置当前数据源 DBType.frontDataSource");
}else{
ContextHolder.setDbType(DBType.dataSource);
System.out.println("设置当前数据源 DBType.dataSource");
}
}
@After("dataSourcePoint()")
public void after(JoinPoint joinPoint){
System.out.println("清除 DB type!");
ContextHolder.clearDbType();
}
//@Pointcut("execution(* com.dao.impl.*.*(..))")
@Pointcut("execution(* com.service.impl.*.*(..))")
private void dataSourcePoint(){
}
}
5.需要配置两个数据源和上面的数据源类型对应,我这里${jdbc.url}是读取的属性文件,你们可以在这里直接写值
<bean id="dataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">
<property name="driverClass" value="${jdbc.driver}" />
<property name="jdbcUrl" value="${jdbc.url}" />
<property name="user" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
<property name="initialPoolSize" value="${connection_pools.initial_pool_size}" />
<property name="minPoolSize" value="${connection_pools.min_pool_size}" />
<property name="maxPoolSize" value="${connection_pools.max_pool_size}" />
<property name="maxIdleTime" value="${connection_pools.max_idle_time}" />
<property name="acquireIncrement" value="${connection_pools.acquire_increment}" />
<property name="checkoutTimeout" value="${connection_pools.checkout_timeout}" />
</bean>
<bean id="frontDataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">
<property name="driverClass" value="${front.jdbc.driver}" />
<property name="jdbcUrl" value="${front.jdbc.url}" />
<property name="user" value="${front.jdbc.username}" />
<property name="password" value="${front.jdbc.password}" />
<property name="initialPoolSize" value="${connection_pools.initial_pool_size}" />
<property name="minPoolSize" value="${connection_pools.min_pool_size}" />
<property name="maxPoolSize" value="${connection_pools.max_pool_size}" />
<property name="maxIdleTime" value="${connection_pools.max_idle_time}" />
<property name="acquireIncrement" value="${connection_pools.acquire_increment}" />
<property name="checkoutTimeout" value="${connection_pools.checkout_timeout}" />
</bean>
6.需要 修改 bean为entityManagerFactory的配置,其中将dataSource属性配置为 mysqlDynamicDataSource,如下:
<property name="dataSource" ref="mysqlDynamicDataSource" />
7.最后一步,配置 bean mysqlDynamicDataSource,对应上一步。
a.此处bean中targetDataSources 为目标数据源,key-type 值为数据源类型类,也就是第一条那个枚举。
b.map 中的key 对应的是数据源类型中的 两个枚举值,value-ref 则是第五条 配置的两个bean。
c.defaultTargetDataSource 为默认的数据源类型,就是本地线程变量中值 为null而默认的数据源对象。
<bean id="mysqlDynamicDataSource" class="com.titanos.dynamic.DynamicDataSource">
<property name="targetDataSources">
<!-- 标识符类型 -->
<map key-type="com.dynamic.DBType">
<entry key="dataSource" value-ref="dataSource"/>
<entry key="frontDataSource" value-ref="frontDataSource"/>
</map>
</property>
<property name="defaultTargetDataSource" ref="dataSource"/>
</bean>