Spring Boot 集成 MongoDB 多数据源

举报
没毛的刷子 发表于 2023/08/08 09:25:32 2023/08/08
【摘要】 MongoClient 是 MongoDB 官方提供的 Java 驱动程序,我们可以通过创建多个 MongoClient 对象来连接多个 MongoDB 数据源。

 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 对象。

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。