随着业务的发展,用户量的增大,单表单库已经满足不了我们的业务需求,因此需要分库分表,但是随之带来的问题是当我们需要从库中捞取某些数据的时候,经常需要在多个数据库来回切换,如何做到无缝切换,这里主要通过配置数据源,自定义注解和aop技术来解决。主要包含以下4步:
1、切换数据源
例如我们有两个数据源,一个是ds0,一个是ds1,我们可以通过如下方式配置一个动态数据源。我们只需要自定义一个类继承AbstractRoutingDatasource,实现determineCurrentLookupKey方法,这个方法可以决定我们每次使用的数据源。contextHolder则是一个线程局部变量,保证线程的安全。也就是只要每次在准备切换数据库的时候,通过contextHolder.setDatabaseName就可以切换数据源了。
/** * 功能:数据库切换工具类 * 版本:1.0 * 日期:2016/12/2 13:44 * 作者:屠苏 */ public class DataSourceContextHolder { // 线程局部变量,用于保存保存数据库的名称 private static final ThreadLocal<String> contextHolder = new InheritableThreadLocal<String>(); public static void setDataBaseName(String dataBaseName) { contextHolder.set(dataBaseName); } public static String getDataBaseName() { return contextHolder.get(); } public static void clearDataBaseName() { contextHolder.remove(); } }
/** * 功能:动态数据源 * 版本:1.0 * 日期:2016/12/2 13:49 * 作者:屠苏 */ public class DynamicDataSource extends AbstractRoutingDataSource { // 返回数据源的key @Override protected Object determineCurrentLookupKey() { return DataSourceContextHolder.getDataBaseName(); } }
<!--配置动态数据源--> <bean id="dataSource" class="com.netease.payment.util.datasource.DynamicDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <entry key="ds0" value-ref="dataSource0"/> <entry key="ds1" value-ref="dataSource1"/> </map> </property> <property name="defaultTargetDataSource" ref="dataSource0"/> </bean>
2、自定义注解
首先做一个自定义注解,然后每次在需要切数据源的方法上加上@Datasource("xx")就可以了。
/** * 描述: 自定义数据源注解 * 版本号: 1.0 * 日期: 2017/2/21 11:49 * 作者: 屠苏 */ @Target(ElementType.METHOD) @Retention(RetentionPolicy.RUNTIME) @Documented @Inherited public @interface Datasource { String value() default "ds0"; }
3、Aop
首先定义一个切入点,对所有存在@Datasource的注释的方法进行切面,然后获取签名方法,获取方法上的注释,替换数据源。
package com.elin4it.ssm.aop; import com.elin4it.ssm.annotation.Datasource; import org.aspectj.lang.JoinPoint; import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.Signature; import org.aspectj.lang.annotation.*; import org.aspectj.lang.reflect.MethodSignature; import org.springframework.stereotype.Component; import java.lang.reflect.Method; /** * 功能:切换数据源aop * 版本:1.0 * 日期:2016/12/29 14:18 * 作者:屠苏 */ @Aspect @Component public class DsAop { /** * 第一个 * 代表所有的返回值类型 * 第二个 * 代表所有的类 * 第三个 * 代表类所有方法 * 最后一个 .. 代表所有的参数 */ @Pointcut("@annotation(com.elin4it.ssm.annotation.Datasource)") private void anyMethod(){}//定义一个切入点 @Before("anyMethod()") public void before(JoinPoint joinPoint){ Object target = joinPoint.getTarget(); Class<?> classObj = target.getClass(); // 获取连接点的签名方法对象 Signature signature = joinPoint.getSignature(); System.out.println("signature method=" + signature.getName()); // 方法名 String methodName = joinPoint.getSignature().getName(); // 方法参数类型 Class<?>[] paraClasses = ((MethodSignature)joinPoint.getSignature()).getMethod().getParameterTypes(); try { Method method = classObj.getMethod(methodName, paraClasses); if(method != null && method.isAnnotationPresent(Datasource.class)) { Datasource dataSource = method.getAnnotation(Datasource.class); /** * 切换数据源 !!!!!!! */ DataSourceContextHolder.setDataBaseName(dataSource.value()); System.out.println("value=" + dataSource.value()); } } catch (Exception e) { e.printStackTrace(); } } @After("anyMethod()") public void after(){} @AfterThrowing("anyMethod()") public void exception(){} @Around("anyMethod()") public Object doBasicProfiling(ProceedingJoinPoint pjp) throws Throwable { Object object = pjp.proceed();//执行该方法 return object; } }
4、调用切换数据源的方法
@Datasource("ds1") public void testSwitchDS() { System.out.println("调用testSwitchDS方法!"); }