ReactiveMongoClientFactory.java

/*
 * Copyright 2012-2017 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.boot.autoconfigure.mongo;

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

import com.mongodb.ConnectionString;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import com.mongodb.async.client.MongoClientSettings;
import com.mongodb.async.client.MongoClientSettings.Builder;
import com.mongodb.connection.ClusterSettings;
import com.mongodb.connection.ConnectionPoolSettings;
import com.mongodb.connection.ServerSettings;
import com.mongodb.connection.SocketSettings;
import com.mongodb.connection.SslSettings;
import com.mongodb.reactivestreams.client.MongoClient;
import com.mongodb.reactivestreams.client.MongoClients;

import org.springframework.core.env.Environment;

/**
 * A factory for a reactive {@link MongoClient} that applies {@link MongoProperties}.
 *
 * @author Mark Paluch
 * @author Stephane Nicoll
 * @since 2.0.0
 */
public class ReactiveMongoClientFactory {

	private final MongoProperties properties;

	private final Environment environment;

	public ReactiveMongoClientFactory(MongoProperties properties,
			Environment environment) {
		this.properties = properties;
		this.environment = environment;
	}

	/**
	 * Creates a {@link MongoClient} using the given {@code settings}. If the environment
	 * contains a {@code local.mongo.port} property, it is used to configure a client to
	 * an embedded MongoDB instance.
	 * @param settings the settings
	 * @return the Mongo client
	 */
	public MongoClient createMongoClient(MongoClientSettings settings) {
		Integer embeddedPort = getEmbeddedPort();
		if (embeddedPort != null) {
			return createEmbeddedMongoClient(settings, embeddedPort);
		}
		return createNetworkMongoClient(settings);
	}

	private Integer getEmbeddedPort() {
		if (this.environment != null) {
			String localPort = this.environment.getProperty("local.mongo.port");
			if (localPort != null) {
				return Integer.valueOf(localPort);
			}
		}
		return null;
	}

	private MongoClient createEmbeddedMongoClient(MongoClientSettings settings,
			int port) {
		Builder builder = builder(settings);
		String host = this.properties.getHost() == null ? "localhost"
				: this.properties.getHost();
		ClusterSettings clusterSettings = ClusterSettings.builder()
				.hosts(Collections.singletonList(new ServerAddress(host, port))).build();
		builder.clusterSettings(clusterSettings);
		return MongoClients.create(builder.build());
	}

	private MongoClient createNetworkMongoClient(MongoClientSettings settings) {
		if (hasCustomAddress() || hasCustomCredentials()) {
			if (this.properties.getUri() != null) {
				throw new IllegalStateException("Invalid mongo configuration, "
						+ "either uri or host/port/credentials must be specified");
			}

			Builder builder = builder(settings);
			if (hasCustomCredentials()) {
				List<MongoCredential> credentials = new ArrayList<>();
				String database = this.properties.getAuthenticationDatabase() == null
						? this.properties.getMongoClientDatabase()
						: this.properties.getAuthenticationDatabase();
				credentials.add(
						MongoCredential.createCredential(this.properties.getUsername(),
								database, this.properties.getPassword()));
				builder.credentialList(credentials);
			}
			String host = this.properties.getHost() == null ? "localhost"
					: this.properties.getHost();
			int port = this.properties.getPort() != null ? this.properties.getPort()
					: MongoProperties.DEFAULT_PORT;
			ClusterSettings clusterSettings = ClusterSettings.builder()
					.hosts(Collections.singletonList(new ServerAddress(host, port)))
					.build();
			builder.clusterSettings(clusterSettings);
			return MongoClients.create(builder.build());
		}
		ConnectionString connectionString = new ConnectionString(
				this.properties.determineUri());
		return MongoClients.create(createBuilder(settings, connectionString).build());
	}

	private Builder createBuilder(MongoClientSettings settings,
			ConnectionString connectionString) {
		Builder builder = builder(settings)
				.clusterSettings(ClusterSettings.builder()
						.applyConnectionString(connectionString).build())
				.connectionPoolSettings(ConnectionPoolSettings.builder()
						.applyConnectionString(connectionString).build())
				.serverSettings(ServerSettings.builder()
						.applyConnectionString(connectionString).build())
				.credentialList(connectionString.getCredentialList())
				.sslSettings(SslSettings.builder().applyConnectionString(connectionString)
						.build())
				.socketSettings(SocketSettings.builder()
						.applyConnectionString(connectionString).build());
		if (connectionString.getReadPreference() != null) {
			builder.readPreference(connectionString.getReadPreference());
		}
		if (connectionString.getReadConcern() != null) {
			builder.readConcern(connectionString.getReadConcern());
		}
		if (connectionString.getWriteConcern() != null) {
			builder.writeConcern(connectionString.getWriteConcern());
		}
		if (connectionString.getApplicationName() != null) {
			builder.applicationName(connectionString.getApplicationName());
		}
		return builder;
	}

	private boolean hasCustomAddress() {
		return this.properties.getHost() != null || this.properties.getPort() != null;
	}

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

	private Builder builder(MongoClientSettings settings) {
		if (settings == null) {
			return MongoClientSettings.builder();
		}

		return MongoClientSettings.builder(settings);
	}

}