SpringBoot配置Mybatis多數據源

1. SpringBoot版本

不一樣的版本其配置方式可能有所區別，我採用的是SpringBoot2.0.4.RELEASE。

2. 項目結構

項目結構

3. 多數據源配置

application.yml配置以下：

spring:
 application:
 name: lab-test
 datasource:
 primary:
 jdbc-url: jdbc:mysql://127.0.0.1:3306/laboratory
 driver-class-name: com.mysql.jdbc.Driver
 username: root
 password: root
 secondary:
 jdbc-url: jdbc:mysql://127.0.0.1:3306/test
 driver-class-name: com.mysql.jdbc.Driver
 username: root
 password: root
mybatis:
 type-aliases-package: cn.ntshare.laboratory.domain
 configuration:
 map-underscore-to-camel-case: true
複製代碼

注意：使用jdbc-url屬性代替以前配置中的url屬性

mybatis中的配置中也再也不寫xml文件的位置，這個位置會在後面的具體數據源中進行配置

建立一個Spring配置類，定義兩個DataSource來讀取application.yml中的不一樣配置，在上述yaml中，spring.datasource.primary爲主數據源配置，spring.datasource.secondary爲第二數據源配置，代碼以下：

@Configuration
public class DataSourceConfig {

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

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

}

複製代碼

4. 配置數據源

4.1 增長對第一數據源的配置

注意其中指定數據源對應的mapper文件定義位置和xml文件定義位置，用@Primary來區分主數據源，代碼以下：

@Configuration
@MapperScan(basePackages = "cn.ntshare.laboratory.mapper.master", sqlSessionFactoryRef = "primarySqlSessionFactory")
public class PrimarySourceConfig {

    @Autowired
    @Qualifier("primaryDataSource")
    private DataSource primaryDataSource;

    @Bean(name = "primarySqlSessionFactory")
    @Primary
    public SqlSessionFactory primarySqlSessionFactory() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(primaryDataSource);       // 使用主數據源
        factoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/mapper/master/*.xml"));     // 配置xml文件位置
        return factoryBean.getObject();
    }

    @Bean(name = "primaryTransactionManager")
    @Primary
    public DataSourceTransactionManager primaryTransactionManager() {
        return new DataSourceTransactionManager(primaryDataSource);
    }

    @Bean(name = "primarySqlSessionTemplate")
    @Primary
    public SqlSessionTemplate primarySqlSessionTemplate() throws Exception {
        return new SqlSessionTemplate(primarySqlSessionFactory());
    }

}

複製代碼

4.2 增長第二數據源的配置

第二數據源的配置與第一數據源配置相似，代碼以下：

@Configuration
@MapperScan(basePackages = {"cn.ntshare.laboratory.mapper.slave"}, sqlSessionFactoryRef = "secondarySqlSessionFactory")
public class SecondarySourceConfig {

    @Autowired
    @Qualifier("secondaryDataSource")
    private DataSource secondaryDataSource;

    @Bean(name = "secondarySqlSessionFactory")
    public SqlSessionFactory secondarySqlSessionFactory() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(secondaryDataSource);       // 使用次數據源
        factoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/mapper/slave/*.xml"));
        return factoryBean.getObject();
    }

    @Bean(name = "secondaryTransactionManager")
    public DataSourceTransactionManager secondaryTransactionManager() {
        return new DataSourceTransactionManager(secondaryDataSource);
    }

    @Bean(name = "secondarySqlSessionTemplate")
    public SqlSessionTemplate secondarySqlSessionTemplate() throws Exception {
        return new SqlSessionTemplate(secondarySqlSessionFactory());
    }

}
複製代碼

完成上述配置以後

主數據源Dao層文件的位置爲：cn.ntshare.laboratory.mapper.master 主數據源xml文件的位置爲：classpath:mybatis/mapper/master/

次數據源Dao層文件的位置爲：cn.ntshare.laboratory.mapper.slave 次數據源xml文件的位置爲：classpath:mybatis/mapper/master/

可參照上文中的項目結構

5. 新Mapper文件和xml文件

主數據源下建立MUserMappper.java和MUserMapper.xml

@Repository
@Mapper
public interface MUserMapper {

    List<MUser> selectAll();
}
複製代碼

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="cn.ntshare.laboratory.mapper.master.MUserMapper">
    <select id="selectAll" resultType="cn.ntshare.laboratory.domain.MUser">
        select * from m_user
    </select>
</mapper>
複製代碼

次數據源下建立SUserMappper.java和SUserMapper.xml

@Repository
@Mapper
public interface SUserMapper {

    List<SUser> selectAll();
}

複製代碼

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="cn.ntshare.laboratory.mapper.slave.SUserMapper">
    <select id="selectAll" resultType="cn.ntshare.laboratory.domain.SUser">
        select * from s_user
    </select>
</mapper>
複製代碼

經過這樣的配置，使用MUserMapper就能夠操做主數據庫，使用SUserMapper就能夠操做從數據庫中的數據。

6. 測試

6.1 數據源

數據庫laboratory中有數據表m_user，如圖：

數據庫test中有數據表s_user,如圖：

6.2 單元測試

MUserMapperTest

@SpringBootTest
@RunWith(SpringJUnit4ClassRunner.class)
public class MUserMapperTest {

    @Autowired
    private MUserMapper mUserMapper;

    @Test
    public void testSelectAll() {
        List<MUser> users = mUserMapper.selectAll();
        users.forEach(e -> System.out.println(e.getUsername()));
    }
}
複製代碼

運行結果：

SUserMapperTest

@SpringBootTest
@RunWith(SpringJUnit4ClassRunner.class)
public class SUserMapperTest {

    @Autowired
    private SUserMapper sUserMapper;

    @Test
    public void testSelectAll() {
        List<SUser> users = sUserMapper.selectAll();
        users.forEach(e -> System.out.println(e.getUsername()));
    }
}
複製代碼

運行結果：

多數據源環境搭建完成