SpringBoot配置多数据源

前言

在日常工作中，难免会遇到一些需要配置多数据源的情况，比如数据迁移，多数据库连接等等，多数据源既可以是同一类型的数据库连接的不同库，也可以是不同类型的数据库连接。下面以MySQL和ClickHouse这两个数据库配置的双数据源进行配置。

一、数据库连接配置

默认的数据源

spring.datasource..type = com.alibaba.druid.pool.DruidDataSource
spring.datasource.url = jdbc:clickhouse://ip:port/database?useSSL=false
spring.datasource.driver-class-name = ru.yandex.clickhouse.ClickHouseDriver
spring.datasource.username = username
spring.datasource.password = password
mybatis.mapper-locations = mapper.xml的位置（例如classpath:mapper/*Mapper.xml）

第二个数据源配置

mysql.datasource.type =com.alibaba.druid.pool.DruidDataSource
mysql.datasource.url=jdbc:mysql://ip:port/database?useSSL=false&useUnicode=true&characterEncoding=utf8&autoReconnect=true&failOverReadOnly=false
mysql.datasource.username=username
mysql.datasource.password=password
mysql.datasource.driver-class-name=com.mysql.jdbc.Driver
mysql.datasource.mapper-locations=classpath:mapper/mysql/*Mapper.xml

这里要注意一个问题，我上面的两个数据源都用datasource.url配置，这是因为我的连接池类型为druid，如果是hikaricp，那么第二个数据源就应该使用datasource.jdbc-url进行配置

二、多数据源配置类

默认数据源配置类

package test.aaa.bbb;

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import javax.sql.DataSource;


@Configuration
@MapperScan(basePackages = "mapper接口的类路径（例如xxx.aaa.bbb.mapper）", sqlSessionTemplateRef  = "sqlSessionTemplate")
//默认数据源配置
public class DefaultSourceConfig {

//这里通过注入DataSourceProperties，去加载默认的数据源配置
    @Autowired
    private DataSourceProperties properties;

    @Value("${mybatis.mapper-locations}")
    private String mapperLocations;

    @Bean
    public DataSource dataSource() {
        return DataSourceBuilder
                .create()
                .type(properties.getType())
                .driverClassName(properties.determineDriverClassName())
                .url(properties.determineUrl())
                .username(properties.determineUsername())
                .password(properties.determinePassword())
                .build();
    }

    @Bean
    public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        Resource[] resource = new PathMatchingResourcePatternResolver().getResources(mapperLocations);
        bean.setMapperLocations(resource);
        return bean.getObject();
    }

    @Bean
    public DataSourceTransactionManager transactionManager(DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean
    public SqlSessionTemplate sqlSessionTemplate(SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

}

第二个数据源配置

package test.xxx;

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import javax.sql.DataSource;

/**
 * mysql数据源配置
 */
@Configuration
@MapperScan(basePackages = "mapper接口的类路径", sqlSessionTemplateRef  = "mysqlSqlSessionTemplate")
public class MysqlDataSourceConfig {

    @Value("$mysql.datasource.mapper-locations}")
    public String mapperLocations;

    @Bean(name = "mysqlDataSource")
    /** application.properteis中对应属性的前缀 */
    @ConfigurationProperties(prefix = "mysql.datasource")
    public DataSource dataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "mysqlSqlSessionFactory")
    public SqlSessionFactory outerSqlSessionFactory(@Qualifier("mysqlDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        Resource[] resource = new PathMatchingResourcePatternResolver().getResources(mapperLocations);
        bean.setMapperLocations(resource);
        return bean.getObject();
    }

    @Bean(name = "mysqlTransactionManager")
    public DataSourceTransactionManager outerTransactionManager(@Qualifier("mysqlDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "mysqlSqlSessionTemplate")
    public SqlSessionTemplate outerSqlSessionTemplate(@Qualifier("mysqlSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }


}

三、启动类修改

因为我们要使用自己定义的数据源配置，所以在启动类上要把自动数据源装配排除，在启动类上加上@SpringBootApplication(exclude = { DataSourceAutoConfiguration.class })，至此，双数据源的配置已经完成。