MongoTemplate手动动态创建(简版)

1、背景

最近遇到一个需求,就是动态获取springboot中MongoTemplate,有哪些mongo数据库在项目启动的时候不确定,运行过程中有可能会增加,也有可能减少。只能通过一个接口获取所有mongo数据库的地址。但是springboot本身提供的mongTemplate配置都是写死的。所以需要自己实现MongoTemplate的创建。

2、依赖情况

下面是我使用的springboot版本和mongo依赖信息

    
        org.springframework.boot
        spring-boot-starter-parent
        2.7.13
         
    

    
        org.springframework.boot
        spring-boot-starter-data-mongodb
    

3、开发思路

既然需要创建MongoTemplate,那么肯定要看一下MongoTemplate的构造器方法,查看代码如下:

MongoTemplate手动动态创建(简版)_第1张图片

可以看到这里有四种实现方式,第一种主要依赖于MongoClient,后三种主要依赖于MongoDatabaseFactory,第一种方式我没有尝试,不过看了传参,构建起来也是比较简单的。这里我选择的是第二种实现方式,即创建一个MongoDatabaseFactory对象。再看一下该对象的结构信息:

MongoTemplate手动动态创建(简版)_第2张图片

 可以看到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 map = new HashMap<>();

    public static MongoTemplate getMongoTemplate(String host)
    {
        if (!map.containsKey(host)){
//            String url = "mongodb://username:[email protected]: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 names = mongoTemplate.getCollectionNames();
            for (String name:names) System.out.println(name);
        }
        System.out.println(123);
    }

}

4、总结

这里只是演示了动态创建使用的过程,具体使用时还需要加一些逻辑判断和url拼接等流程,另外还可以设置一些客户端连接参数,比如超时时间等,这些配置可以在MongoClient或者其它的构造器中配置。具体的可以按照上面的思路点开对应的构造器查看。

你可能感兴趣的:(spring,boot,mongodb,MongoTemplate)