Spirngboot 2.X 整合mybatis-plus +druid 多数据源的实现

首先加入依赖:

        <!-- AOP-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>

        <!--MySQL 连接驱动依赖-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>

        <!--mybatis-plus数据持久层-->
        <!-- https://mvnrepository.com/artifact/com.baomidou/mybatis-plus -->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus</artifactId>
            <version>3.3.1</version>
        </dependency>
  <!-- https://mvnrepository.com/artifact/org.mybatis.spring.boot/mybatis-spring-boot-starter -->
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>2.1.2</version>
        </dependency>
  <!--druid数据连接池-->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid-spring-boot-starter</artifactId>
            <version>1.1.22</version>
        </dependency>
 <!-- https://mvnrepository.com/artifact/org.springframework/spring-jdbc -->
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-jdbc</artifactId>
            <version>5.2.6.RELEASE</version>
        </dependency>

在配置文件增加下面配置:

spring:
  jackson:
    time-zone: GMT+8
    date-format: yyyy-MM-dd HH:mm:ss
  aop:
    auto: true
    proxy-target-class: true
  datasource:
    druid:
      db1:
        url: jdbc:mysql://ip:端口/db1?useSSL=false&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useUnicode=true&rewriteBatchedStatements=true
        username: ****#mysql账号
        password: ***#mysql密码
        driver-class-name: com.mysql.cj.jdbc.Driver
        initialSize: 5
        minIdle: 5
        maxActive: 20
      db2:
        url: jdbc:mysql://****/db2?useSSL=false&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useUnicode=true&rewriteBatchedStatements=true
        username: ****#mysql账号
        password: ***#mysql密码
        driver-class-name: com.mysql.cj.jdbc.Driver
        initialSize: 5
        minIdle: 5
        maxActive: 20


mybatis:
  type-aliases-package: com.*.*.entity #实体映射类包的位置
  configuration:
    # 驼峰下划线转换
    map-underscore-to-camel-case: true
  mapper-locations: classpath*:/mapper/**/*.xml

#设置druid后台访问白名单
druid:
  config:
    white-list: 127.0.0.1,loclhost

下面配置DruidConfig

import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.List;

@Configuration
public class DruidConfig {

  @Value("#{'${druid.config.white-list}'.split(',')}")
  private List<String> whitelist;
  /**
   * 注册ServletRegistrationBean
   *
   * @return
   */
  @Bean
  public ServletRegistrationBean druidServlet() {
    ServletRegistrationBean reg = new ServletRegistrationBean(new StatViewServlet(), "/druid/*");
    // IP白名单
    for (String ip : whitelist) {
      reg.addInitParameter("allow", ip);
    }
    // IP黑名单(共同存在时，deny优先于allow)
    //        reg.addInitParameter("deny","127.0.0.1");
    // 控制台管理用户
    reg.addInitParameter("loginUsername", "**");//设置登陆账号
    reg.addInitParameter("loginPassword", "***");//设置登陆密码
    // 是否能够重置数据
    reg.addInitParameter("resetEnable", "false");
    return reg;
  }

  /**
   * 注册FilterRegistrationBean
   *
   * @return
   */
  @Bean
  public FilterRegistrationBean filterRegistrationBean() {
    FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean();
    filterRegistrationBean.setFilter(new WebStatFilter());
    filterRegistrationBean.addUrlPatterns("/*");
    filterRegistrationBean.addInitParameter(
        "exclusions", "*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
    filterRegistrationBean.addInitParameter("profileEnable", "true");
    filterRegistrationBean.addInitParameter("principalCookieName", "USER_COOKIE");
    filterRegistrationBean.addInitParameter("principalSessionName", "USER_SESSION");
    filterRegistrationBean.addInitParameter("DruidWebStatFilter", "/*");
    return filterRegistrationBean;
  }
}

定义数据源枚举

public enum DBTYPE {
    db1("db1"),
    db2("db2"),
    db3("db3");

    private String value;

    private DBTYPE(String value) {
        this.value = value;
    }

    public String getValue() {
        return this.value;
    }
}

定义数据源上下文

public class DbContextHolder {
    private static final ThreadLocal contextHolder = new ThreadLocal();

    public DbContextHolder() {
    }

    public static void setDbType(DBTYPE dbTypeEnum) {
        contextHolder.set(dbTypeEnum.getValue());
    }

    public static String getDbType() {
        return (String)contextHolder.get();
    }

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

动态数据源设置

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

/**
 * @program: common
 * @description: 动态数据源
 * @author: linwl
 * @create: 2020-06-20 15:33
 **/
public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    public Object determineCurrentLookupKey()
    {
        return DbContextHolder.getDbType();
    }
}

定义动态数据源切换类

import db.DBTYPE;
import db.DbContextHolder;
import lombok.extern.slf4j.Slf4j;
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.core.annotation.Order;
import org.springframework.stereotype.Component;

@Component
@Order(value = -100)
@Slf4j
@Aspect
public class DataSourceSwitchAspect {
  //数据源1切点
  @Pointcut("execution(* com.*.mapper.db1..*.*(..))")
  public void db1Aspect() {}
  //数据源2切点
  @Pointcut("execution(* com.*.mapper.db2..*.*(..))")
  public void db2Aspect() {}
  //数据源3切点
  @Pointcut("execution(* com.*.mapper.db3..*.*(..))")
  public void db3Aspect() {}

  @Before("db1Aspect()")
  public void getdb1() {
    log.info("切换到db1 数据源...");
    DbContextHolder.setDbType(DBTYPE.db1);
  }

  @Before("db2Aspect()")
  public void getdb2() {
    log.info("切换到db2 数据源...");
    DbContextHolder.setDbType(DBTYPE.db2);
  }

  @Before("db3Aspect()")
  public void getdb3() {
    log.info("切换到db3 数据源...");
    DbContextHolder.setDbType(DBTYPE.db3);
  }

  @After("db1Aspect()")
  public void romvedb1() {
    log.info("移除数据源");
    DbContextHolder.clearDbType();
  }

  @After("db2Aspect()")
  public void romvedb2() {
    log.info("移除数据源");
    DbContextHolder.clearDbType();
  }

  @After("db3Aspect()")
  public void romvedb3() {
    log.info("移除数据源");
    DbContextHolder.clearDbType();
  }
}

前期的配置基本完成了,接下来就是要进入MybatisPlus的配置了
配置如下:

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.MybatisXMLLanguageDriver;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import db.DBTYPE;
import db.DynamicDataSource;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.mapping.DatabaseIdProvider;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.annotation.MapperScan;
import org.mybatis.spring.boot.autoconfigure.MybatisProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.ResourceLoader;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import org.springframework.util.StringUtils;

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

@Configuration
@MapperScan("com.***.mapper")
@EnableTransactionManagement
@Slf4j
public class MybatisPlusConfig {

  @Autowired private MybatisProperties properties;

  private ResourceLoader resourceLoader = new DefaultResourceLoader();

  @Autowired(required = false)
  private DatabaseIdProvider databaseIdProvider;

  /**
   * mybatis-plus分页插件<br>
   * 文档：http://mp.baomidou.com<br>
   */
  @Bean
  public PaginationInterceptor paginationInterceptor() {
    PaginationInterceptor page = new PaginationInterceptor();
    page.setDbType(DbType.MYSQL);
    return new PaginationInterceptor();
  }

  @Bean(name = "db1")
  @ConfigurationProperties(prefix = "spring.datasource.druid.db1")
  public DataSource db1() {
    return DruidDataSourceBuilder.create().build();
  }

  @Bean(name = "db2")
  @ConfigurationProperties(prefix = "spring.datasource.druid.db2")
  public DataSource db2() {
    return DruidDataSourceBuilder.create().build();
  }

  //  @Bean(name = "db3")
  //  @ConfigurationProperties(prefix = "spring.datasource.druid.db3")
  //  public DataSource db3() {
  //    return DruidDataSourceBuilder.create().build();
  //  }

  /**
   * 动态数据源配置
   *
   * @return
   */
  @Bean
  @Primary
  public DataSource multipleDataSource(
      @Qualifier("db1") DataSource db1, @Qualifier("db2") DataSource db2) {
    DynamicDataSource dynamicDataSource = new DynamicDataSource();
    Map<Object, Object> targetDataSources = new HashMap<>(3);
    targetDataSources.put(DBTYPE.db1.getValue(), db1);
    targetDataSources.put(DBTYPE.db2.getValue(), db2);
    dynamicDataSource.setTargetDataSources(targetDataSources);
    dynamicDataSource.setDefaultTargetDataSource(db1);
    return dynamicDataSource;
  }

  @Bean("sqlSessionFactory")
  public SqlSessionFactory sqlSessionFactory() throws Exception {
    try {
      MybatisSqlSessionFactoryBean mybatisPlus = new MybatisSqlSessionFactoryBean();
      mybatisPlus.setDataSource(multipleDataSource(db1(), db2()));
      if (StringUtils.hasText(this.properties.getConfigLocation())) {
        mybatisPlus.setConfigLocation(
            this.resourceLoader.getResource(this.properties.getConfigLocation()));
      }
      mybatisPlus.setPlugins(paginationInterceptor());
      MybatisConfiguration mc = new MybatisConfiguration();
      mc.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);
      mybatisPlus.setConfiguration(mc);
      if (this.databaseIdProvider != null) {
        mybatisPlus.setDatabaseIdProvider(this.databaseIdProvider);
      }
      if (StringUtils.hasLength(this.properties.getTypeAliasesPackage())) {
        mybatisPlus.setTypeAliasesPackage(this.properties.getTypeAliasesPackage());
      }
      if (StringUtils.hasLength(this.properties.getTypeHandlersPackage())) {
        mybatisPlus.setTypeHandlersPackage(this.properties.getTypeHandlersPackage());
      }
      if (!ObjectUtils.isEmpty(this.properties.resolveMapperLocations())) {
        mybatisPlus.setMapperLocations(this.properties.resolveMapperLocations());
      }
      return mybatisPlus.getObject();
    } catch (Exception e) {
      log.error(MessageFormat.format("初始化Mybatis配置异常:{0}!", e.getMessage()));
      throw e;
    }
  }
}

多数据源动态切换就大功告成!