com.lordofthejars.nosqlunit.mongodb.ReplicationMongoDbConfigurationBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nosqlunit-mongodb Show documentation
There is a newer version: 1.0.0
package com.lordofthejars.nosqlunit.mongodb;

import com.mongodb.MongoClient;
import com.mongodb.ServerAddress;
import com.mongodb.WriteConcern;

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

public class ReplicationMongoDbConfigurationBuilder {

	private static final String DEFAULT_HOST = "localhost";
	private static final int DEFAULT_PORT = 27017;

	private MongoDbConfiguration mongoDbConfiguration = new MongoDbConfiguration();
	private List serverAddresses = new ArrayList();

	private boolean enableSharding = false;

	private WriteConcern writeConcern = WriteConcern.ACKNOWLEDGED;

	private ReplicationMongoDbConfigurationBuilder() {
		super();
	}

	public static ReplicationMongoDbConfigurationBuilder replicationMongoDbConfiguration() {
		return new ReplicationMongoDbConfigurationBuilder();
	}

	public ReplicationMongoDbConfigurationBuilder seed(String host, int port) {
		serverAddresses.add(new ServerAddress(host, port));
		return this;
	}

	public ReplicationMongoDbConfigurationBuilder databaseName(
			String databaseName) {
		mongoDbConfiguration.setDatabaseName(databaseName);
		return this;
	}

	public ReplicationMongoDbConfigurationBuilder username(String username) {
		mongoDbConfiguration.setUsername(username);
		return this;
	}

	public ReplicationMongoDbConfigurationBuilder password(String password) {
		mongoDbConfiguration.setPassword(password);
		return this;
	}

	public ReplicationMongoDbConfigurationBuilder connectionIdentifier(
			String identifier) {
		mongoDbConfiguration.setConnectionIdentifier(identifier);
		return this;
	}

	public ReplicationMongoDbConfigurationBuilder enableSharding() {
		this.enableSharding = true;
		return this;
	}

	public ReplicationMongoDbConfigurationBuilder writeConcern(
			WriteConcern writeConcern) {
		this.writeConcern = writeConcern;
		return this;
	}

	public MongoDbConfiguration configure() {

		if (this.serverAddresses.isEmpty()) {
			addDefaultSeed();
		}

		
		MongoClient mongoClient = new MongoClient(this.serverAddresses);

		if(this.enableSharding) {
			enableSharding(mongoClient);
		}
		
		mongoDbConfiguration.setWriteConcern(writeConcern);
		mongoDbConfiguration.setMongo(mongoClient);

		return mongoDbConfiguration;
	}

	private void enableSharding(MongoClient mongoClient) {

		MongoDbCommands.enableSharding(mongoClient,
				this.mongoDbConfiguration.getDatabaseName());
	}

	private boolean isAuthenticationSet() {
		return this.mongoDbConfiguration.getUsername() != null
				&& this.mongoDbConfiguration.getPassword() != null;
	}

	private void addDefaultSeed() {
		serverAddresses.add(new ServerAddress(DEFAULT_HOST, DEFAULT_PORT));
	}

}