实现Mybatis接口模式下的数据库调用分离

目标：

多项目，多数据库，多连接池，程序级动态切换数据库调用

环境基础：

         框架：SPRING+MYBATIS+MYSQL/ORACLE

设想：

Mapper分包处理不同的库

BaseDao分包处理不同的库

BaseService分包处理不同的库

实现：

多个数据源管理结构：

 配置文件：

<!-- MyBatis begin && Alias-->
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="multipleDataSource"/>
        <property name="typeAliasesPackage" value="com.thinkgem.jeesite,com.qysxy.product"/>
        <!--com.thinkgem.jeesite.common.persistence.BaseEntity-->
        <property name="typeAliasesSuperType" value="com.thinkgem.jeesite.common.persistence.SuperEntity"/>
        <property name="mapperLocations" value="classpath:/mappings/**/*.xml"/>
        <property name="configLocation" value="classpath:/mybatis-config.xml"></property>
    </bean>

    <!--多数据源管理配置-->
    <bean id="multipleDataSource" class="com.qysxy.framework.spring.support.MultipleDataSource">
        <property name="defaultTargetDataSource" ref="dataSource"/>
        <property name="targetDataSources">
            <map>
                <entry key="MYSQL-PROJECT1" value-ref="dataSource"/>
                <entry key="MYSQL-PROJECT2" value-ref="dataSource-project2"/>
            </map>
        </property>
    </bean>


    <!-- 数据源配置, 使用 BoneCP 数据库连接池 -->
    <bean id="dataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> 
        <!-- 数据源驱动类可不写，Druid默认会自动根据URL识别DriverClass -->
        <property name="driverClassName" value="${jdbc.driver}" />
        
        <!-- 基本属性 url、user、password -->
        <property name="url" value="${jdbc.url}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />
        
        <!-- 配置初始化大小、最小、最大 -->
        <property name="initialSize" value="${jdbc.pool.init}" />
        <property name="minIdle" value="${jdbc.pool.minIdle}" /> 
        <property name="maxActive" value="${jdbc.pool.maxActive}" />
        
        <!-- 配置获取连接等待超时的时间 -->
        <property name="maxWait" value="60000" />
        
        <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        
        <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />
        
        <property name="validationQuery" value="${jdbc.testSql}" />
        <property name="testWhileIdle" value="true" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />
        
        <!-- 打开PSCache，并且指定每个连接上PSCache的大小（Oracle使用）
        <property name="poolPreparedStatements" value="true" />
        <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> -->
        
        <!-- 配置监控统计拦截的filters -->
        <property name="filters" value="stat" /> 
    </bean>

    <!-- 数据源配置, 使用 BoneCP 数据库连接池,针对平台插拔系统2 -->
    <bean id="dataSource-project2" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <!-- 数据源驱动类可不写，Druid默认会自动根据URL识别DriverClass -->
        <property name="driverClassName" value="${jdbc.driver}" />

        <!-- 基本属性 url、user、password -->
        <property name="url" value="${jdbc.url.project2}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />

        <!-- 配置初始化大小、最小、最大 -->
        <property name="initialSize" value="${jdbc.pool.init}" />
        <property name="minIdle" value="${jdbc.pool.minIdle}" />
        <property name="maxActive" value="${jdbc.pool.maxActive}" />

        <!-- 配置获取连接等待超时的时间 -->
        <property name="maxWait" value="60000" />

        <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />

        <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />

        <property name="validationQuery" value="${jdbc.testSql}" />
        <property name="testWhileIdle" value="true" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />

        <!-- 打开PSCache，并且指定每个连接上PSCache的大小（Oracle使用）
        <property name="poolPreparedStatements" value="true" />
        <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> -->

        <!-- 配置监控统计拦截的filters -->
        <property name="filters" value="stat" />
    </bean>

<!--切面注解驱动-->
    <aop:aspectj-autoproxy proxy-target-class="false"/>

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
 * @author fuguangli
 * @description 数据库选择路由
 * @Create date:    2017/2/24
 */
public class MultipleDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DatabaseContextHolder.getDataBaseType();
    }
}

public class DatabaseContextHolder {
    public static String DATABASE_MYSQL_PROJECT1 = "MYSQL-PROJECT1";//默认
    public static String DATABASE_MYSQL_PROJECT2 = "MYSQL-PROJECT2";

    private static final ThreadLocal<String> contextHolder = new ThreadLocal();

    public static void setDataBaseType(String context) {
        contextHolder.set(context);
    }

    public static String getDataBaseType() {
        if (contextHolder.get() == null) {
            return DATABASE_MYSQL_PROJECT1;
        }

        return contextHolder.get();
    }

    public static void clearDatabaseType() {
        contextHolder.remove();
    }

}

@Test
    public void changeDataSource() {
        Long time  = System.currentTimeMillis();
        System.out.println("start ="+time);
        DatabaseContextHolder.setDataBaseType(DatabaseContextHolder.DATABASE_MYSQL_PROJECT2);
        List<Object> list = testService.findBySql("select * from jee_user");
        System.out.println("end ="+(System.currentTimeMillis()-time));//185-126
    }

此时，可以进行切换数据库，但是还没有做到真正的程序自动切换，此时利用spring的AOP进行分包检测，自动切换数据源。

 

原理：

目前，spring AOP的使用条件：

1、代理目标必须是实现类

2、代理的方法是实现的接口的方法

3、程序需要调用实现的接口方法

4、代理的类的所有上层接口不能有过代理

 

还有，在本次框架中需要注意的是，mybatis采用的接口调用，所以mybatis框架对DAO接口已经进行了一次java反射代理，所以，为了动态的切换数据源，需要重新写一个接口和实现类调用DAO方法，然后对此类进行代理。

 

此时有两种方法实现代理：

一、代理mybatis的DAO接口

基本实现就是，你需要写一个接口和实现类去封装mybatis的调用接口，很像是我们经常做的service层。

import com.qysxy.framework.spring.support.DatabaseContextHolder;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.*;
import org.springframework.stereotype.Component;

import java.lang.annotation.Annotation;

/**
 * @author fuguangli
 * @description 切面通知类
 * @Create date:    2017/2/24
 */
@Component("dataSourceAdvice ")
@Aspect
public class DataSourceAdvice{

    @Pointcut("execution(*com.thinkgem.jeesite.modules.test.service.*Service.*(..))")
    private   void pointcut(){}

//    @Pointcut("@annotation(daoAOP)")
//    private   void pointcut(DaoAOP daoAOP){}

    /**
     * 在调用方法执行前执行，不能阻止方法的调用。
     * @param joinPoint
     */
    @Before("pointcut()")
    public void doBefore(JoinPoint joinPoint) {
       DatabaseContextHolder.setDataBaseType(DatabaseContextHolder.DATABASE_MYSQL_PROJECT2);

        System.out.println("--------调用数据库之前切换设置数据源--------");
    }
}

二、自定义注解，实现切面

其实原理和上面是一样的，也是需要写一层去封装mybatis调用接口，然后做切面配置的时候，加上自定义注解。

 

贴上自定义注解：

import java.lang.annotation.*;

/**
 * @author fuguangli
 * @description 切面注解
 * @Create date:    2017/2/27
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DaoAOP {
    String value() default "";
}