前言
多数据源的核心就是向 IOC 容器注入 AbstractRoutingDataSource 和如何切换数据源。注入的方式可以是注册 BeanDefinition 或者是构建好的 Bean,切换数据源的方式可以是方法参数或者是注解切换(其他的没想象出来),具体由需求决定。
我的需求是统计多个库的数据,将结果写入另一个数据库,统计的数据库数量是不定的,无法通过 @Bean 直接注入,又是统计任务,DAO 层注解切换无法满足,因此选择注册(AbstractRoutingDataSource 的)BeanDefinition 和方法参数切换来实现。下面以统计统计中日韩用户到结果库为例。
配置文件
master 为结果库,其他为被统计的数据库(china、japan 可以用枚举唯一标识,当然也可以用 String):
dynamic: dataSources: master: driver-class-name: com.mysql.cj.jdbc.Driver <b style="color:transparent">来&源gao@dai!ma.com搞$代^码%网</b> url: jdbc:mysql://localhost:3306/result?useUnicode=true&characterEncoding=utf8xxxxxxxx username: root password: 123456 china: driver-class-name: com.mysql.cj.jdbc.Driver url: jdbc:mysql://localhost:3306/china?useUnicode=true&characterEncoding=utf8xxxxxxxx username: root password: 123456 japan: driver-class-name: com.mysql.cj.jdbc.Driver url: jdbc:mysql://localhost:3306/japan?useUnicode=true&characterEncoding=utf8xxxxxxxx username: root password: 123456 korea: driver-class-name: com.mysql.cj.jdbc.Driver url: jdbc:mysql://localhost:3306/korea?useUnicode=true&characterEncoding=utf8xxxxxxxx username: root password: 123456
对应的配置类:
package com.statistics.dynamicds.core.config; import com.statistics.dynamicds.core.Country; import lombok.Data; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.context.annotation.Configuration; import java.util.Map; import static com.statistics.dynamicds.core.config.MultiDataSourceProperties.PREFIX; @Data @Configuration @ConfigurationProperties(prefix = PREFIX) public class MultiDataSourceProperties { public static final String PREFIX = "dynamic"; private Map<Country, DataSourceProperties> dataSources; @Data public static class DataSourceProperties { private String driverClassName; private String url; private String username; private String password; } } package com.statistics.dynamicds.core; public enum Country { MASTER("master", 0), CHINA("china", 86), JAPAN("japan", 81), KOREA("korea", 82), // 其他国家省略 private final String name; private final int id; Country(String name, int id) { this.name = name; this.id = id; } public int getId() { return id; } public String getName() { return name; } }
依赖
ORM 用的 JPA,SpringBoot 版本为 2.3.7.RELEASE,通过 Lombok 简化 GetSet。
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-jpa</artifactId> </dependency> <dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> <version>1.18.22</version> <scope>provided</scope> </dependency>
构建 AbstractRoutingDataSource
Spring 的动态数据源需要注入 AbstractRoutingDataSource,因为配置文件中被统计数据源不是固定的,所以不能通过 @Bean 注解注入,需要手动构建。
要在启动类加上 @Import(MultiDataSourceImportBeanDefinitionRegistrar.class)。