Mybatis-Plus配置多数据源

因实验室项目需要，要用mybatisplus从两个不同的数据库中读取数据。

配置数据源

在 application.yml文件中加上多个数据源的配置。默认使用的是 DruidDataSource

spring:
  datasource:
    master:
      driver-class-name: com.mysql.cj.jdbc.Driver
      type: com.alibaba.druid.pool.DruidDataSource
      url: @master_jdbc_url@?useSSL=false&useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&serverTimezone=GMT%2B8
      username: @master_db_username@
      password: @master_db_password@
    ontology:
      driver-class-name: com.mysql.cj.jdbc.Driver
      type: com.alibaba.druid.pool.DruidDataSource
      url: @ontology_jdbc_url@?useSSL=false&useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&serverTimezone=GMT%2B8
      username: @ontology_db_username@
      password: @ontology_db_password@

配置DataSource 和 SessionFactory

针对两个数据库配置两个数据源。用 @ConfigurationProperties(prefix = "spring.datasource.master") 配置数据源的信息。因此，配置文件中的属性必须和 DruidDataSource的set方法相对应。

package com.wx.sleep.configuration;

import com.alibaba.druid.pool.DruidDataSource;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;

/**
 * 配置不同的数据源
 */
@Configuration
public class DataSourceConfig {
    @Bean("masterDatasource")
    @ConfigurationProperties(prefix = "spring.datasource.master")
    public DataSource masterDataSource(){
        return new DruidDataSource();
    }

    @Bean("ontologyDatasource")
    @ConfigurationProperties(prefix = "spring.datasource.ontology")
    public DataSource slaveDataSource(){
        return new DruidDataSource();
    }
}

如下所示，自定义两个Bean，并在首部使用@MapperScan(basePackages = "com.wx.sleep.core.mapper", sqlSessionFactoryRef = "masterSqlSessionFactory")引用它。可以自己执行locationPattern将mapper放在自定义的目录上。因此需要自己准备好mapper和对应的xml文件。然后在这个文件中指定即可。

package com.wx.sleep.configuration;

import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

import javax.sql.DataSource;

@Configuration
@MapperScan(basePackages = "com.wx.sleep.core.mapper", sqlSessionFactoryRef = "masterSqlSessionFactory")
public class MasterMybatisConfig {
    /**
     * 注意，此处需要使用MybatisSqlSessionFactoryBean，不是SqlSessionFactoryBean，
     * 否则，使用mybatis-plus的内置函数时就会报invalid bound statement (not found)异常
     */
    @Bean("masterSqlSessionFactory")
    @Primary
    public SqlSessionFactory masterSqlSessionFactory(@Qualifier("masterDatasource") DataSource dataSource) throws Exception {
        // 设置数据源
        MybatisSqlSessionFactoryBean mybatisSqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
        mybatisSqlSessionFactoryBean.setDataSource(dataSource);
        // mapper的xml文件位置
        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        String locationPattern = "classpath*:/mapper/master/*.xml";
        mybatisSqlSessionFactoryBean.setMapperLocations(resolver.getResources(locationPattern));

        //对应数据库的entity位置
//        String typeAliasesPackage = "me.mason.demo.basicmultidatasource.entity.master";
//        mybatisSqlSessionFactoryBean.setTypeAliasesPackage(typeAliasesPackage);
        return mybatisSqlSessionFactoryBean.getObject();
    }
}

package com.wx.sleep.configuration;

import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

import javax.sql.DataSource;

@Configuration
@MapperScan(basePackages = "com.wx.sleep.neo4j.ontology.mapper", sqlSessionFactoryRef = "ontologySqlSessionFactory")
public class OntologyMybatisConfig {
    /**
     * 注意，此处需要使用MybatisSqlSessionFactoryBean，不是SqlSessionFactoryBean，
     * 否则，使用mybatis-plus的内置函数时就会报invalid bound statement (not found)异常
     */
    @Bean("ontologySqlSessionFactory")
    public SqlSessionFactory masterSqlSessionFactory(@Qualifier("ontologyDatasource") DataSource dataSource) throws Exception {
        // 设置数据源
        MybatisSqlSessionFactoryBean mybatisSqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
        mybatisSqlSessionFactoryBean.setDataSource(dataSource);
        // mapper的xml文件位置
        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        String locationPattern = "classpath*:/mapper/ontology/*.xml";
        mybatisSqlSessionFactoryBean.setMapperLocations(resolver.getResources(locationPattern));

        //对应数据库的entity位置
//        String typeAliasesPackage = "me.mason.demo.basicmultidatasource.entity.master";
//        mybatisSqlSessionFactoryBean.setTypeAliasesPackage(typeAliasesPackage);
        return mybatisSqlSessionFactoryBean.getObject();
    }
}

测试

直接注入对应的mapper就可以使用。

package com.wx.sleep.core.mapper;

import com.wx.sleep.neo4j.ontology.entity.Concept;
import com.wx.sleep.neo4j.ontology.mapper.ConceptMapper;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import javax.annotation.Resource;
import javax.sql.DataSource;
import java.util.List;

@SpringBootTest
public class DynamicTest {
    @Resource(name = "masterDatasource")
    DataSource dataSource;

    @Autowired
    ConceptMapper conceptMapper;

    @Autowired
    EventInfoMapper eventInfoMapper;

    @Test
    public void test(){
//        System.out.println(dataSource);
        conceptMapper.selectList(null).forEach(System.out::println);
        eventInfoMapper.countBUGByStation().forEach(System.out::println);
    }
}