Spring+SpringMvc+MybatisPlus+Aop实现多数据源切换

1、需求描述

• 分别获取【server_1,server_2,server_3】数据库中t_person表信息。默认获取server_1数据库数据。

2、效果展示

• 项目启动时
  
• 访问：http://localhost:8080/server_1/resultJson、动态切换数据库并读取Person信息
  
• 访问：http://localhost:8080/server_2/resultJson、动态切换数据库并读取Person信息
  
• 访问：http://localhost:8080/server_3/resultJson、动态切换数据库并读取Person信息
  

3、项目目录简介

4、sql文件介绍

• server_1
  
• server_2
  
• server_3
  

5、核心配置文件代码

• jdbc.properties

#server_1
jdbc1.driver=com.mysql.jdbc.Driver
jdbc1.url=jdbc:mysql://127.0.0.1:3306/server_1?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true&zeroDateTimeBehavior=convertToNull
jdbc1.username=root
jdbc1.password=000

#server_2
jdbc2.driver=com.mysql.jdbc.Driver
jdbc2.url=jdbc:mysql://127.0.0.1:3306/server_2?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true&zeroDateTimeBehavior=convertToNull
jdbc2.username=root
jdbc2.password=00

#server_3
jdbc3.driver=com.mysql.jdbc.Driver
jdbc3.url=jdbc:mysql://127.0.0.1:3306/server_3?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true&zeroDateTimeBehavior=convertToNull
jdbc3.username=root
jdbc3.password=00

• spring-beans.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:context="http://www.springframework.org/schema/context"
       xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
		http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-4.3.xsd">

    <context:component-scan base-package="com.hf.db">
        <context:exclude-filter type="annotation" expression="org.springframework.stereotype.Controller"/>
    </context:component-scan>

    <context:property-placeholder location="classpath:jdbc.properties"/>

    <!--server_1-->
    <bean id="server_1" class="com.alibaba.druid.pool.DruidDataSource">
        <property name="username" value="${jdbc1.username}"/>
        <property name="password" value="${jdbc1.password}"/>
        <property name="url" value="${jdbc1.url}"/>
        <property name="driverClassName" value="${jdbc1.driver}"/>
        <property name="initialSize" value="5"/>
    </bean>

    <!--server_2-->
    <bean id="server_2" class="com.alibaba.druid.pool.DruidDataSource">
        <property name="username" value="${jdbc2.username}"/>
        <property name="password" value="${jdbc2.password}"/>
        <property name="url" value="${jdbc2.url}"/>
        <property name="driverClassName" value="${jdbc2.driver}"/>
        <property name="initialSize" value="5"/>
    </bean>

    <!--server_3-->
    <bean id="server_3" class="com.alibaba.druid.pool.DruidDataSource">
        <property name="username" value="${jdbc3.username}"/>
        <property name="password" value="${jdbc3.password}"/>
        <property name="url" value="${jdbc3.url}"/>
        <property name="driverClassName" value="${jdbc3.driver}"/>
        <property name="initialSize" value="5"/>
    </bean>

    <!--读取当前的返回值key值,选择数据源-->
    <bean id="dataSourceOption" class="com.hf.db.utils.db.DynamicDataSourceUtils">
        <!--默认链接server_1-->
        <property name="defaultTargetDataSource" ref="server_1"/>
        <property name="targetDataSources">
            <map>
                <entry key="server_1" value-ref="server_1"/>
                <entry key="server_2" value-ref="server_2"/>
                <entry key="server_3" value-ref="server_3"/>
            </map>
        </property>
    </bean>

</beans>

• spring-mybatis.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd">

    <!-- 配置用mybatis-plus操作数据库 -->
    <bean id="sqlSessionFactoryBean" class="com.baomidou.mybatisplus.spring.MybatisSqlSessionFactoryBean">
        <!-- 指定数据源 -->
        <property name="dataSource" ref="dataSourceOption"/>
        <property name="configLocation" value="classpath:mybatis/mybatis-config.xml"/>
        <property name="mapperLocations" value="classpath:mybatis/mapper/*.xml"/>
        <property name="typeAliasesPackage" value="com.hf.db.entity"/>
        <property name="globalConfig" ref="globalConfiguration"/>

    </bean>

    <!-- 定义MybatisPlus的全局策略配置-->
    <bean id="globalConfiguration" class="com.baomidou.mybatisplus.entity.GlobalConfiguration">
        <property name="dbColumnUnderline" value="true"/>
        <property name="idType" value="0"/>
        <property name="tablePrefix" value="t_"/>
    </bean>

    <!-- 将所有mapper接口的实现类自动加入到ioc容器中  -->
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.hf.db.dao"/>
    </bean>

</beans>

• spring-aop.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:aop="http://www.springframework.org/schema/aop"
       xmlns:tx="http://www.springframework.org/schema/tx"
       xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
		http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-4.3.xsd
		http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-4.3.xsd">

    <!-- spring配置事务 -->
    <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSourceOption"></property>
    </bean>
    <tx:annotation-driven transaction-manager="transactionManager"/>

    <aop:config>
        <aop:pointcut expression="execution(* com.hf.db.service.*.*(..))" id="txPoint"/>
        <aop:advisor advice-ref="myAdvice" pointcut-ref="txPoint"/>
    </aop:config>

    <tx:advice id="myAdvice" transaction-manager="transactionManager">
        <tx:attributes>
            <tx:method name="*"/>
            <tx:method name="get*" propagation="SUPPORTS" read-only="true"/>
        </tx:attributes>
    </tx:advice>

</beans>

• springMvc.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:context="http://www.springframework.org/schema/context"
       xmlns:mvc="http://www.springframework.org/schema/mvc" xmlns:aop="http://www.springframework.org/schema/aop"
       xsi:schemaLocation="http://www.springframework.org/schema/mvc http://www.springframework.org/schema/mvc/spring-mvc-4.3.xsd
		http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
		http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-4.3.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop.xsd">

    <!--配置只扫描controller -->
    <context:component-scan base-package="com.hf.db.controller" use-default-filters="false">
        <context:include-filter type="annotation" expression="org.springframework.stereotype.Controller"/>
    </context:component-scan>

    <!-- 映射动态资源 -->
    <mvc:annotation-driven/>

    <!-- 映射静态资源 -->
    <mvc:default-servlet-handler/>

    <!-- 启动@aspectj的自动代理支持-->
    <aop:aspectj-autoproxy expose-proxy="true"/>

    <!--允许跨域访问-->
    <mvc:cors>
        <mvc:mapping path="/**"/>
    </mvc:cors>

</beans>

• ChooseDataSource.java

package com.hf.db.aspect;

import java.lang.annotation.*;

/**
 * 功能描述: <br>
 * 〈
 * 动态切换数据源自定义注解类
 * 〉
 *
 * @className: ChooseDataSource
 * @author: hf
 * @version: 1.0.0
 * @date: 2018/12/25 18:21
 */
@Target({ElementType.PARAMETER, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ChooseDataSource {

    /**
     * 数据库名称
     */
    String dataSourceName() default "server_1";
}

• ChooseDataSourceAspect.java

package com.hf.db.aspect;

import com.hf.db.utils.db.DataSourceCheckUtils;
import com.hf.db.utils.db.DynamicDataSourceHolder;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * 功能描述: <br>
 * 〈
 * Aop动态切换数据源的具体实现
 * 〉
 *
 * @className: ChooseDataSource
 * @author: hf
 * @version: 1.0.0
 * @date: 2018/12/25 18:21
 */
@Aspect
@Component
public class ChooseDataSourceAspect {

    /**
     * 功能描述: <br>
     * 〈
     * 配置接入点,切入controller层
     * 〉
     *
     * @className: ChooseDataSourceAspect
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/25 18:25
     * @param: []
     * @return: void
     */
    @Pointcut("execution(* com.hf.db.controller..*.*(..))")
    private void controllerAspect() {
    }

    /**
     * 功能描述: <br>
     * 〈
     * 配置切面动态切换数据源
     * 〉
     *
     * @className: ChooseDataSourceAspect
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/25 18:25
     * @param: [joinPoint 切点]
     * @return: java.lang.Object
     */
    @Around("controllerAspect()")
    public Object permission(ProceedingJoinPoint joinPoint) throws Throwable {
        Object target = joinPoint.getTarget();
        Object[] args = joinPoint.getArgs();
        Method method = getMethod(joinPoint, args);
        //获取数据库名称参数
        ChooseDataSource chooseDataSource = method.getAnnotation(ChooseDataSource.class);
        if (chooseDataSource != null) {
            String dataSourceName = chooseDataSource.dataSourceName();
            //检查数据库名称是否存在
            if (DataSourceCheckUtils.check(dataSourceName)) {
                DynamicDataSourceHolder.putDataSourceName(dataSourceName);
            } else {
                DynamicDataSourceHolder.putDataSourceName("server_1");
            }
        }
        return joinPoint.proceed();
    }

    /**
     * 功能描述: <br>
     * 〈
     * 获取切面信息
     * 〉
     *
     * @className: ChooseDataSourceAspect
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/7 15:28
     * @param: [joinPoint 切点, args 参数]
     * @return: java.lang.reflect.Method
     */
    private Method getMethod(ProceedingJoinPoint joinPoint, Object[] args) throws NoSuchMethodException {
        String methodName = joinPoint.getSignature().getName();
        Class clazz = joinPoint.getTarget().getClass();
        Method[] methods = clazz.getMethods();
        for (Method method : methods) {
            if (methodName.equals(method.getName())) {
                return method;
            }
        }
        return null;
    }
}

• DynamicDataSourceHolder.java

package com.hf.db.utils.db;

/**
 * 功能描述: <br>
 * 〈
 * 解决线程安全问题
 * 〉
 *
 * @className: DynamicDataSourceHolder
 * @author: hf
 * @version: 1.0.0
 * @date: 2018/12/25 18:18
 * <p>
 * History:
 * <author:>          <time:>          <version:>          <desc:>
 */
public class DynamicDataSourceHolder {

    //解决线程安全问题
    private static final ThreadLocal<String> holder = new ThreadLocal<String>();

    public static void putDataSourceName(String dataName) {
        holder.set(dataName);
    }

    public static String getDataSourceName() {
        return holder.get();
    }

    //默认链接server_1服务器
    public static class DataSourceName {
        public final static String dataSource = "server_1";
    }
}

• DataSourceCheckUtils.java

package com.hf.db.utils.db;

import java.util.Arrays;
import java.util.List;

/**
 * 功能描述: <br>
 * 〈
 * 判断数据源是否存在
 * 〉
 *
 * @className: DataSourceCheckUtils
 * @author: hf
 * @version: 1.0.0
 * @date: 2018/12/25 18:16
 */
public class DataSourceCheckUtils {

    static List<String> souList = Arrays.asList("server_1", "server_2", "server_3");

    /**
     * 功能描述: <br>
     * 〈
     * 检查动态切换的数据源是否存在
     * 〉
     *
     * @className: DataSourceCheckUtils
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/25 18:16
     * @param: [dataSourceName 传递过来的数据源名称]
     * @return: boolean
     */
    public static boolean check(String dataSourceName) {
        boolean flag = false;
        if (souList.contains(dataSourceName)) {
            flag = true;
        }
        return flag;
    }
}

• PersonController.java

package com.hf.db.controller;


import com.hf.db.aspect.ChooseDataSource;
import com.hf.db.service.PersonService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * <p>
 * 前端控制器
 * </p>
 *
 * @author hf
 * @since 2018-12-25
 */
@RestController
public class PersonController {

    @Autowired
    PersonService personService;

    /**
     * 功能描述: <br>
     * 〈
     * 动态切换一号数据库【server_1】、获取所有人员信息
     * 〉
     *
     * @className: PersonController
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/25 18:08
     * @param: []
     * @return: java.lang.Object
     */
    @ChooseDataSource()
    @GetMapping("/server_1/resultJson")
    public Object resultJson1() {
        return personService.selectList(null);
    }

    /**
     * 功能描述: <br>
     * 〈
     * 动态切换一号数据库【server_2】、获取所有人员信息
     * 〉
     *
     * @className: PersonController
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/25 18:09
     * @param: []
     * @return: java.lang.Object
     */
    @ChooseDataSource(dataSourceName = "server_2")
    @GetMapping("/server_2/resultJson")
    public Object resultJson2() {

        return personService.selectList(null);
    }

    /**
     * 功能描述: <br>
     * 〈
     * 动态切换一号数据库【server_3】、获取所有人员信息
     * 〉
     *
     * @className: PersonController
     * @author: hf
     * @version: 1.0.0
     * @date: 2018/12/25 18:09
     * @param: []
     * @return: java.lang.Object
     */
    @ChooseDataSource(dataSourceName = "server_3")
    @GetMapping("/server_3/resultJson")
    public Object resultJson3() {

        return personService.selectList(null);
    }
}

7、其它

• 若有不足之处、敬请谅解
• 源码：链接：https://pan.baidu.com/s/1A4QJsrs-6MIkPigVhZk67w 提取码：8cqs
• 本文有较多MybatisPlus配置未做赘述、建议参考:https://blog.csdn.net/weixin_44187730/article/details/85131048