Add support for MongoDB AOT Repositories.

Initial Support for generating repository source code at build time.

Closes: #4939

Author: christophstrobl

spring-data-mongodb/pom.xml

@@ -288,6 +288,12 @@
 			<scope>test</scope>
 		</dependency>
 
+		<dependency>
+			<groupId>org.springframework</groupId>
+			<artifactId>spring-core-test</artifactId>
+			<scope>test</scope>
+		</dependency>
+
 		<!-- Kotlin extension -->
 		<dependency>
 			<groupId>org.jetbrains.kotlin</groupId>

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/aot/generated/AotQueryCreator.java

@@ -0,0 +1,199 @@
+/*
+ * Copyright 2025 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.data.mongodb.aot.generated;
+
+import java.util.Iterator;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+import org.bson.conversions.Bson;
+import org.springframework.data.domain.Pageable;
+import org.springframework.data.domain.Range;
+import org.springframework.data.domain.ScrollPosition;
+import org.springframework.data.domain.Sort;
+import org.springframework.data.geo.Distance;
+import org.springframework.data.geo.Point;
+import org.springframework.data.mongodb.core.convert.MongoCustomConversions;
+import org.springframework.data.mongodb.core.convert.MongoWriter;
+import org.springframework.data.mongodb.core.mapping.MongoMappingContext;
+import org.springframework.data.mongodb.core.mapping.MongoPersistentProperty;
+import org.springframework.data.mongodb.core.query.Collation;
+import org.springframework.data.mongodb.core.query.CriteriaDefinition.Placeholder;
+import org.springframework.data.mongodb.core.query.Query;
+import org.springframework.data.mongodb.core.query.TextCriteria;
+import org.springframework.data.mongodb.core.query.UpdateDefinition;
+import org.springframework.data.mongodb.repository.query.ConvertingParameterAccessor;
+import org.springframework.data.mongodb.repository.query.MongoParameterAccessor;
+import org.springframework.data.mongodb.repository.query.MongoQueryCreator;
+import org.springframework.data.repository.query.parser.PartTree;
+import org.springframework.data.util.TypeInformation;
+import org.springframework.lang.Nullable;
+
+import com.mongodb.DBRef;
+
+/**
+ * @author Christoph Strobl
+ * @since 2025/01
+ */
+public class AotQueryCreator {
+
+	private MongoMappingContext mappingContext;
+
+	public AotQueryCreator() {
+
+		MongoMappingContext mongoMappingContext = new MongoMappingContext();
+		mongoMappingContext.setSimpleTypeHolder(
+				MongoCustomConversions.create((cfg) -> cfg.useNativeDriverJavaTimeCodecs()).getSimpleTypeHolder());
+		mongoMappingContext.setAutoIndexCreation(false);
+		mongoMappingContext.afterPropertiesSet();
+
+		this.mappingContext = mongoMappingContext;
+	}
+
+	StringQuery createQuery(PartTree partTree, int parameterCount) {
+
+		Query query = new MongoQueryCreator(partTree,
+				new PlaceholderConvertingParameterAccessor(new PlaceholderParameterAccessor(parameterCount)), mappingContext)
+				.createQuery();
+
+		if(partTree.isLimiting()) {
+			query.limit(partTree.getMaxResults());
+		}
+		return new StringQuery(query);
+	}
+
+	static class PlaceholderConvertingParameterAccessor extends ConvertingParameterAccessor {
+
+		/**
+		 * Creates a new {@link ConvertingParameterAccessor} with the given {@link MongoWriter} and delegate.
+		 *
+		 * @param delegate must not be {@literal null}.
+		 */
+		public PlaceholderConvertingParameterAccessor(PlaceholderParameterAccessor delegate) {
+			super(PlaceholderWriter.INSTANCE, delegate);
+		}
+	}
+
+	enum PlaceholderWriter implements MongoWriter<Object> {
+
+		INSTANCE;
+
+		@Nullable
+		@Override
+		public Object convertToMongoType(@Nullable Object obj, @Nullable TypeInformation<?> typeInformation) {
+			return obj instanceof Placeholder p ? p.getValue() : obj;
+		}
+
+		@Override
+		public DBRef toDBRef(Object object, @Nullable MongoPersistentProperty referringProperty) {
+			return null;
+		}
+
+		@Override
+		public void write(Object source, Bson sink) {
+
+		}
+	}
+
+	static class PlaceholderParameterAccessor implements MongoParameterAccessor {
+
+		private final List<Placeholder> placeholders;
+
+		public PlaceholderParameterAccessor(int parameterCount) {
+			if (parameterCount == 0) {
+				placeholders = List.of();
+			} else {
+				placeholders = IntStream.range(0, parameterCount).mapToObj(it -> new Placeholder("?" + it))
+						.collect(Collectors.toList());
+			}
+		}
+
+		@Override
+		public Range<Distance> getDistanceRange() {
+			return null;
+		}
+
+		@Nullable
+		@Override
+		public Point getGeoNearLocation() {
+			return null;
+		}
+
+		@Nullable
+		@Override
+		public TextCriteria getFullText() {
+			return null;
+		}
+
+		@Nullable
+		@Override
+		public Collation getCollation() {
+			return null;
+		}
+
+		@Override
+		public Object[] getValues() {
+			return placeholders.toArray();
+		}
+
+		@Nullable
+		@Override
+		public UpdateDefinition getUpdate() {
+			return null;
+		}
+
+		@Nullable
+		@Override
+		public ScrollPosition getScrollPosition() {
+			return null;
+		}
+
+		@Override
+		public Pageable getPageable() {
+			return null;
+		}
+
+		@Override
+		public Sort getSort() {
+			return null;
+		}
+
+		@Nullable
+		@Override
+		public Class<?> findDynamicProjection() {
+			return null;
+		}
+
+		@Nullable
+		@Override
+		public Object getBindableValue(int index) {
+			return placeholders.get(index).getValue();
+		}
+
+		@Override
+		public boolean hasBindableNullValue() {
+			return false;
+		}
+
+		@Override
+		@SuppressWarnings({ "unchecked", "rawtypes" })
+		public Iterator<Object> iterator() {
+			return ((List) placeholders).iterator();
+		}
+	}
+
+}