源码分析 – MyBatis Plus 多数据源踩坑

公司准备把报表数据库从MySQL迁到TiDB，先用几个功能试点。于是在系统上需要配置多个数据源。之前在上家公司用MyBatis时配置过4个数据源，MyBatis Plus下应该是大同小异。不出意外的话就要出意外，结果还是踩了坑。

环境

MyBatis Plus版本：3.0.6

配置

项目中使用shardingjdbc管理数据源，这里只贴其中一个配置

@Bean("tidbSqlSessionFactory")
public SqlSessionFactoryBean getTidbSqlSessionFactory(@Autowired ApplicationContext context, @Autowired MetaObjectHandler metaObjectHandler) throws Exception {

    MasterSlaveDataSource dataSource = (MasterSlaveDataSource) context.getBean(DataSource.class);
    DataSource tidbDataSource = dataSource.getConnection().getDataSourceMap().get("ds-tidb");

    SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
    bean.setDataSource(tidbDataSource);

    List<ClassPathResource> resourceList = new ArrayList<>();
    String path = MybatisPlusConfig.class.getResource("/mapper2").getPath();
    FileUtils.scan(path, f -> {
        if (f.getPath().endsWith(".xml")) {
            String xml = f.getPath().substring(f.getPath().lastIndexOf("mapper2"));
            log.info("loading -> {}", xml);
            resourceList.add(new ClassPathResource(xml));
        }
    });
    bean.setMapperLocations(resourceList.toArray(new Resource[resourceList.size()]));

    MybatisConfiguration mybatisConfig = new MybatisConfiguration();
    mybatisConfig.addInterceptor(new PaginationInterceptor());
    mybatisConfig.addInterceptor(new OptimisticLockerInterceptor());
    bean.setConfiguration(mybatisConfig);

    GlobalConfig globalConfig = GlobalConfigUtils.getGlobalConfig(mybatisConfig);
    globalConfig.setSqlInjector(new LogicSqlInjector());
    globalConfig.setMetaObjectHandler(metaObjectHandler);

    return bean;
}

使用注解扫描

@MapperScan(basePackages = "com.ienjoys.reportform.dao2", sqlSessionFactoryRef = "tidbSqlSessionFactory")

也可以创建 MapperScannerConfigurer 配置扫描

遇坑

启动项目正常，点击测试了一些功能也没发现问题。在跑定时任务时抛出错误：

定位错误行，未获取到 sqlSessionFactory 。

查看源码发现，MyBatis Plus 的 IService 接口，save 方法和 saveBatch 方法获取sqlSessionFactory方式不一样。

save方法是获取baseMapper，而批量操作的saveBatch方法是从全局配置 GlobalConfig 里获取的。

在配置多数据源时，每个sqlSession的配置是独立的，不能共用一个，否则会出现部分配置被覆盖。而自己创建 GlobalConfig ，sqlSessionFactory是没有指定的。需要手动设置。

globalConfig.signGlobalConfig(bean.getObject());

重启项目，测试通过。

新版已修复

我在github上查看最新的MyBatis Plus源码，发现该问题在3.4.0之后的版本修复。以下是最新源码：

软件开发是一个持续过程，平时用的依赖库也要不断更新才行。