SpringBoot+Mybatis多數據源配置和切換

作者：九天銀河聊編程 2023-01-04 09:33:31

數據庫其他數據庫

今天介紹一個SpringBoot+mybatis的多數據源的解決方案。

前言

在項目開發(fā)中，經常會涉及到一個應用程序調用多個數據的情況。今天介紹一個SpringBoot+mybatis的多數據源的解決方案。

數據庫準備

創(chuàng)建兩個數據庫，兩個數據庫都有Im_person表，兩個表中無數據。

代碼結構

說明：我這里只是為了體現(xiàn)效果，就省略了service步驟。各位大牛開發(fā)，不喜勿噴，理解萬歲，嘻嘻！！

application.yml中配置兩個數據源，配置如下：

master:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://127.0.0.1:3306/db1?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&serverTimezone=GMT%2B8
    username: root
    password: 123456

slave:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://127.0.0.1:3306/db2?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&serverTimezone=GMT%2B8
    username: root
    password: 123456

2.SpringBoot多數據源配置類。

@Configuration
@MapperScan(basePackages = "com.zhangls.multipledatasource.dao.master", sqlSessionFactoryRef = "masterSqlSessionFactory")
public class MasterDataSourceConfig {
    @Value("${master.datasource.driver-class-name}")
    private String driverClassName;

    @Value("${master.datasource.url}")
    private String url;

    @Value("${master.datasource.username}")
    private String username;

    @Value("${master.datasource.password}")
    private String password;

    @Bean(name = "masterDataSource")
    @Primary
    public DataSource dataSource() {
        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setDriverClassName(this.driverClassName);
        dataSource.setUrl(this.url);
        dataSource.setUsername(this.username);
        dataSource.setPassword(this.password);
        return dataSource;
    }

    @Bean(name = "masterSqlSessionFactory")
    @Primary
    public SqlSessionFactory sqlSessionFactory(@Qualifier("masterDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
        List<String> mapperLocations = new ArrayList<>();
        mapperLocations.add("classpath*:/mapper/master/*.xml");
        List<Resource> resources = new ArrayList();
        if (mapperLocations != null) {
            for (String mapperLocation : mapperLocations) {
                try {
                    Resource[] mappers = resourceResolver.getResources(mapperLocation);
                    resources.addAll(Arrays.asList(mappers));
                } catch (IOException e) {
                    // ignore
                }
            }
        }

        bean.setMapperLocations(resources.toArray(new Resource[resources.size()]));
        return bean.getObject();
    }

    @Bean(name = "masterTransactionManager")
    @Primary
    public DataSourceTransactionManager transactionManager(@Qualifier("masterDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "masterSqlSessionTemplate")
    @Primary
    public SqlSessionTemplate testSqlSessionTemplate(
            @Qualifier("masterSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

@Configuration
@MapperScan(basePackages = "com.zhangls.multipledatasource.dao.slave", sqlSessionFactoryRef = "slaveSqlSessionFactory")
public class SlaveDataSourceConfig {
    @Value("${slave.datasource.driver-class-name}")
    private String driverClassName;

    @Value("${slave.datasource.url}")
    private String url;

    @Value("${slave.datasource.username}")
    private String username;

    @Value("${slave.datasource.password}")
    private String password;

    @Bean(name = "slaveDataSource")
    public DataSource dataSource() {
        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setDriverClassName(this.driverClassName);
        dataSource.setUrl(this.url);
        dataSource.setUsername(this.username);
        dataSource.setPassword(this.password);
        return dataSource;
    }

    @Bean(name = "slaveSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("slaveDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(
                new PathMatchingResourcePatternResolver().getResources("classpath*:/mapper/slave/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "slaveTransactionManager")
    public DataSourceTransactionManager transactionManager(@Qualifier("slaveDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "slaveSqlSessionTemplate")
    public SqlSessionTemplate testSqlSessionTemplate(
            @Qualifier("masterSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

說明：兩個數據的代碼基本相同，不同的是要配置好掃描包的路徑，以及mybatis的SQL文件的路徑。所以不同的連接DAO和mapper文件要放在不同的文件夾，方便配置管理。

實現(xiàn)效果

以上的配置已經實現(xiàn)了多數據源的配置，下面實現(xiàn)一個功能。通過一個接口同時向兩個庫的兩個表中導入不同的數據。

controller代碼如下：

@RestController
public class DataSourceController {
    @Resource
    private MasterPersonMapper masterPersonMapper;
    @Resource
    private SlavePersonMapper slavePersonMapper;

    @GetMapping("/datasource")
    public String datasource() {
        ImPerson person1 = new ImPerson();
        person1.setPersonId("1");
        person1.setGender("男");
        person1.setBirthday(new Date());
        person1.setLocation("中國");
        masterPersonMapper.insertSelective(person1);

        ImPerson person2 = new ImPerson();
        person2.setPersonId("2");
        person2.setGender("女");
        person2.setBirthday(new Date());
        person2.setLocation("中國北京");
        slavePersonMapper.insertSelective(person2);
        return "導入成功";
    }
}