关于架构心得-mybatis多个数据源使用start包

前言：

最近新搭建了一个项目，用到了多个数据源，并且需要动态切换，所以写了一个切换数据源的starter包。

spring-boot-starter-multiple-data-source

以读写分离两个数据源举例

一.数据源配置

配置多个数据源

spring:
    datasource:
        master:
          jdbc-url: jdbc:mysql://127.0.0.1:3306/demo1?useUnicode=true&useLegacyDatetimeCode=false&serverTimezone=GMT%2b8&useSSL=false
          username: xxxxxxxxx
          password: xxxxxxxxx
          driver-class-name: com.mysql.cj.jdbc.Driver
        slave:
          jdbc-url: jdbc:mysql://127.0.0.1:3306/demo2?useUnicode=true&useLegacyDatetimeCode=false&serverTimezone=GMT%2b8&useSSL=false
          username: xxxxxx
          password: xxxxxx
          driver-class-name: com.mysql.cj.jdbc.Driver

基础类：

DataSourceType.java 数据源枚举类

public enum DataSourceType {

    MASTER("master"),
    SLAVE("slave");

    private String dataSource;

    DataSourceType(String dataSource){
        this.dataSource = dataSource;
    }

    public String getDataSource() {
        return dataSource;
    }
}

MybatisConfig类：

加载多个数据源，并默认数据源为master

@Configuration
@MapperScan(basePackages = {"**.*Mapper"}) // 扫描DAO
public class MybatisConfig {


    @Bean("master")
    @Primary
    @ConfigurationProperties(prefix = "spring.datasource.master")
    public DataSource master(){
        return DataSourceBuilder.create().build();
    }

    @Bean("slave")
    @ConfigurationProperties(prefix = "spring.datasource.slave")
    public DataSource slave(){
        return DataSourceBuilder.create().build();
    }

    @Bean("dynamicDataSource")
    public DataSource dynamicDataSource() {
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        Map dataSourceMap = new HashMap<>(2);
        dataSourceMap.put(DataSourceType.MASTER.getDataSource(), master());
        dataSourceMap.put(DataSourceType.SLAVE.getDataSource(), slave());
        // 将 master 数据源作为默认指定的数据源
        dynamicDataSource.setDefaultDataSource(master());
        // 将 master 和 slave 数据源作为指定的数据源
        dynamicDataSource.setDataSources(dataSourceMap);
        return dynamicDataSource;
    }

    @Bean
    public SqlSessionFactoryBean sqlSessionFactoryBean() throws Exception {
        SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        // 配置数据源，此处配置为关键配置，如果没有将 dynamicDataSource作为数据源则不能实现切换
        sessionFactory.setDataSource(dynamicDataSource());
        // 扫描Model
        //sessionFactory.setTypeAliasesPackage("com.danke.**.entity");
        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        // 扫描映射文件
        sessionFactory.setMapperLocations(resolver.getResources("**.*Mapper.xml"));
        return sessionFactory;
    }

    @Bean
    public PlatformTransactionManager transactionManager() {
        // 配置事务管理, 使用事务时在方法头部添加@Transactional注解即可
        return new DataSourceTransactionManager(dynamicDataSource());
    }
}

上下文类，用于保证线程安全

public class DynamicDataSourceContextHolder {

    private static final ThreadLocal contextHolder = new ThreadLocal() {
        /**
         * 将 master 数据源的 key作为默认数据源的 key
         */
        @Override
        protected String initialValue() {
            return DataSourceType.MASTER.getDataSource();
        }
    };


    /**
     * 数据源的 key集合，用于切换时判断数据源是否存在
     */
    public static List