数据源的无缝切换

随着业务的发展，用户量的增大，单表单库已经满足不了我们的业务需求，因此需要分库分表，但是随之带来的问题是当我们需要从库中捞取某些数据的时候，经常需要在多个数据库来回切换，如何做到无缝切换，这里主要通过配置数据源，自定义注解和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方法！");
}

转载于:https://my.oschina.net/u/3203060/blog/842939