1、背景
最近遇到一个需求,就是动态获取springboot中MongoTemplate,有哪些mongo数据库在项目启动的时候不确定,运行过程中有可能会增加,也有可能减少。只能通过一个接口获取所有mongo数据库的地址。但是springboot本身提供的mongTemplate配置都是写死的。所以需要自己实现MongoTemplate的创建。
2、依赖情况
下面是我使用的springboot版本和mongo依赖信息
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.7.13</version>
<relativePath/>
</parent>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
3、开发思路
既然需要创建MongoTemplate,那么肯定要看一下MongoTemplate的构造器方法,查看代码如下:
可以看到这里有四种实现方式,第一种主要依赖于MongoClient,后三种主要依赖于MongoDatabaseFactory,第一种方式我没有尝试,不过看了传参,构建起来也是比较简单的。这里我选择的是第二种实现方式,即创建一个MongoDatabaseFactory对象。再看一下该对象的结构信息:
可以看到MongoDatabaseFactory是一个接口类,其有多个实现类,需要关注的有两个,一个是GridFsMongoDatabaseFactory,它就是springboot自动创建mongoTemplate时使用的工厂对象(这里用不上,主要是参考)。另外一个是SimpleMongoClientDatabaseFactory对象,它是一个简单的工厂类,只需要传递连接url就可以。这里为了便于演示,我使用的就是该对象。下面来看下具体的代码:
import org.springframework.data.mongodb.MongoDatabaseFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoClientDatabaseFactory;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
public class MongoClientUtils {
private static Map<String,MongoTemplate> map = new HashMap<>();
public static MongoTemplate getMongoTemplate(String host)
{
if (!map.containsKey(host)){
// String url = "mongodb://username:password@127.0.0.1:27017/dbname";
String url = "mongodb://127.0.0.1:27017/admin";
MongoDatabaseFactory factory = new SimpleMongoClientDatabaseFactory(url);
return new MongoTemplate(factory);
}
return map.get(host);
}
public static void main(String[] args) {
MongoTemplate mongoTemplate = getMongoTemplate("127.0.0.1");
if (mongoTemplate!=null){
Set<String> names = mongoTemplate.getCollectionNames();
for (String name:names) System.out.println(name);
}
System.out.println(123);
}
}
4、总结
这里只是演示了动态创建使用的过程,具体使用时还需要加一些逻辑判断和url拼接等流程,另外还可以设置一些客户端连接参数,比如超时时间等,这些配置可以在MongoClient或者其它的构造器中配置。具体的可以按照上面的思路点开对应的构造器查看。