多数据源
实现AbstractRoutingDataSource抽象类
新建数据源
在configuration中注入多个数据源
@Bean
public DataSource dataSource1() {
HikariDataSource hikariDataSource = new HikariDataSource();
hikariDataSource.setJdbcUrl("jdbc:mysql://192.168.18.117:3306/study");
hikariDataSource.setUsername("root");
hikariDataSource.setPassword("365373011");
hikariDataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
return hikariDataSource;
}
@Bean
public DataSource dataSource2() {
HikariDataSource hikariDataSource = new HikariDataSource();
hikariDataSource.setJdbcUrl("jdbc:mysql://192.168.18.226:3306/study");
hikariDataSource.setUsername("root");
hikariDataSource.setPassword("52701314");
hikariDataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
return hikariDataSource;
}
实现抽象AbstractRoutingDataSource类
配置数据源key与value之间的对应关系，根据ThreadLocal中的字符串对象来获取数据源进行切换，ThreadLocal等于w时那么就是使用DataSource1，ThreadLocal等于r时，那么就使用Data Source2.
@Configuration
public class DynamicDatasource extends AbstractRoutingDataSource {
public static ThreadLocal<String> name = new ThreadLocal<>();
@Autowired
@Qualifier("dataSource1")
private DataSource dataSource1;
@Autowired
@Qualifier("dataSource2")
private DataSource dataSource2;
@Override
protected Object determineCurrentLookupKey() {
return name.get();
}
@Override
public void afterPropertiesSet() {
Map<Object, Object> dataSourceMap = new HashMap<>();
// 设置数据源key与value之间的对应关系
dataSourceMap.put("W", dataSource1);
dataSourceMap.put("R", dataSource2);
super.setTargetDataSources(dataSourceMap);
// 设置默认数据源
super.setDefaultTargetDataSource(dataSource2);
super.afterPropertiesSet();
}
}
使用DynamicDatasource
将DynamicDatasource注入到SqlSessionFactoryBean中，依靠DynamicDatasource实现myabtis数据源的动态切换。
@Bean
public SqlSessionFactory sqlSessionFactory(@Autowired DynamicDatasource dynamicDatasource) {
SqlSessionFactoryBean sqlSessionFactory = new SqlSessionFactoryBean();
sqlSessionFactory.setDataSource(dynamicDatasource);
try {
return sqlSessionFactory.getObject();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
在controller中进行测试
ThreadLocal等于w时那么就是使用DataSource1，ThreadLocal等于r时，那么就使用Data Source2，这里可以采用AOP + 注解的方式进行优化，利用AOP扫描所有包，判断包下的是否有注解，如果有注解则获取注解中的value，根据value切换ThreadLocal设置ThreadLocal，以此来实现切换数据源。
@RestController
public class TestController {
@Autowired
private UserDAO userDAO;
@GetMapping("/R")
public Object test() {
DynamicDatasource.name.set("W");
return userDAO.list();
}
@GetMapping("/W")
public Object w() {
DynamicDatasource.name.set("R");
return userDAO.list();
}
}
数据库存储内容如下
数据源1:
数据源2:
执行结果如下
W接口:
R接口:
创建多个SqlSessionFactory
根据数据源的不同，创建不同的SqlSessionFactoryBean，根据不同的SqlSessionFactoryBean扫描不同的包，创建不同的mapper，根据数据源的不同导入不同数据源的mapper即可
创建多个SqlSessionFactory
mapperScan中指定了mapper路径和使用的SqlSessionFactoryBean的bean名称。
@Configuration
@MapperScan(basePackages = "com.lyra.dao.w", sqlSessionFactoryRef = "wSqlSessionFactory")
public class WDatasouce {
@Bean
public DataSource dataSource2() {
HikariDataSource hikariDataSource = new HikariDataSource();
hikariDataSource.setJdbcUrl("jdbc:mysql://192.168.18.227:3306/study");
hikariDataSource.setUsername("root");
hikariDataSource.setPassword("52701314");
hikariDataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
return hikariDataSource;
}
@Bean(value = "wSqlSessionFactory")
public SqlSessionFactory wSqlSessionFactory() {
SqlSessionFactoryBean sqlSessionFactory = new SqlSessionFactoryBean();
sqlSessionFactory.setDataSource(dataSource2());
try {
return sqlSessionFactory.getObject();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}
@Configuration
@MapperScan(basePackages = "com.lyra.dao.r", sqlSessionFactoryRef = "rSqlSessionFactory")
public class DataSourceConfig {
@Bean
public DataSource dataSource1() {
HikariDataSource hikariDataSource = new HikariDataSource();
hikariDataSource.setJdbcUrl("jdbc:mysql://192.168.18.117:3306/study");
hikariDataSource.setUsername("root");
hikariDataSource.setPassword("365373011");
hikariDataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
return hikariDataSource;
}
@Bean(value = "rSqlSessionFactory")
public SqlSessionFactory rSqlSessionFactory() {
SqlSessionFactoryBean sqlSessionFactory = new SqlSessionFactoryBean();
sqlSessionFactory.setDataSource(dataSource1());
try {
return sqlSessionFactory.getObject();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}
在Mapper中根据不同的数据源创建不同的mapper
测试
执行结果和之前方法1也是一样的
W接口:
R接口:
dynamic-datasource-spring
文档: https://www.kancloud.cn/tracy5546/dynamic-datasource/2264611
导依赖
<!-- https://mvnrepository.com/artifact/com.baomidou/dynamic-datasource-spring-boot-starter -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>dynamic-datasource-spring-boot-starter</artifactId>
<version>3.6.1</version>
</dependency>
配置数据源
配置两个数据源分别是w和r，和一个primary数据源，如果没有使用ds注解，那么默认使用的就是primary中配置的数据源
spring:
datasource:
dynamic:
primary: w
datasource:
w:
url: jdbc:mysql://192.168.18.117:3306/study
username: root
password: 365373011
driver-class-name: com.mysql.cj.jdbc.Driver
r:
url: jdbc:mysql://192.168.18.227:3306/study
username: root
password: 52701314
driver-class-name: com.mysql.cj.jdbc.Driver
使用DS注解
利用DS注解切换不同的数据源
@RestController
public class TestController {
@Autowired
private UserDAO userDAO;
@Autowired
private WUserDAO wUserDAO;
@GetMapping("/R")
@DS("r")
public Object test() {
return wUserDAO.list();
}
@GetMapping("/W")
@DS("w")
public Object w() {
return userDAO.list();
}
}
测试
执行结果和之前方法1也是一样的
W接口:
R接口: