欢迎您的访问
专注架构,Java,数据结构算法,Python技术分享

SpringBoot集成Mybatis动态多数据源后,MybatisPlus的IPage失效的问题解决方案


    @Value("{spring.datasource.test3.password}")
    private String test3Password;

    @Bean(name="test3DataSource")
    public DataSource test3DataSource() throws Exception{
        HikariDataSource dataSource = new HikariDataSource();
        dataSource.setDriverClassName(test3Driver);
        dataSource.setJdbcUrl(test3Url);
        dataSource.setUsername(test3Username);
        dataSource.setPassword(test3Password);
        return dataSource;
    }
}

定义一个枚举类管理数据源

public enum DatabaseType {

    test1("test1", "test1"),
    test2("test2", "test2"),
    test3("test3","test3");

    private String name;
    private String value;

    DatabaseType(String name, String value){
        this.name = name;
        this.value = value;
    }

    public String getName(){
        return name;
    }

    public String getValue(){
        return value;
    }
}

定义一个线程安全的数据源容器

public class DatabaseContextHolder {

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

    public static void setDatabaseType(DatabaseType type){
        contextHolder.set(type);
    }

    public static DatabaseType getDatabaseType(){
        return contextHolder.get();
    }
}

定义动态数据源

public class DynamicDataSource extends AbstractRoutingDataSource{

    protected Object determineCurrentLookupKey() {
        return DatabaseContextHolder.getDatabaseType();
    }
}

mybatis配置类

网上的很多文章配置出来都会产生数据源循环依赖的问题,这里解决了这个问题。

@Configuration
@MapperScan(basePackages="cn.test.jichi", sqlSessionFactoryRef="sessionFactory")
public class MybatisConfig {

    /**
     *  @Description:设置动态数据源
     */
    @Bean(name="dynamicDataSource")
    @Primary
    public DynamicDataSource DataSource(
            @Qualifier("test1DataSource") DataSource test1DataSource,
            @Qualifier("test2DataSource") DataSource test2DataSource,
            @Qualifier("test3DataSource") DataSource test3DataSource){
        Map<Object, Object> targetDataSource = new HashMap<>();
        targetDataSource.put(DatabaseType.test1, test1DataSource);
        targetDataSource.put(DatabaseType.test2, test2DataSource);
        targetDataSource.put(DatabaseType.test3, test3DataSource);
        DynamicDataSource dataSource = new DynamicDataSource();
        dataSource.setTargetDataSources(targetDataSource);
        dataSource.setDefaultTargetDataSource(test1DataSource);
        return dataSource;
    }

    /**
     *  @Description:根据动态数据源创建sessionFactory
     */
    @Bean(name="sessionFactory")
    public SqlSessionFactory sessionFactory(
            @Qualifier("test1DataSource") DataSource test1DataSource,
            @Qualifier("test2DataSource") DataSource test2DataSource,
            @Qualifier("test3DataSource") DataSource test3DataSource) throws Exception{
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        //构造方法,解决动态数据源循环依赖问题。
        sessionFactoryBean.setDataSource(this.DataSource(test1DataSource,test2DataSource, test3DataSource));
        return sessionFactoryBean.getObject();
    }
}

提供一个示例

    public void testDymnaicDatasource(){
        //不切换数据源默认是自己的。
        System.out.println("-----默认数据源");
        DemoEntity totalCount = demoMapper.getTotalCount();
        String nameCount1 = totalCount.getNameCount();
        String ageCount2 = totalCount.getAgeCount();
        System.out.println("nameCount:"+nameCount1);
        System.out.println("ageCount:"+ageCount2);
        //数据源切换为branch
        System.out.println("-----数据源为test2");
        DynamicDataSourceUtils.chooseBranchDataSource();
        Integer nameCount = demoMapper.getNameCount();
        Integer ageCount = demoMapper.getAgeCount();
        System.out.println("nameCount:"+nameCount);
        System.out.println("ageCount:"+ageCount);
        //数据源为basic
        System.out.println("-----数据源为test3");
        DynamicDataSourceUtils.chooseBasicDataSource();
        Integer ageCount1 = demoMapper.getAgeCount();
        System.out.println("ageCount:"+ageCount1);

    }

总结

至此我们标题探讨的问题就已经解决了,同时给大家提供了动态数据源的解决方案

赞(0) 打赏
版权归原创作者所有,任何形式转载请联系作者;码农code之路 博客站点 » SpringBoot集成Mybatis动态多数据源后,MybatisPlus的IPage失效的问题解决方案

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏