Add Docker Compose service connection support for MongoDB Atlas

Author: eddumelendez

spring-ai-docs/src/main/antora/modules/ROOT/pages/api/docker-compose.adoc

@@ -34,6 +34,9 @@ The following service connection factories are provided in the `spring-ai-spring
 | `ChromaConnectionDetails`
 | Containers named `chromadb/chroma`, `ghcr.io/chroma-core/chroma`
 
+| `MongoConnectionDetails`
+| Containers named `mongodb/mongodb-atlas-local`
+
 | `OllamaConnectionDetails`
 | Containers named `ollama/ollama`
 

spring-ai-spring-boot-docker-compose/pom.xml

@@ -123,6 +123,13 @@
 			<optional>true</optional>
 		</dependency>
 
+		<dependency>
+			<groupId>org.springframework.ai</groupId>
+			<artifactId>spring-ai-mongodb-atlas-store</artifactId>
+			<version>${project.parent.version}</version>
+			<optional>true</optional>
+		</dependency>
+
         <!-- test dependencies -->
 
         <dependency>

spring-ai-spring-boot-docker-compose/src/main/java/org/springframework/ai/docker/compose/service/connection/mongo/MongoDbAtlasLocalDockerComposeConnectionDetailsFactory.java

@@ -0,0 +1,63 @@
+/*
+ * Copyright 2023 - 2024 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
+ *
+ * https://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.ai.docker.compose.service.connection.mongo;
+
+import com.mongodb.ConnectionString;
+import org.springframework.boot.autoconfigure.mongo.MongoConnectionDetails;
+import org.springframework.boot.docker.compose.core.RunningService;
+import org.springframework.boot.docker.compose.service.connection.DockerComposeConnectionDetailsFactory;
+import org.springframework.boot.docker.compose.service.connection.DockerComposeConnectionSource;
+
+/**
+ * @author Eddú Meléndez
+ */
+class MongoDbAtlasLocalDockerComposeConnectionDetailsFactory
+		extends DockerComposeConnectionDetailsFactory<MongoConnectionDetails> {
+
+	private static final int MONGODB_PORT = 27017;
+
+	protected MongoDbAtlasLocalDockerComposeConnectionDetailsFactory() {
+		super("mongodb/mongodb-atlas-local");
+	}
+
+	@Override
+	protected MongoConnectionDetails getDockerComposeConnectionDetails(DockerComposeConnectionSource source) {
+		return new MongoDbAtlasLocalContainerConnectionDetails(source.getRunningService());
+	}
+
+	/**
+	 * {@link MongoConnectionDetails} backed by a {@code MongoDB Atlas}
+	 * {@link RunningService}.
+	 */
+	static class MongoDbAtlasLocalContainerConnectionDetails extends DockerComposeConnectionDetails
+			implements MongoConnectionDetails {
+
+		private final String connectionString;
+
+		MongoDbAtlasLocalContainerConnectionDetails(RunningService service) {
+			super(service);
+			this.connectionString = String.format("mongodb://%s:%d/?directConnection=true", service.host(),
+					service.ports().get(MONGODB_PORT));
+		}
+
+		@Override
+		public ConnectionString getConnectionString() {
+			return new ConnectionString(this.connectionString);
+		}
+
+	}
+
+}

spring-ai-spring-boot-docker-compose/src/main/resources/META-INF/spring.factories

@@ -1,5 +1,6 @@
 org.springframework.boot.autoconfigure.service.connection.ConnectionDetailsFactory=\
 org.springframework.ai.docker.compose.service.connection.chroma.ChromaDockerComposeConnectionDetailsFactory,\
+org.springframework.ai.docker.compose.service.connection.mongo.MongoDbAtlasLocalDockerComposeConnectionDetailsFactory,\
 org.springframework.ai.docker.compose.service.connection.ollama.OllamaDockerComposeConnectionDetailsFactory,\
 org.springframework.ai.docker.compose.service.connection.opensearch.OpenSearchDockerComposeConnectionDetailsFactory,\
 org.springframework.ai.docker.compose.service.connection.qdrant.QdrantDockerComposeConnectionDetailsFactory,\

spring-ai-spring-boot-docker-compose/src/test/java/org/springframework/ai/docker/compose/service/connection/mongo/MongoDbAtlasLocalDockerComposeConnectionDetailsFactoryTests.java

@@ -0,0 +1,22 @@
+package org.springframework.ai.docker.compose.service.connection.mongo;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.autoconfigure.mongo.MongoConnectionDetails;
+import org.springframework.boot.docker.compose.service.connection.test.AbstractDockerComposeIntegrationTests;
+import org.testcontainers.utility.DockerImageName;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class MongoDbAtlasLocalDockerComposeConnectionDetailsFactoryTests extends AbstractDockerComposeIntegrationTests {
+
+	protected MongoDbAtlasLocalDockerComposeConnectionDetailsFactoryTests() {
+		super("mongo-compose.yaml", DockerImageName.parse("mongodb/mongodb-atlas-local"));
+	}
+
+	@Test
+	void runCreatesConnectionDetails() {
+		MongoConnectionDetails connectionDetails = run(MongoConnectionDetails.class);
+		assertThat(connectionDetails.getConnectionString()).isNotNull();
+	}
+
+}

spring-ai-spring-boot-docker-compose/src/test/resources/org/springframework/ai/docker/compose/service/connection/mongo/mongo-compose.yaml

@@ -0,0 +1,5 @@
+services:
+  mongo:
+    image: '{imageName}'
+    ports:
+      - '27017'