SpringBoot配置两个数据库

2022-08-02

SpringBoot配置两个数据库

yml配置

spring:
  datasource:
    test1:
      driver-class-name: org.postgresql.Driver
      jdbc-url: jdbc:postgresql://127.0.0.1:5432/test  #测试数据库
      username: root
      password: root
 
    test2:
      driver-class-name: oracle.jdbc.driver.OracleDriver
      jdbc-url: jdbc:oracle:thin:@127.0.0.1:8888:orcl  #测试数据库
      username: root
      password: root

特别注意：

（1）使用 test1、test2 区分两个数据库连接

（2）url 改为：jdbc-url

config配置

使用代码进行数据源注入，和扫描 dao 层路径 （以前是在 yml 文件里配置 mybatis 扫描 dao 的路径）

DataSource1Config.java

package com.test.config;
 
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.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
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 org.springframework.jdbc.datasource.DataSourceTransactionManager;
 
import javax.sql.DataSource;
 
@Configuration
@MapperScan(basePackages = "com.test.dao.test1", sqlSessionTemplateRef  = "test1SqlSessionTemplate")
public class DataSource1Config {
 
    @Bean(name = "test1DataSource")
    @ConfigurationProperties(prefix = "spring.datasource.test1")
    @Primary
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }
 
    @Bean(name = "test1SqlSessionFactory")
    @Primary
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:test1/*.xml"));
        return bean.getObject();
    }
 
    @Bean(name = "test1TransactionManager")
    @Primary
    public DataSourceTransactionManager testTransactionManager(@Qualifier("test1DataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }
 
    @Bean(name = "test1SqlSessionTemplate")
    @Primary
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

DataSource2Config.java

package com.test.config;
 
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.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
 
import javax.sql.DataSource;
 
@Configuration
@MapperScan(basePackages = "com.test.dao.test2", sqlSessionTemplateRef  = "test2SqlSessionTemplate")
public class DataSource2Config {
 
    @Bean(name = "test2DataSource")
    @ConfigurationProperties(prefix = "spring.datasource.test2")
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }
 
    @Bean(name = "test2SqlSessionFactory")
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("test2DataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:test2/*.xml"));
        return bean.getObject();
    }
 
    @Bean(name = "test2TransactionManager")
    public DataSourceTransactionManager testTransactionManager(@Qualifier("test2DataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }
 
    @Bean(name = "test2SqlSessionTemplate")
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test2SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

特别注意：

（1）主数据库都有 @Primary 注解，从数据库都没有

（2）第二个数据库作为从数据库

Dao层

在 dao 文件夹下，新建 test1 和 test2 两个包，分别放两个不同数据库的 dao 层文件

TestDao1.java

@Component
public interface TestDao1 {
 
    List<DailyActivityDataMiddle> selectDailyActivity();
 
}

TestDao2.java

@Component
public interface TestDao2 {
 
    List<MovieShowTest> selectDailyActivity();
 
}

XML文件

在 resource 下新建 test1 和 test2 两个文件夹，分别放入对应 dao 层的 xml 文件

TestDao1.xml

<?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="com.test.dao.test1.TestDao1">
 
    <select id="selectDailyActivity" resultType="com.test.pojo.DailyActivityDataMiddle">
 
        SELECT * FROM daily_activity_data_middle
 
    </select>
 
</mapper>

TestDao2.xml

<?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="com.test.dao.test2.TestDao2">
 
    <select id="selectDailyActivity" resultType="com.test.pojo.MovieShowTest">
 
        SELECT * FROM movieshowtest
 
    </select>
 
</mapper>

控制层测试

在 controller 文件里，注入两个数据库的 dao，分别查询数据

@RestController
public class TestController extends BaseController{
 
    @Autowired
    private PropertiesUtils propertiesUtils;
 
    @Autowired
    private TestDao1 testDao1;
 
    @Autowired
    private TestDao2 testDao2;
 
    @RequestMapping(value = {"/test/test1"},method = RequestMethod.POST)
    public Result<JSONObject> DataStatistics (@RequestBody JSONObject body) throws Exception {
        Result<JSONObject> result = new Result<>(ICommon.SUCCESS, propertiesUtils.get(ICommon.SUCCESS));
 
        JSONObject object = new JSONObject();
        object.put("data",testDao1.selectDailyActivity());
        result.setResult(object);
        return result;
    }
 
    @RequestMapping(value = {"/test/test2"},method = RequestMethod.POST)
    public Result<JSONObject> DataStatisticsaa (@RequestBody JSONObject body) throws Exception {
        Result<JSONObject> result = new Result<>(ICommon.SUCCESS, propertiesUtils.get(ICommon.SUCCESS));
 
        JSONObject object = new JSONObject();
        object.put("data",testDao2.selectDailyActivity());
        result.setResult(object);
        return result;
    }
}