SpringBoot MongoDB 多数据源连接池配置

MongoDB 多数据源连接池配置

1.Config配置类

import com.mongodb.ConnectionString;
import com.mongodb.MongoClientSettings;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.mongo.MongoProperties;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.mongodb.MongoDatabaseFactory;
import org.springframework.data.mongodb.SpringDataMongoDB;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoClientDatabaseFactory;
import org.springframework.data.mongodb.core.convert.DefaultMongoTypeMapper;
import org.springframework.data.mongodb.core.convert.MappingMongoConverter;

import java.util.concurrent.TimeUnit;

/**
 * @author: curyu
 * @date: 2022/9/5 13:51
 * @description: mongodb连接池配置
 */
@Configuration
public class MongoConfiguration {

    @Bean
    @Primary
    @ConfigurationProperties(prefix = "spring.data.mongodb.primary")
    public MongoProperties masterMongoProperties() {
        return new MongoProperties();
    }

    @Bean
    @ConfigurationProperties(prefix = "spring.data.mongodb.second")
    public MongoProperties secondMongoProperties() {
        return new MongoProperties();
    }


    /**
     * 连接池自定义配置处理
     *
     * @return
     */
    @Bean
    @Primary
    public MongoDatabaseFactory mongoDatabaseFactory(@Qualifier("masterMongoProperties") MongoProperties properties) {
        MongoClientSettings.Builder builder = MongoClientSettings.builder();
        builder.applyConnectionString(new ConnectionString(properties.getUri()));
        builder.applyToConnectionPoolSettings(b -> {
            // 允许的最大连接数。这些连接在空闲时将保留在池中。一旦池耗尽，任何需要连接的操作都将阻塞等待可用连接 默认: 100
            b.maxSize(100);
            // 最小连接数。这些连接在空闲时将保留在池中，并且池将确保它至少包含这个最小数量 默认: 0
            b.minSize(5);
            // 池连接可以存活的最长时间。零值表示寿命没有限制。超过其生命周期的池连接将被关闭并在必要时由新连接替换
            b.maxConnectionLifeTime(0, TimeUnit.SECONDS);
            // 池连接的最大空闲时间。零值表示对空闲时间没有限制。超过其空闲时间的池连接将被关闭并在必要时由新连接替换
            b.maxConnectionIdleTime(5, TimeUnit.MINUTES);
            // 默认最大连接时间120s;
            b.maxWaitTime(60000, TimeUnit.MILLISECONDS);
        });
        MongoClient mongoClient = MongoClients.create(builder.build(), SpringDataMongoDB.driverInformation());
        return new SimpleMongoClientDatabaseFactory(mongoClient, properties.getDatabase());
    }


    /**
     * 连接池自定义配置处理
     *
     * @return
     */
    @Bean
    public MongoDatabaseFactory secondMongoDatabaseFactory(@Qualifier("secondMongoProperties") MongoProperties properties) {
        MongoClientSettings.Builder builder = MongoClientSettings.builder();
        builder.applyConnectionString(new ConnectionString(properties.getUri()));
        builder.applyToConnectionPoolSettings(b -> {
            // 允许的最大连接数。这些连接在空闲时将保留在池中。一旦池耗尽，任何需要连接的操作都将阻塞等待可用连接 默认: 100
            b.maxSize(100);
            // 最小连接数。这些连接在空闲时将保留在池中，并且池将确保它至少包含这个最小数量 默认: 0
            b.minSize(5);
            // 池连接可以存活的最长时间。零值表示寿命没有限制。超过其生命周期的池连接将被关闭并在必要时由新连接替换
            b.maxConnectionLifeTime(0, TimeUnit.SECONDS);
            // 池连接的最大空闲时间。零值表示对空闲时间没有限制。超过其空闲时间的池连接将被关闭并在必要时由新连接替换
            b.maxConnectionIdleTime(5, TimeUnit.MINUTES);
            // 默认最大连接时间120s;
            b.maxWaitTime(60000, TimeUnit.MILLISECONDS);
        });
        MongoClient mongoClient = MongoClients.create(builder.build(), SpringDataMongoDB.driverInformation());
        return new SimpleMongoClientDatabaseFactory(mongoClient, properties.getDatabase());
    }


    @Bean
    @Primary
    public MongoTemplate mongoTemplate(@Qualifier("mongoDatabaseFactory") MongoDatabaseFactory mongoDatabaseFactory) {
        MongoTemplate mongoTemplate = new MongoTemplate(mongoDatabaseFactory);
        //去除保存实体时，spring data mongodb 自动添加的_class字段
        MappingMongoConverter mongoMapping = (MappingMongoConverter) mongoTemplate.getConverter();
        mongoMapping.setTypeMapper(new DefaultMongoTypeMapper(null));
        mongoMapping.afterPropertiesSet();

        return mongoTemplate;

    }

    @Bean
    public MongoTemplate secondMongoTemplate(@Qualifier("secondMongoDatabaseFactory") MongoDatabaseFactory mongoDatabaseFactory) {
        MongoTemplate mongoTemplate = new MongoTemplate(mongoDatabaseFactory);
        //去除保存实体时，spring data mongodb 自动添加的_class字段
        MappingMongoConverter mongoMapping = (MappingMongoConverter) mongoTemplate.getConverter();
        mongoMapping.setTypeMapper(new DefaultMongoTypeMapper(null));
        mongoMapping.afterPropertiesSet();
        return mongoTemplate;

    }


}

2. 配置文件

spring:
  data:
    mongodb:
      # mongodb 默认数据源
      primary:
        uri: mongodb://root:[email protected]:27017/${spring.data.mongodb.primary.database}?authSource=admin&authMechanism=SCRAM-SHA-1
        database: test-db

      # mongodb 第二数据源
      second:
        uri: mongodb://admin:[email protected]:27017/${spring.data.mongodb.second.database}?authSource=admin&authMechanism=SCRAM-SHA-1
        database: test2-db

3. 数据源测试

import com.mongodb.client.FindIterable;
import lombok.Data;
import org.bson.Document;
import org.junit.jupiter.api.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.test.context.junit4.SpringRunner;

/**
 * @author: curyu
 * @date: 2021/9/4 16:56
 * @description:
 */
@Data
@RunWith(SpringRunner.class)
@SpringBootTest
class MongoUtilTest {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Qualifier("secondMongoTemplate")
    @Autowired
    private MongoTemplate secondMongoTemplate;

    @Test
    void getById() {

        System.out.println(" mongo DataSource 1 ...");
        // 数据源1
        FindIterable cc = mongoTemplate.getCollection("order").find().limit(10);
        for (Document document : cc) {
            System.out.println(document);
        }

        System.out.println(" mongo DataSource 2 ...");
        // 数据源2
        FindIterable cc2 = secondMongoTemplate.getCollection("order2").find().limit(10);
        for (Document document : cc2) {
            System.out.println(document);
        }

    }
}