多数据源Mongo集群配置

Mongo配置:

mongodb:
    mongo1:
      replica: 集群ip:端口
      userName: 
      password: 
      dbName: 
      maxWaitTime: 10000
      authDB: 
    mongo2:
      replica: 集群ip:端口
      userName: 
      password:
      dbName:
      maxWaitTime: 10000
      authDB: 

MongoProperties :

public static class MongoProperties {

        /**
         * 副本数,host:port 多个,号隔开
         */
        private String replica;

        /**
         * 账户名称
         */
        private String userName;

        /**
         * 密码
         */
        private String password;

        /**
         * 数据库
         */
        private String dbName;

        /**
         * 最大等待时间
         */
        private Integer maxWaitTime;

        /**
         * 验证库
         */
        private String authDB;
    }

集群分割:

/**
     * @param replica 副本
     * @return List<ServerAddress>
     */
    private static List<ServerAddress> getServerAddress(String replica) {
        String[] replicas = replica.split(",");
        List<ServerAddress> serverAddresses = new ArrayList<>();
        for (String s : replicas) {
            String host = s.substring(0, s.lastIndexOf(":"));
            String port = s.substring(s.lastIndexOf(":") + 1);
            serverAddresses.add(new ServerAddress(host, Integer.parseInt(port)));
        }
        return serverAddresses;
    }

配置流程:

MongoTemplate-MongoDatabaseFactoryMongoClient-MongoClientSettings

MongoClient mongoClient = MongoClients.create(MongoClientSettings);

return new SimpleMongoClientDatabaseFactory(mongoClient, mongoProperties.getDbName());

MongoClientSettings

ConnectionPoolSettings poolSetting = ConnectionPoolSettings.builder()
                .maxWaitTime(mongoProperties.getMaxWaitTime(), TimeUnit.MILLISECONDS)
                .build();

存在连接认证的话

MongoCredential credential = MongoCredential.createScramSha1Credential(mongoProperties.getUserName(),
                mongoProperties.getAuthDB(), mongoProperties.getPassword().toCharArray());
MongoClientSettings.builder()
                .credential(credential)
                .applyToConnectionPoolSettings(builder -> builder.applySettings(poolSetting))
                .applyToClusterSettings(builder -> builder.hosts(serverAddresses)).build();
在 Spring Boot 中整合 MongoDB 多数据源,需要按照以下步骤操作: 1. 引入 MongoDB 的依赖 在 pom.xml 文件中引入 MongoDB 的依赖: ```xml <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-mongodb</artifactId> </dependency> ``` 2. 配置数据源 在 application.properties 文件中配置多个数据源: ```properties # 数据源1 spring.data.mongodb.uri=mongodb://localhost:27017/db1 # 数据源2 mongodb2.uri=mongodb://localhost:27017/db2 ``` 3. 配置多数据源 创建多个数据源的配置类,继承自 `AbstractMongoConfiguration`,并重写 `mongoClient()` 方法: ```java @Configuration public class DataSourceConfig1 extends AbstractMongoConfiguration { @Value("${spring.data.mongodb.uri}") private String uri; @Override protected String getDatabaseName() { return "db1"; } @Override public MongoClient mongoClient() { return new MongoClient(new MongoClientURI(uri)); } } @Configuration public class DataSourceConfig2 extends AbstractMongoConfiguration { @Value("${mongodb2.uri}") private String uri; @Override protected String getDatabaseName() { return "db2"; } @Override public MongoClient mongoClient() { return new MongoClient(new MongoClientURI(uri)); } } ``` 4. 配置动态数据源 创建一个动态数据源,实现 `AbstractRoutingDataSource` 接口,重写 `determineCurrentLookupKey()` 方法: ```java public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { return DataSourceContextHolder.getDataSource(); } } ``` 5. 配置数据源上下文 创建一个数据源上下文,用于保存当前使用的数据源的名称: ```java public class DataSourceContextHolder { private static final ThreadLocal<String> contextHolder = new ThreadLocal<>(); public static void setDataSource(String dataSource) { contextHolder.set(dataSource); } public static String getDataSource() { return contextHolder.get(); } public static void clearDataSource() { contextHolder.remove(); } } ``` 6. 配置事务管理器 创建一个事务管理器,用于管理多个数据源的事务: ```java @Configuration @EnableTransactionManagement public class TransactionConfig implements TransactionManagementConfigurer { @Autowired private DynamicDataSource dynamicDataSource; @Override public PlatformTransactionManager annotationDrivenTransactionManager() { return new DataSourceTransactionManager(dynamicDataSource); } } ``` 7. 完成动态数据源配置 在 `Application` 类中完成动态数据源的配置: ```java @SpringBootApplication public class Application { public static void main(String[] args) { SpringApplication.run(Application.class, args); } @Bean public DynamicDataSource dynamicDataSource(DataSourceConfig1 dataSourceConfig1, DataSourceConfig2 dataSourceConfig2) { Map<Object, Object> targetDataSources = new HashMap<>(); targetDataSources.put("db1", dataSourceConfig1.mongoClient()); targetDataSources.put("db2", dataSourceConfig2.mongoClient()); DynamicDataSource dynamicDataSource = new DynamicDataSource(); dynamicDataSource.setDefaultTargetDataSource(dataSourceConfig1.mongoClient()); dynamicDataSource.setTargetDataSources(targetDataSources); return dynamicDataSource; } @Bean public MongoClient mongoClient(DynamicDataSource dynamicDataSource) { return dynamicDataSource; } @Bean public MongoTemplate mongoTemplate(DynamicDataSource dynamicDataSource) { return new MongoTemplate(dynamicDataSource); } } ``` 至此,就完成了 Spring Boot 整合 MongoDB 多数据源动态切换的配置。在需要切换数据源的地方,可以使用 `DataSourceContextHolder.setDataSource("db1")` 进行动态切换。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值