Mongo多数据源操作

项目中遇到Mongo多数据源操作，网上查找资料，做一个总结。项目框架：SpringBoot 

• 配置文件

  #mongdb
  data:
    mongodb:
      primary:
        host: localhost
        port: 27017
        database: jury
      secondary:
        host: localhost
        port: 27017
        database: DM

config文件

MultipleMongoConfig.java

package com.ps.jury.stream.dao.config;

import com.mongodb.MongoClient;
import com.mongodb.MongoClientURI;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.mongo.MongoProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.util.StringUtils;

/**
 * @author yds
 * @version 1.0
 * @description:
 * @date 2020/5/29 0029 下午 17:06
 */
@Configuration
public class MultipleMongoConfig {

    @Autowired
    private MultipleMongoProperties mongoProperties;

    @Primary
    @Bean(name = PrimaryMongoConfig.MONGO_TEMPLATE)
    public MongoTemplate primaryMongoTemplate() {
        return new MongoTemplate(primaryFactory(this.mongoProperties.getPrimary()));
    }

    @Bean
    @Qualifier(SecondaryMongoConfig.MONGO_TEMPLATE)
    public MongoTemplate secondaryMongoTemplate() {
        return new MongoTemplate(secondaryFactory(this.mongoProperties.getSecondary()));
    }

    @Bean
    @Primary
    public MongoDbFactory primaryFactory(MongoProperties mongo) {
        if (StringUtils.isEmpty(mongo.getUsername()) || mongo.getPassword() == null) {
            return new SimpleMongoDbFactory(new MongoClient(mongo.getHost(), mongo.getPort()),
                    mongo.getDatabase());
        } else {
            String password = String.valueOf(mongo.getPassword());
            String url = "mongodb://" + mongo.getUsername()
                    + ":" + password
                    + "@" + mongo.getHost() + ":" + mongo.getPort() + "/" + mongo.getDatabase()
                    + "?authSource=admin&authMechanism=SCRAM-SHA-1";
            return new SimpleMongoDbFactory(new MongoClientURI(url));
        }
    }

    @Bean
    public MongoDbFactory secondaryFactory(MongoProperties mongo) {
        if (StringUtils.isEmpty(mongo.getUsername()) || mongo.getPassword() == null) {
            return new SimpleMongoDbFactory(new MongoClient(mongo.getHost(), mongo.getPort()),
                    mongo.getDatabase());
        } else {
            String password = String.valueOf(mongo.getPassword());
            String url = "mongodb://" + mongo.getUsername()
                    + ":" + password
                    + "@" + mongo.getHost() + ":" + mongo.getPort() + "/" + mongo.getDatabase()
                    + "?authSource=admin&authMechanism=SCRAM-SHA-1";
            return new SimpleMongoDbFactory(new MongoClientURI(url));
        }
    }

}

MultipleMongoProperties.java

package com.ps.jury.stream.dao.config;

import lombok.Data;
import org.springframework.boot.autoconfigure.mongo.MongoProperties;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

/**
 * @author yds
 * @version 1.0
 * @description:
 * @date 2020/5/29 0029 下午 16:55
 */
@Data
@Component
@ConfigurationProperties(prefix = "spring.data.mongodb")
public class MultipleMongoProperties {
    private MongoProperties primary = new MongoProperties();

    private MongoProperties secondary = new MongoProperties();

}

PrimaryMongoConfig.java

package com.ps.jury.stream.dao.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;

/**
 * @author yds
 * @version 1.0
 * @description:
 * @date 2020/5/29 0029 下午 17:04
 */
@Configuration
@EnableMongoRepositories(mongoTemplateRef = PrimaryMongoConfig.MONGO_TEMPLATE)
public class PrimaryMongoConfig {
    public static final String MONGO_TEMPLATE = "primaryMongoTemplate";
}

SecondaryMongoConfig.java

package com.ps.jury.stream.dao.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;

/**
 * @author yds
 * @version 1.0
 * @description:
 * @date 2020/5/29 0029 下午 17:05
 */

@Configuration
@EnableMongoRepositories(mongoTemplateRef = SecondaryMongoConfig.MONGO_TEMPLATE)
public class SecondaryMongoConfig {
    public static final String MONGO_TEMPLATE = "secondaryMongoTemplate";

}

使用：

说明：

@Autowired 注入：默认第一数据源， @Resource：指定数据源

package com.ps.jury.stream.dao.mongodao;

import com.ps.jury.stream.common.constant.MongoConstant;
import com.ps.jury.stream.dao.config.SecondaryMongoConfig;
import com.ps.jury.stream.dao.dto.var.IndBlackList;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

/**
 * @author yds
 * @version 1.0
 * @description:
 * @date 2020/5/29 0029 下午 17:41
 */
@Service
public class VarseelctImpl<T> implements VarSelect<T> {

    /**
     * MD库
     */
    @Resource(name = SecondaryMongoConfig.MONGO_TEMPLATE)
    private MongoTemplate mongoTemplate;


    @Override
    public IndBlackList selectBk(String category, String level, String value) {
        List<IndBlackList> indBlackLists = mongoTemplate.find(
                new Query(Criteria
                        .where("blacklist_category").is(category)
                        .and("blacklist_level").is(level)
                        .and("blacklist_value").is(value)), IndBlackList.class, MongoConstant.IND_BLACKLIST);
        if (indBlackLists.size() > 0) {
            return indBlackLists.get(0);
        }
        return null;
    }

    @Override
    public List<IndBlackList> getAllbk() {
        List<IndBlackList> all = mongoTemplate.findAll(IndBlackList.class, "ind_blacklist");
        return all;
    }
}