1、添加 MongoDB 相关依赖
在 pom.xml 文件中添加以下依赖:
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
<dependency>
<groupId>org.mongodb</groupId>
<artifactId>mongodb-driver-sync</artifactId>
<version>4.2.3</version>
</dependency>
<dependency>
<groupId>org.mongodb</groupId>
<artifactId>bson</artifactId>
<version>4.2.3</version>
</dependency>
2、在 application.properties 中配置 MongoDB 数据源
在 application.properties 文件中添加以下配置:
# 主数据源
spring.data.mongodb.uri=mongodb://localhost:27017/test
# 第二个数据源
mongodb.second.uri=mongodb://localhost:27018/test
3、创建多个 MongoClient 对象
MongoClient 是 MongoDB 官方提供的 Java 驱动程序,我们可以通过创建多个 MongoClient 对象来连接多个 MongoDB 数据源。
首先,将下面的代码作为配置类,用于创建多个 MongoClient 对象。
@EnableMongoRepositories(
basePackages = "com.example.demo.repository",
mongoTemplateRef = "firstMongoTemplate"
)
@Configuration
public class MultipleMongoConfig {
@Primary
@Bean(name = "firstMongoClient")
public MongoClient firstMongoClient() {
return MongoClients.create("mongodb://localhost:27017");
}
@Bean(name = "secondMongoClient")
public MongoClient secondMongoClient() {
return MongoClients.create("mongodb://localhost:27018");
}
@Primary
@Bean(name = "firstMongoTemplate")
public MongoTemplate firstMongoTemplate(@Qualifier("firstMongoClient") MongoClient mongoClient) {
return new MongoTemplate(mongoClient, "test");
}
@Bean(name = "secondMongoTemplate")
public MongoTemplate secondMongoTemplate(@Qualifier("secondMongoClient") MongoClient mongoClient) {
return new MongoTemplate(mongoClient, "test");
}
}
这个配置类中定义了两个 MongoClient 对象,分别对应了两个 MongoDB 数据源。其中:
@Primary 注解表示该 Bean 是默认的 Bean,如果有多个同类型的 Bean 被注册时,会使用该 Bean;
@Bean 注解用于标注方法,表示将方法返回的实例添加到 Spring Boot 的容器中;
@Qualifier 注解可以指定使用哪个 MongoClient 对象。
4、使用多个 MongoTemplate 对象
在服务实现类中,通过注入 @Qualifier 指定具体使用哪一个 MongoTemplate 来操作数据库。
@Service
public class UserServiceImpl implements UserService {
private final MongoTemplate firstMongoTemplate;
private final MongoTemplate secondMongoTemplate;
public UserServiceImpl(@Qualifier("firstMongoTemplate") MongoTemplate firstMongoTemplate,
@Qualifier("secondMongoTemplate") MongoTemplate secondMongoTemplate) {
this.firstMongoTemplate = firstMongoTemplate;
this.secondMongoTemplate = secondMongoTemplate;
}
@Override
public void saveUser(User user) {
// 在第一个数据源上插入用户数据
firstMongoTemplate.insert(user);
// 在第二个数据源上插入用户数据
secondMongoTemplate.insert(user);
}
}
在上面的示例中,我们使用了 @Qualifier 注解来指定要使用的 MongoTemplate 对象。其中,@Qualifier("firstMongoTemplate") 和 @Qualifier("secondMongoTemplate") 分别对应了配置类中定义的两个 MongoTemplate 对象。