SpringBoot MongoDB副本集配置

MongoDB配置文件

#mongoDB config
spring.data.mongodb.host=10.0.10.1:3717,10.0.10.2:3717
#采用副本集配置,端口直接配在host后面,不需要单独配置,单独配置也不会读取
#spring.data.mongodb.port=3717
spring.data.mongodb.username=db_ums_app
spring.data.mongodb.password=db_ums_app5402
spring.data.mongodb.database=db_ums_app

MongoDBProperties.java

import org.springframework.boot.context.properties.ConfigurationProperties;

@ConfigurationProperties("spring.data.mongodb")
public class MongoDBProperties {

    private String host;
    private String username;
    private String password;
    private String database;

    //省略getter,setter
    ....
}

MongoDBConfig.java MongoDB配置

import cn.com.flaginfo.platform.umsapp.configuration.config.MongoDBProperties;
import com.mongodb.MongoClient;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

@Component
public class MongoDBConfig {

    private final static Logger logger = LoggerFactory.getLogger(MongoDBConfig.class);

    @Autowired
    private MongoDBProperties mongoDBProperties;

    @Bean
    //覆盖
    @ConditionalOnMissingBean(MongoClient.class)
    public MongoClient getMongodbClients() {
        String host = mongoDBProperties.getHost();
        if(StringUtils.isEmpty(host)){
            throw new RuntimeException("MongoDB host is empty.");
        }
        String[] hosts = host.split(",");
        String[] address;
        List<ServerAddress> addresses = new ArrayList<ServerAddress>();
        for( String h : hosts ){
            if( StringUtils.isEmpty(h) ){
                logger.warn("This MongoDB host [{}] is empty, continue init this host.", h);
                continue;
            }

            if( h.indexOf(":") != -1 ){
                address = h.split(":");
                logger.info("Init MongoDB : host [{}], port [{}]", address[0], address[1]);
                addresses.add(new ServerAddress(address[0], Integer.valueOf(address[1])));
            }else{
                logger.info("Init MongoDB : host [{}], no port.", h);
                addresses.add(new ServerAddress(h));
            }
        }
        MongoCredential credential = MongoCredential.createCredential(mongoDBProperties.getUsername(),
                mongoDBProperties.getDatabase(), mongoDBProperties.getPassword().toCharArray());
        MongoClient client = new MongoClient(addresses, Arrays.asList(credential));
        return client;
    }

}