在
Spring Boot 中整合
MongoDB 多
数据源，需要按照以下步骤操作：
1. 引入
MongoDB 的依赖
在 pom.xml 文件中引入
MongoDB 的依赖：
```xml
<dependency>
<groupId>org.
springframework.
boot</groupId>
<artifactId>
spring-
boot-starter-
data-
mongodb</artifactId>
</dependency>
```
2.
配置数据源
在 ap
plication.properties 文件中
配置多个数据源：
```properties
#
数据源1
spring.
data.
mongodb.uri=
mongodb://lo
calhost:27017/db1
#
数据源2
mongodb2.uri=
mongodb://lo
calhost:27017/db2
```
3.
配置多
数据源
创建
多个数据源的
配置类，继承自 `Abstract
MongoConfiguration`，并重写 `
mongoClient()` 方法：
```java
@Configuration
public class
DataSourceConfig1 extends Abstract
MongoConfiguration {
@Value("${
spring.
data.
mongodb.uri}")
private String uri;
@Override
protected String get
DatabaseName() {
return "db1";
}
@Override
public
MongoClient
mongoClient() {
return new
MongoClient(new
MongoClientURI(uri));
}
}
@Configuration
public class
DataSourceConfig2 extends Abstract
MongoConfiguration {
@Value("${
mongodb2.uri}")
private String uri;
@Override
protected String get
DatabaseName() {
return "db2";
}
@Override
public
MongoClient
mongoClient() {
return new
MongoClient(new
MongoClientURI(uri));
}
}
```
4.
配置动态
数据源
创建一个动态
数据源，实现 `AbstractRouting
DataSource` 接口，重写 `determineCurrentLookupKey()` 方法：
```java
public class Dynamic
DataSource extends AbstractRouting
DataSource {
@Override
protected Object determineCurrentLookupKey() {
return
DataSourceContextHolder.get
DataSource();
}
}
```
5.
配置数据源上下文
创建一个
数据源上下文，用于保存当前使用的
数据源的名称：
```java
public class
DataSourceContextHolder {
private static final ThreadLo
cal<String> contextHolder = new ThreadLo
cal<>();
public static void
setDataSource(String
dataSource) {
contextHolder.
set(
dataSource);
}
public static String get
DataSource() {
return contextHolder.get();
}
public static void clear
DataSource() {
contextHolder.remove();
}
}
```
6.
配置事务管理器
创建一个事务管理器，用于管理
多个数据源的事务：
```java
@Configuration
@EnableTransactionManagement
public class TransactionConfig implements TransactionManagementConfigurer {
@Autowired
private Dynamic
DataSource dynamic
DataSource;
@Override
public PlatformTransactionManager annotationDrivenTransactionManager() {
return new
DataSourceTransactionManager(dynamic
DataSource);
}
}
```
7. 完成动态
数据源配置
在 `Ap
plication` 类中完成动态
数据源的
配置：
```java
@
SpringBootAp
plication
public class Ap
plication {
public static void main(String[] args) {
SpringAp
plication.run(Ap
plication.class, args);
}
@Bean
public Dynamic
DataSource dynamic
DataSource(
DataSourceConfig1
dataSourceConfig1,
DataSourceConfig2
dataSourceConfig2) {
Map<Object, Object> target
DataSources = new HashMap<>();
target
DataSources.put("db1",
dataSourceConfig1.
mongoClient());
target
DataSources.put("db2",
dataSourceConfig2.
mongoClient());
Dynamic
DataSource dynamic
DataSource = new Dynamic
DataSource();
dynamic
DataSource.
setDefaultTarget
DataSource(
dataSourceConfig1.
mongoClient());
dynamic
DataSource.
setTarget
DataSources(target
DataSources);
return dynamic
DataSource;
}
@Bean
public
MongoClient
mongoClient(Dynamic
DataSource dynamic
DataSource) {
return dynamic
DataSource;
}
@Bean
public
MongoTemplate
mongoTemplate(Dynamic
DataSource dynamic
DataSource) {
return new
MongoTemplate(dynamic
DataSource);
}
}
```
至此，就完成了
Spring Boot 整合
MongoDB 多
数据源动态切换的
配置。在需要切换
数据源的地方，可以使用 `
DataSourceContextHolder.
setDataSource("db1")` 进行动态切换。
712
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
