com.testquack.dal.MongoConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dal Show documentation
There is a newer version: 1.21
package com.testquack.dal;

import com.mongodb.ConnectionString;
import com.mongodb.MongoClientSettings;
import com.mongodb.ServerAddress;
import com.mongodb.ServerApi;
import com.mongodb.ServerApiVersion;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.config.AbstractMongoClientConfiguration;

import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import static com.mongodb.MongoCredential.createCredential;
import static com.mongodb.internal.connection.ServerAddressHelper.createServerAddress;
import static org.apache.commons.lang3.StringUtils.isEmpty;

@Configuration
public class MongoConfig extends AbstractMongoClientConfiguration {

    @Value("${mongo.replicaSet}")
    String replicaSet;

    @Value("${mongo.dbname}")
    String dbname;

    @Value("${mongo.username}")
    String username;

    @Value("${mongo.password}")
    String password;

    @Value("${mongo.uri}")
    String uri;

    @Override
    public MongoClient mongoClient() {
        if(!isEmpty(uri)){
            return getClientByUri();
        }

        List addresses = Stream.of(replicaSet.split(",")).
                map(String::trim).
                map(host-> {
                    String[] tokens = host.split(":");
                    return tokens.length == 2 ?
                            createServerAddress(tokens[0], Integer.parseInt(tokens[1])) :
                            createServerAddress(host);

                }).
                collect(Collectors.toList());

        MongoClientSettings.Builder settingsBuilder = MongoClientSettings.builder()
                .applyToClusterSettings(builder ->
                        builder.hosts(addresses)
                );

        if (!isEmpty(username)){
            settingsBuilder.credential(createCredential(username, dbname, password.toCharArray()));
        }

        return MongoClients.create(settingsBuilder.build());
    }

    private MongoClient getClientByUri(){
        ConnectionString connectionString = new ConnectionString(uri);
        MongoClientSettings settings = MongoClientSettings.builder()
                .applyConnectionString(connectionString)
                .build();
        return MongoClients.create(settings);
    }

    @Override
    protected String getDatabaseName() {
        return dbname;
    }
}