Add spring.data.mongodb.protocol property

This update allows configuring the MongoDB connection
protocol using the 'spring.data.mongodb.protocol' property.

See gh-43851

Signed-off-by: Dmytro Nosan <[email protected]>

Author: nosan

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/mongo/MongoProperties.java

@@ -51,6 +51,11 @@ public class MongoProperties {
 	 */
 	public static final String DEFAULT_URI = "mongodb://localhost/test";
 
+	/**
+	 * Protocol to be used for the MongoDB connection.
+	 */
+	private String protocol = "mongodb";
+
 	/**
 	 * Mongo server host. Cannot be set with URI.
 	 */
@@ -117,6 +122,14 @@ public class MongoProperties {
 	 */
 	private Boolean autoIndexCreation;
 
+	public void setProtocol(String protocol) {
+		this.protocol = protocol;
+	}
+
+	public String getProtocol() {
+		return this.protocol;
+	}
+
 	public String getHost() {
 		return this.host;
 	}

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/mongo/PropertiesMongoConnectionDetails.java

@@ -51,11 +51,11 @@ public PropertiesMongoConnectionDetails(MongoProperties properties, SslBundles s
 
 	@Override
 	public ConnectionString getConnectionString() {
-		// mongodb://[username:password@]host1[:port1][,host2[:port2],...[,hostN[:portN]]][/[database.collection][?options]]
+		// [protocol]://[username:password@]host1[:port1][,host2[:port2],...[,hostN[:portN]]][/[database.collection][?options]]
 		if (this.properties.getUri() != null) {
 			return new ConnectionString(this.properties.getUri());
 		}
-		StringBuilder builder = new StringBuilder("mongodb://");
+		StringBuilder builder = new StringBuilder(this.properties.getProtocol()).append("://");
 		if (this.properties.getUsername() != null) {
 			builder.append(encode(this.properties.getUsername()));
 			builder.append(":");

spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/mongo/MongoAutoConfigurationTests.java

@@ -24,6 +24,7 @@
 import com.mongodb.client.MongoClient;
 import com.mongodb.client.MongoClients;
 import com.mongodb.client.internal.MongoClientImpl;
+import com.mongodb.connection.ClusterConnectionMode;
 import com.mongodb.connection.SslSettings;
 import org.junit.jupiter.api.Test;
 
@@ -98,6 +99,22 @@ void configuresSslWithBundle() {
 			});
 	}
 
+	@Test
+	void configuresProtocol() {
+		this.contextRunner.withPropertyValues("spring.data.mongodb.protocol=mongodb+srv").run((context) -> {
+			MongoClientSettings settings = getSettings(context);
+			assertThat(settings.getClusterSettings().getMode()).isEqualTo(ClusterConnectionMode.MULTIPLE);
+		});
+	}
+
+	@Test
+	void defaultProtocol() {
+		this.contextRunner.run((context) -> {
+			MongoClientSettings settings = getSettings(context);
+			assertThat(settings.getClusterSettings().getMode()).isEqualTo(ClusterConnectionMode.SINGLE);
+		});
+	}
+
 	@Test
 	void configuresWithoutSslWhenDisabledWithBundle() {
 		this.contextRunner

spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/mongo/PropertiesMongoConnectionDetailsTests.java

@@ -86,6 +86,13 @@ void databaseHasDefaultWhenNotConfigured() {
 		assertThat(connectionString.getDatabase()).isEqualTo("test");
 	}
 
+	@Test
+	void protocolCanBeConfigured() {
+		this.properties.setProtocol("mongodb+srv");
+		ConnectionString connectionString = this.connectionDetails.getConnectionString();
+		assertThat(connectionString.getConnectionString()).startsWith("mongodb+srv://");
+	}
+
 	@Test
 	void authenticationDatabaseCanBeConfigured() {
 		this.properties.setUsername("user");