Spring Boot 集成 MongoDB 多数据源

时间:2023-04-01 18:05:09

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