机票网站建设爱站网站长工具

我想要的效果是，一个类统一管理多数据源，我传个参数进去，它就能返回我对应的mongotemplate

但是根据"mongbodb 多数据源"的关键词，找不到我想要的效果。

网上大多都是明确知道自己是几个数据源，然后每个数据源具体配置，然后业务代码调用的时候，直接注入不同的数据源，这样的话，不太利于以后的数据库扩容。

目录

网上找到的多数据源写法

自用易扩展的多数据源写法

---

比如如下的这种，我在网上找到的写法。

网上找到的多数据源写法

配的时候是这样的

spring:data:mongodb:primary:uri: mongodb://localhost:27017/primarysecondary:uri: mongodb://localhost:27017/secondary

写配置class

@Configuration
public class PrimaryMongoConfig {@Bean@Primary@ConfigurationProperties(prefix = "spring.data.mongodb.primary")public MongoClient primaryMongoClient() {return MongoClients.create();}@Primary@Bean(name = "primaryMongoTemplate")public MongoTemplate primaryMongoTemplate() {return new MongoTemplate(primaryMongoClient(), "primary");}
}@Configuration
public class SecondaryMongoConfig {@Bean@ConfigurationProperties(prefix = "spring.data.mongodb.secondary")public MongoClient secondaryMongoClient() {return MongoClients.create();}@Bean(name = "secondaryMongoTemplate")public MongoTemplate secondaryMongoTemplate() {return new MongoTemplate(secondaryMongoClient(), "secondary");}
}

这样注入了2个Bean

@Service
public class ProductService {@Autowired@Qualifier("secondaryMongoTemplate")private MongoTemplate mongoTemplate;public void saveProduct(Product product) {mongoTemplate.save(product);}
}

通过标明名称的方式，来注入具体的数据源。

这种有个坏处，就是以后如果想扩展数据库，就必须来个类似于thirdMongoTemplate的玩意，然后需要写个switchCase的东西，返回具体的mongoTemplate

所以，我需要自己写个符合我需求的mongoTemplate的聚集类。

自用易扩展的多数据源写法

我的设计是，多数据源以配置文件里，数组的方式进行传入

如下

spring.data.mongodb.sources[0].url=mongodb://localhost:27017
spring.data.mongodb.sources[0].name=mongodb_0
spring.data.mongodb.sources[0].dbName=myTest
spring.data.mongodb.sources[1].url=mongodb://localhost:27017
spring.data.mongodb.sources[1].name=mongodb_1
spring.data.mongodb.sources[1].dbName=myTest_1

我现在暂时设置了2个数据源，url为真实链接，dbName为数据库名称，name为我之后要获取这玩意，设置的key名称。不用拘泥于这些名字，都是我自定义出来的。

对应这些配置，我定义了一个配置类去接。

@Data
@Component
@ConfigurationProperties(prefix = "spring.data.mongodb")
public class MongoConfig {//sources跟配置文件的键名对应上，如果配置文件不叫这个，也跟着换下private List<MongoSource> sources;
}

因为配的是数组对象，也设置下该对象

@Data
public class MongoSource {private String name;private String url;private String dbName;
}

@Data 是lombada插件的

然后构造一下我想要的，管理多数据源的类

public class MongoFactory {// 这里的修饰符自己定吧，按你自己的需求private Map<String, MongoTemplate> dbMap = new HashMap<>();public MongoFactory() {}public void init(MongoConfig config) {// 初始化多数据源的map吧for (MongoSource item : config.getSources()) {MongoClient client = MongoClients.create(item.getUrl());dbMap.put(item.getName(), new MongoTemplate(client, item.getDbName()));}}public MongoTemplate getTemplate(String dbName) {return dbMap.get(dbName);}
}

这里有个很坑的地方，new MongoTemplate(client, item.getDbName())，这里的第2个参数，是指链接里要填的dataBase名称，别乱填了

将这个类进行注册

@Configuration
public class BeanFactoryConfig {//引入配置@Resource private MongoConfig config;//注册一下我自己造的factory@Beanpublic MongoFactory mongoFactory() {MongoFactory factory = new MongoFactory();factory.init(config);return factory;}
}

进行使用，直接进行注入

@Resource private MongoFactory factory;

比如说，判断该数据源是否存在某集合 

boolean flag =factory.getTemplate(name).collectionExists(collectionName);

factory.getTemplate(name) 根据name，可以从map拿到对应的数据源。

以上，我的需求就实现完毕。