MybatisPlus的IPage失效的問題解決方案

更新時(shí)間：2021年01月19日 14:24:17 作者：經(jīng)典雞翅

這篇文章主要介紹了MybatisPlus的IPage失效的問題解決方案，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

背景

之前做數(shù)據(jù)抽取的時(shí)候，搭了一個(gè)mybatis動(dòng)態(tài)數(shù)據(jù)源切換的架子。方便他們寫抽取的代碼。今天同事問我，架子里面的mybatisplus的IPage失效了是什么問題。想了一下，應(yīng)該是寫動(dòng)態(tài)數(shù)據(jù)源的時(shí)候，我自定義的mybatis的配置覆蓋了已有的配置。于是我讓他先把我寫的配置進(jìn)行刪除，看是否正常。得到回復(fù)，刪除后正常。那么到此問題原因找到，接下來的解決方法，只要在配置中增加分頁器即可。

解決方案

建立一個(gè)分頁器的bean配置

@Bean
public PaginationInterceptor paginationInterceptor() {
  return new PaginationInterceptor();
}

我們隨便自定義一個(gè)類即可，這里主要是將這個(gè)類作為一個(gè)bean交給spring容器管理。

在sqlSessionFactory中注入

  @Bean(name="sessionFactory")
  public SqlSessionFactory sessionFactory(
      @Qualifier("bigDataDataSource") DataSource bigDataDataSource,
      @Qualifier("branchDataSource") DataSource branchDataSource,
      @Qualifier("basicDataSource") DataSource basicDataSource,
      org.apache.ibatis.session.Configuration config) throws Exception{
    SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
    //構(gòu)造方法，解決動(dòng)態(tài)數(shù)據(jù)源循環(huán)依賴問題。
    MybatisConfiguration configuration = new MybatisConfiguration();
    configuration.addInterceptor(new PaginationInterceptor());
    sessionFactoryBean.setConfiguration(configuration);
    sessionFactoryBean.setConfiguration(config);
    sessionFactoryBean.setDataSource(this.DataSource(bigDataDataSource,branchDataSource, basicDataSource));
    return sessionFactoryBean.getObject();
  }

文中總結(jié)

至此，我們的mybatisplus的分頁插件就好使了。下面給大家提供MyBatis多數(shù)據(jù)源的解決方案。

動(dòng)態(tài)數(shù)據(jù)源解決方案

數(shù)據(jù)庫(kù)配置文件

我們項(xiàng)目使用的是yml形式的配置文件，采用的是hikari的數(shù)據(jù)庫(kù)連接池。第一步我們自然是配置多個(gè)數(shù)據(jù)庫(kù)源頭。
我們找到spring的datasource，在下方配置三個(gè)數(shù)據(jù)源。

spring:
 application:
  name: dynamicDatasource
 datasource:
  test1:
   driver-class-name: com.mysql.jdbc.Driver
   url: jdbc:mysql://127.0.0.1:3306/test1?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
   username: root
   password: 123456
  test2:
   driver-class-name: com.mysql.jdbc.Driver
   url: jdbc:mysql://127.0.0.1:3306/test2?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
   username: root
   password: 123456
  test3:
   driver-class-name: com.mysql.jdbc.Driver
   url: jdbc:mysql://127.0.0.1:3306/test3?serverTimezone=GMT%2B8&characterEncoding=utf-8&useSSL=false
   username: root
   password: 123456
  hikari:
   leak-detection-threshold: 2000

定義數(shù)據(jù)源實(shí)體類

我們可以建立個(gè)datasourceBean文件夾專門管理數(shù)據(jù)源的實(shí)體類。
我們這里要建立三個(gè)實(shí)體類。分別對(duì)應(yīng)test1，test2，test3

@Configuration
public class Test1DataSourceBean {

  @Value("${spring.datasource.test1.driver-class-name}")
  private String test1Driver;

  @Value("${spring.datasource.test1.url}")
  private String test1Url;

  @Value("${spring.datasource.test1.username}")
  private String test1Username;

  @Value("${spring.datasource.test1.password}")
  private String test1Password;

  @Bean(name="test1DataSource")
  public DataSource test1DataSource() throws Exception{
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setDriverClassName(test1Driver);
    dataSource.setJdbcUrl(test1Url);
    dataSource.setUsername(test1Username);
    dataSource.setPassword(test1Password);
    return dataSource;
  }
}


@Configuration
public class Test2DataSourceBean {

  @Value("${spring.datasource.test2.driver-class-name}")
  private String test2Driver;

  @Value("${spring.datasource.test2.url}")
  private String test2Url;

  @Value("${spring.datasource.test2.username}")
  private String test2Username;

  @Value("${spring.datasource.test2.password}")
  private String test2Password;

  @Bean(name="test2DataSource")
  public DataSource test2DataSource() throws Exception{
    HikariDataSource dataSource = new HikariDataSource();
    dataSource.setDriverClassName(test2Driver);
    dataSource.setJdbcUrl(test2Url);
    dataSource.setUsername(test2Username);
    dataSource.setPassword(test2Password);
    return dataSource;
  }
}


@Configuration
public class Test3DataSourceBean {

  @Value("${spring.datasource.test3.driver-class-name}")
  private String test3Driver;

  @Value("${spring.datasource.test3.url}")
  private String test3Url;

  @Value("${spring.datasource.test3.username}")
  private String test3Username;

  @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;
  }
}

定義一個(gè)枚舉類管理數(shù)據(jù)源

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;
  }
}

定義一個(gè)線程安全的數(shù)據(jù)源容器

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();
  }
}

定義動(dòng)態(tài)數(shù)據(jù)源

public class DynamicDataSource extends AbstractRoutingDataSource{

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

mybatis配置類

網(wǎng)上的很多文章配置出來都會(huì)產(chǎn)生數(shù)據(jù)源循環(huán)依賴的問題，這里解決了這個(gè)問題。

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

  /**
   * @Description:設(shè)置動(dòng)態(tài)數(shù)據(jù)源
   */
  @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:根據(jù)動(dòng)態(tài)數(shù)據(jù)源創(chuàng)建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();
    //構(gòu)造方法，解決動(dòng)態(tài)數(shù)據(jù)源循環(huán)依賴問題。
    sessionFactoryBean.setDataSource(this.DataSource(test1DataSource,test2DataSource, test3DataSource));
    return sessionFactoryBean.getObject();
  }
}

提供一個(gè)示例

  public void testDymnaicDatasource(){
    //不切換數(shù)據(jù)源默認(rèn)是自己的。
    System.out.println("-----默認(rèn)數(shù)據(jù)源");
    DemoEntity totalCount = demoMapper.getTotalCount();
    String nameCount1 = totalCount.getNameCount();
    String ageCount2 = totalCount.getAgeCount();
    System.out.println("nameCount:"+nameCount1);
    System.out.println("ageCount:"+ageCount2);
    //數(shù)據(jù)源切換為branch
    System.out.println("-----數(shù)據(jù)源為test2");
    DynamicDataSourceUtils.chooseBranchDataSource();
    Integer nameCount = demoMapper.getNameCount();
    Integer ageCount = demoMapper.getAgeCount();
    System.out.println("nameCount:"+nameCount);
    System.out.println("ageCount:"+ageCount);
    //數(shù)據(jù)源為basic
    System.out.println("-----數(shù)據(jù)源為test3");
    DynamicDataSourceUtils.chooseBasicDataSource();
    Integer ageCount1 = demoMapper.getAgeCount();
    System.out.println("ageCount:"+ageCount1);

  }

總結(jié)

至此我們標(biāo)題探討的問題就已經(jīng)解決了，同時(shí)給大家提供了動(dòng)態(tài)數(shù)據(jù)源的解決方案。更多相關(guān)MybatisPlus IPage失效內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: