Spring Boot 多数据源切换

前面我们已经介绍了三种方式来操作数据库，在实际开发中，往往会出现一个服务连接多个数据库的需求，这时候就需要在项目中进行灵活切换数据源来完成多个数据库操作。这一章中，我们使用jdbcTemplate来学习多数据源的配置。

一、准备工作

1.1 建库、建表

我们新建两个库db1和db2，数据结构还是用前面演示的，分别在两个库中新建表student。

CREATE TABLE `student` (
   `student_id` int(30) NOT NULL AUTO_INCREMENT,
   `age` int(1) DEFAULT NULL COMMENT '年龄',
   `name` varchar(45) DEFAULT NULL COMMENT '姓名',
   `sex` int(1) DEFAULT NULL COMMENT '性别：1：男，2：女，0：未知',
   `create_time` datetime DEFAULT NULL COMMENT '创建时间',
   `status` int(1) DEFAULT NULL COMMENT '状态：1：正常，-1：删除',
   PRIMARY KEY (`student_id`)
 ) ENGINE=InnoDB CHARSET=utf8mb4 CHECKSUM=1 DELAY_KEY_WRITE=1 ROW_FORMAT=DYNAMIC COMMENT='学生表'

1.2 引入mysql和jdbcTemplate依赖

<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.11</version>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>

1.3 写入两个数据源配置

spring:
  datasource:
    db1:
      driver-class-name: com.mysql.cj.jdbc.Driver
      jdbc-url: jdbc:mysql://localhost:3306/db1
      username: root
      password: root
    db2:
      driver-class-name: com.mysql.cj.jdbc.Driver
      jdbc-url: jdbc:mysql://localhost:3306/db2
      username: root
      password: root

二、多数据源配置

@Configuration
public class DataSourceConfig {

    @Primary
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.db1")
    public DataSource db1DataSource(){
        return DataSourceBuilder.create().build();
    }

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

• @Primary：表示主的，即出现多个bean的时候如果不指定具体的bean，则会采用这个
• @bean：标注为一个bean，如果不指定name属性，则会使用创建bean方法的名字做为bean的名字
• @ConfigurationProperties：读取配置文件

三、配置JdbcTemplate对象

@Configuration
public class DataSourceConfig {

    @Bean
    public JdbcTemplate db1JdbcTemplate(@Qualifier("db1DataSource") DataSource dataSource){
        return new JdbcTemplate(dataSource);
    }
    @Primary
    @Bean
    public JdbcTemplate db2JdbcTemplate(@Qualifier("db2DataSource") DataSource dataSource){
        return new JdbcTemplate(dataSource);
    }
}

• @Qualifier：bean类型相同后，指定使用的bean的name

四、测试类

4.1 测试@Primary属性

不指定使用哪个JdbcTemplate对象时，会使用标注了@Primary属性的对象

@SpringBootTest
class SpringBootDatasourceApplicationTests {

    @Autowired
    private JdbcTemplate jdbcTemplate;

    @Test
    void testPrimary() {
        jdbcTemplate.update("insert into student(name,age) values(?,?)",new Object[]{"Java旅途",18});
    }
}

4.2 测试多数据源

@SpringBootTest
class SpringBootDatasourceApplicationTests {


    @Autowired
    private JdbcTemplate db1JdbcTemplate;
    @Autowired
    private JdbcTemplate db2JdbcTemplate;

    @Autowired
    private JdbcTemplate jdbcTemplate;
    @Test
    void contextLoads() {

        db1JdbcTemplate.update("insert into student(name,age) values(?,?)",new Object[]{"Java旅途",18});
        db2JdbcTemplate.update("insert into student(name,age) values(?,?)",new Object[]{"Java旅途",18});
    }
    }
}

这里分享一道面试题：@Autowired 与@Resource有什么区别？

@Autowired是Spring提供的，@Resource是JDK提供的；

@Autowired是根据bean的类型匹配的，@Resource是根据bean的name匹配的；

如果@Autowird想要根据name匹配应该怎么做呢？

1. 配合@Qualifier注解指定bean的name
2. 使用变量名称作为bean的id，@Autowired如果匹配到多个符合条件的对象后，会自动根据变量名称做为bean的id继续匹配。我们在4.2中采用的就是这种方式。

代码示例

本文的相关代码已上传至github：

• Github：https://github.com/binzh303/spring-boot-route/

如果您觉得本文不错，欢迎 Star 支持！