SpringBoot настраивает несколько источников данных Mybatis

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Атрибуты

Расположение xml-файла больше не прописывается в конфигурации в mybatis. Это расположение будет настроено в конкретном источнике данных позже.

Создайте класс конфигурации Spring и определите два источника данных для чтения разных конфигураций в 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 Источники данных

В базе данных Laboratories есть таблица данных m_user, как показано на рисунке:

a.png

В базе данных test есть таблица данных s_user, как показано на рисунке:

b.png

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

результат операции:

c.png

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

результат операции:

d.png

Создана среда с несколькими источниками данных.