springboot+mybatisplus 动态切换数据源——数据库（通过AOP获取head参数动态切换）

写这个博客，是记录一下自己的学习过程。
因为网上信息杂乱不清，所以此博客只做记录和分享作用。
参考博客链接：https://blog.csdn.net/QiuHaoqian/article/details/120479890
再此基础上修改了一些，使得mybatisplus无mapper.xml文件可以使用此代码。
先上效果图：

可以看出这个数据库没有这个帐号。

然后更换请求头Language参数，即可更换数据源-数据库

登录成功，返回数据。

实现，只需要写四个类，外加配置数据库和导入相关jar包

话不多说，先上pom.xml

      <dependency>
          <groupId>com.alibaba</groupId>
          <artifactId>druid-spring-boot-starter</artifactId>
          <version>1.2.3</version>
      </dependency>
      <dependency>
          <groupId>org.springframework.boot</groupId>
          <artifactId>spring-boot-starter-aop</artifactId>
      </dependency>

然后配置文件 application.yml

mybatis-plus:
  configuration:
    map-underscore-to-camel-case: true
spring:
  datasource:
    primary:
    //此处配置自己的第一个数据库
      jdbc-url: jdbc:mysql://localhost:3306/XXXX1?useSSL=false&useUnicode=true&characterEncoding=utf-8&autoReconnect=true&serverTimezone=Asia/Shanghai
      username: root
      password: XXXXX
      driver‐class‐name: com.mysql.cj.jdbc.Driver
    secondary:
    //此处配置自己的第二个数据库
      jdbc-url: jdbc:mysql://localhost:3306/XXXX2?useSSL=false&useUnicode=true&characterEncoding=utf-8&autoReconnect=true&serverTimezone=Asia/Shanghai
      username: root
      password: XXXXX
      driver‐class‐name: com.mysql.cj.jdbc.Driver
server:
  port:
    8081

再上相关类：

DataSourceAspect.java

package com.xxxx.aop;

import com.xxxx.datasource.DataSourceHolder;
import org.apache.commons.lang3.StringUtils;
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.springframework.context.annotation.Configuration;
import org.springframework.core.annotation.Order;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.Objects;


/**
 * 拦截controller方法，从请求头中获取使用的数据库编号
 *
 * @author qyz
 */
@Aspect
@Order(1)
@Configuration
public class DataSourceAspect {

    /**
     * 数据库名称在请求头中的key（从header中取）
     */
    private static final String LANGUAGE = "Language";

    /**
     * 切入点,放在controller的每个方法上进行切入,更新数据源
     */
     //这里填入自己项目的controller路径
    @Pointcut("execution(* com.1这里要改.controller..*.*(..))")
    private void anyMethod() {
    }

    @Before("anyMethod()")
    public void dataSourceChange() {
        // 请求头head中获取对应数据库编号 name=Language
        String Language = ((ServletRequestAttributes) Objects
                .requireNonNull(RequestContextHolder.getRequestAttributes()))
                .getRequest()
                .getHeader(LANGUAGE);
        System.out.println("当前数据源: " + Language);
        if (StringUtils.isBlank(Language)) {
            // TODO 根据业务抛异常
            throw new NullPointerException("请求头中没有" + LANGUAGE);
        }
        // 根据请求头中数据库名称来更改对应的数据源（核心）
        DataSourceHolder.setDataSource(Language);
    }

    @After("anyMethod()")
    public void after() {
        // 数据源重置（必须在请求完成后清空ThreadLocal线程上下文，否则会内存溢出）
        DataSourceHolder.clearDataSource();
    }
}

DataSourceHolder.java

package com.xxxx.datasource;

public class DataSourceHolder {

    /**
     * 线程本地环境 （存储数据库名称）
     */
    private static final ThreadLocal<String> DATA_SOURCES = new ThreadLocal<>();

    /**
     * 设置数据源(动态切换数据源),就是调用这个setDataSource方法
     */
    public static void setDataSource(String customerType) {
        DATA_SOURCES.set(customerType);
    }

    /**
     * 获取数据源
     */
    public static String getDataSource() {
        return DATA_SOURCES.get();
    }

    /**
     * 清除数据源
     */
    public static void clearDataSource() {
        DATA_SOURCES.remove();
    }
}

DynamicDataSource.java

package com.xxxx.datasource;

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

public class DynamicDataSource extends AbstractRoutingDataSource {
    /**
     * 每次请求动态请求哪一个数据源
     *
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceHolder.getDataSource();
    }
}

DynamicDataSourceConfig.java

package com.xxxx.datasource;

import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.MybatisXMLLanguageDriver;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.type.JdbcType;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.mybatis.spring.transaction.SpringManagedTransactionFactory;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;


@Configuration
//改成你项目的mapper路径
@MapperScan(basePackages = "com.2这里也要改.mapper", sqlSessionFactoryRef = "SqlSessionFactory")
public class DynamicDataSourceConfig {

    /**
     * 将第1个数据源对象放入Spring容器中
     *
     * @ConfigurationProperties 读取application.properties中的前缀为spring.datasource.primary的配置参数并映射成为一个对象
     */
    @Bean(name = "dateSource1")
    @ConfigurationProperties(prefix = "spring.datasource.primary")
    public DataSource DateSource1() {
        return DataSourceBuilder.create().build();
    }

    /**
     * 将第2个数据源对象放入Spring容器中
     */
    @Bean(name = "dateSource2")
    @ConfigurationProperties(prefix = "spring.datasource.secondary")
    public DataSource DateSource2() {
        return DataSourceBuilder.create().build();
    }

    /**
     * 将动态代理数据源对象放入Spring容器中
     */
    @Bean(name = "dynamicDataSource")
    public DynamicDataSource DynamicDataSource(@Qualifier("dateSource1") DataSource primaryDataSource,
                                               @Qualifier("dateSource2") DataSource secondaryDataSource) {
        // 这个地方是比较核心的targetDataSource 集合是我们数据库和名字之间的映射
        Map<Object, Object> targetDataSource = new HashMap<>();
        targetDataSource.put("ch", primaryDataSource);
        targetDataSource.put("en", secondaryDataSource);

        DynamicDataSource dataSource = new DynamicDataSource();
        // 设置所有的数据源
        dataSource.setTargetDataSources(targetDataSource);
        // 设置默认使用的数据源对象
        dataSource.setDefaultTargetDataSource(primaryDataSource);

        return dataSource;
    }


    @Bean(name = "SqlSessionFactory")
    public SqlSessionFactory SqlSessionFactory(@Qualifier("dynamicDataSource") DataSource dynamicDataSource)
            throws Exception {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dynamicDataSource);
        //也是改成pojo路径（pojo实体类）
        sqlSessionFactoryBean.setTypeAliasesPackage("com.3这里要改.pojo");

        MybatisConfiguration configuration = new MybatisConfiguration();
        configuration.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);
        configuration.setJdbcTypeForNull(JdbcType.NULL);
        sqlSessionFactoryBean.setConfiguration(configuration);

        sqlSessionFactoryBean.setTransactionFactory(new SpringManagedTransactionFactory());
        return sqlSessionFactoryBean.getObject();
    }
}

基本的增删改查代码就不贴了。
测试如上方效果图在请求头填入参数即可。
至此项目完成。