Christoph Strobl
7 months ago
committed by
Mark Paluch
34 changed files with 2185 additions and 819 deletions
@ -0,0 +1,360 @@
@@ -0,0 +1,360 @@
|
||||
/* |
||||
* 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.repository.aot; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
import java.util.stream.Collectors; |
||||
import java.util.stream.Stream; |
||||
|
||||
import org.bson.Document; |
||||
import org.jspecify.annotations.NullUnmarked; |
||||
import org.springframework.core.annotation.MergedAnnotation; |
||||
import org.springframework.data.domain.SliceImpl; |
||||
import org.springframework.data.domain.Sort.Order; |
||||
import org.springframework.data.mongodb.core.MongoOperations; |
||||
import org.springframework.data.mongodb.core.aggregation.Aggregation; |
||||
import org.springframework.data.mongodb.core.aggregation.AggregationOptions; |
||||
import org.springframework.data.mongodb.core.aggregation.AggregationPipeline; |
||||
import org.springframework.data.mongodb.core.aggregation.AggregationResults; |
||||
import org.springframework.data.mongodb.core.aggregation.TypedAggregation; |
||||
import org.springframework.data.mongodb.core.mapping.MongoSimpleTypes; |
||||
import org.springframework.data.mongodb.core.query.Collation; |
||||
import org.springframework.data.mongodb.repository.Hint; |
||||
import org.springframework.data.mongodb.repository.ReadPreference; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryMethod; |
||||
import org.springframework.data.repository.aot.generate.AotQueryMethodGenerationContext; |
||||
import org.springframework.data.util.ReflectionUtils; |
||||
import org.springframework.javapoet.CodeBlock; |
||||
import org.springframework.javapoet.CodeBlock.Builder; |
||||
import org.springframework.util.ClassUtils; |
||||
import org.springframework.util.CollectionUtils; |
||||
import org.springframework.util.StringUtils; |
||||
|
||||
/** |
||||
* @author Christoph Strobl |
||||
* @since 5.0 |
||||
*/ |
||||
class AggregationBlocks { |
||||
|
||||
@NullUnmarked |
||||
static class AggregationExecutionCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
private String aggregationVariableName; |
||||
|
||||
AggregationExecutionCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
AggregationExecutionCodeBlockBuilder referencing(String aggregationVariableName) { |
||||
|
||||
this.aggregationVariableName = aggregationVariableName; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
String mongoOpsRef = context.fieldNameOf(MongoOperations.class); |
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.add("\n"); |
||||
|
||||
Class<?> outputType = queryMethod.getReturnedObjectType(); |
||||
if (MongoSimpleTypes.HOLDER.isSimpleType(outputType)) { |
||||
outputType = Document.class; |
||||
} else if (ClassUtils.isAssignable(AggregationResults.class, outputType)) { |
||||
outputType = queryMethod.getReturnType().getComponentType().getType(); |
||||
} |
||||
|
||||
if (ReflectionUtils.isVoid(queryMethod.getReturnedObjectType())) { |
||||
builder.addStatement("$L.aggregate($L, $T.class)", mongoOpsRef, aggregationVariableName, outputType); |
||||
return builder.build(); |
||||
} |
||||
|
||||
if (ClassUtils.isAssignable(AggregationResults.class, context.getMethod().getReturnType())) { |
||||
builder.addStatement("return $L.aggregate($L, $T.class)", mongoOpsRef, aggregationVariableName, outputType); |
||||
return builder.build(); |
||||
} |
||||
|
||||
if (outputType == Document.class) { |
||||
|
||||
Class<?> returnType = ClassUtils.resolvePrimitiveIfNecessary(queryMethod.getReturnedObjectType()); |
||||
|
||||
if (queryMethod.isStreamQuery()) { |
||||
|
||||
builder.addStatement("$T<$T> $L = $L.aggregateStream($L, $T.class)", Stream.class, Document.class, |
||||
context.localVariable("results"), mongoOpsRef, aggregationVariableName, outputType); |
||||
|
||||
builder.addStatement("return $1L.map(it -> ($2T) convertSimpleRawResult($2T.class, it))", |
||||
context.localVariable("results"), returnType); |
||||
} else { |
||||
|
||||
builder.addStatement("$T $L = $L.aggregate($L, $T.class)", AggregationResults.class, |
||||
context.localVariable("results"), mongoOpsRef, aggregationVariableName, outputType); |
||||
|
||||
if (!queryMethod.isCollectionQuery()) { |
||||
builder.addStatement( |
||||
"return $1T.<$2T>firstElement(convertSimpleRawResults($2T.class, $3L.getMappedResults()))", |
||||
CollectionUtils.class, returnType, context.localVariable("results")); |
||||
} else { |
||||
builder.addStatement("return convertSimpleRawResults($T.class, $L.getMappedResults())", returnType, |
||||
context.localVariable("results")); |
||||
} |
||||
} |
||||
} else { |
||||
if (queryMethod.isSliceQuery()) { |
||||
builder.addStatement("$T $L = $L.aggregate($L, $T.class)", AggregationResults.class, |
||||
context.localVariable("results"), mongoOpsRef, aggregationVariableName, outputType); |
||||
builder.addStatement("boolean $L = $L.getMappedResults().size() > $L.getPageSize()", |
||||
context.localVariable("hasNext"), context.localVariable("results"), context.getPageableParameterName()); |
||||
builder.addStatement( |
||||
"return new $1T<>($2L ? $3L.getMappedResults().subList(0, $4L.getPageSize()) : $3L.getMappedResults(), $4L, $2L)", |
||||
SliceImpl.class, context.localVariable("hasNext"), context.localVariable("results"), |
||||
context.getPageableParameterName()); |
||||
} else { |
||||
|
||||
if (queryMethod.isStreamQuery()) { |
||||
builder.addStatement("return $L.aggregateStream($L, $T.class)", mongoOpsRef, aggregationVariableName, |
||||
outputType); |
||||
} else { |
||||
|
||||
builder.addStatement("return $L.aggregate($L, $T.class).getMappedResults()", mongoOpsRef, |
||||
aggregationVariableName, outputType); |
||||
} |
||||
} |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
} |
||||
|
||||
@NullUnmarked |
||||
static class AggregationCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
private final List<CodeBlock> arguments; |
||||
|
||||
private AggregationInteraction source; |
||||
|
||||
private String aggregationVariableName; |
||||
private boolean pipelineOnly; |
||||
|
||||
AggregationCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.arguments = context.getBindableParameterNames().stream().map(CodeBlock::of).collect(Collectors.toList()); |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
AggregationCodeBlockBuilder stages(AggregationInteraction aggregation) { |
||||
|
||||
this.source = aggregation; |
||||
return this; |
||||
} |
||||
|
||||
AggregationCodeBlockBuilder usingAggregationVariableName(String aggregationVariableName) { |
||||
|
||||
this.aggregationVariableName = aggregationVariableName; |
||||
return this; |
||||
} |
||||
|
||||
AggregationCodeBlockBuilder pipelineOnly(boolean pipelineOnly) { |
||||
|
||||
this.pipelineOnly = pipelineOnly; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
builder.add("\n"); |
||||
|
||||
String pipelineName = context.localVariable(aggregationVariableName + (pipelineOnly ? "" : "Pipeline")); |
||||
builder.add(pipeline(pipelineName)); |
||||
|
||||
if (!pipelineOnly) { |
||||
|
||||
builder.addStatement("$1T<$2T> $3L = $4T.newAggregation($2T.class, $5L.getOperations())", |
||||
TypedAggregation.class, context.getRepositoryInformation().getDomainType(), aggregationVariableName, |
||||
Aggregation.class, pipelineName); |
||||
|
||||
builder.add(aggregationOptions(aggregationVariableName)); |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock pipeline(String pipelineVariableName) { |
||||
|
||||
String sortParameter = context.getSortParameterName(); |
||||
String limitParameter = context.getLimitParameterName(); |
||||
String pageableParameter = context.getPageableParameterName(); |
||||
|
||||
boolean mightBeSorted = StringUtils.hasText(sortParameter); |
||||
boolean mightBeLimited = StringUtils.hasText(limitParameter); |
||||
boolean mightBePaged = StringUtils.hasText(pageableParameter); |
||||
|
||||
int stageCount = source.stages().size(); |
||||
if (mightBeSorted) { |
||||
stageCount++; |
||||
} |
||||
if (mightBeLimited) { |
||||
stageCount++; |
||||
} |
||||
if (mightBePaged) { |
||||
stageCount += 3; |
||||
} |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
builder.add(aggregationStages(context.localVariable("stages"), source.stages(), stageCount, arguments)); |
||||
|
||||
if (mightBeSorted) { |
||||
builder.add(sortingStage(sortParameter)); |
||||
} |
||||
|
||||
if (mightBeLimited) { |
||||
builder.add(limitingStage(limitParameter)); |
||||
} |
||||
|
||||
if (mightBePaged) { |
||||
builder.add(pagingStage(pageableParameter, queryMethod.isSliceQuery())); |
||||
} |
||||
|
||||
builder.addStatement("$T $L = createPipeline($L)", AggregationPipeline.class, pipelineVariableName, |
||||
context.localVariable("stages")); |
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock aggregationOptions(String aggregationVariableName) { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
List<CodeBlock> options = new ArrayList<>(5); |
||||
if (ReflectionUtils.isVoid(queryMethod.getReturnedObjectType())) { |
||||
options.add(CodeBlock.of(".skipOutput()")); |
||||
} |
||||
|
||||
MergedAnnotation<Hint> hintAnnotation = context.getAnnotation(Hint.class); |
||||
String hint = hintAnnotation.isPresent() ? hintAnnotation.getString("value") : null; |
||||
if (StringUtils.hasText(hint)) { |
||||
options.add(CodeBlock.of(".hint($S)", hint)); |
||||
} |
||||
|
||||
MergedAnnotation<ReadPreference> readPreferenceAnnotation = context.getAnnotation(ReadPreference.class); |
||||
String readPreference = readPreferenceAnnotation.isPresent() ? readPreferenceAnnotation.getString("value") : null; |
||||
if (StringUtils.hasText(readPreference)) { |
||||
options.add(CodeBlock.of(".readPreference($T.valueOf($S))", com.mongodb.ReadPreference.class, readPreference)); |
||||
} |
||||
|
||||
if (queryMethod.hasAnnotatedCollation()) { |
||||
options.add(CodeBlock.of(".collation($T.parse($S))", Collation.class, queryMethod.getAnnotatedCollation())); |
||||
} |
||||
|
||||
if (!options.isEmpty()) { |
||||
|
||||
Builder optionsBuilder = CodeBlock.builder(); |
||||
optionsBuilder.add("$1T $2L = $1T.builder()\n", AggregationOptions.class, |
||||
context.localVariable("aggregationOptions")); |
||||
optionsBuilder.indent(); |
||||
for (CodeBlock optionBlock : options) { |
||||
optionsBuilder.add(optionBlock); |
||||
optionsBuilder.add("\n"); |
||||
} |
||||
optionsBuilder.add(".build();\n"); |
||||
optionsBuilder.unindent(); |
||||
builder.add(optionsBuilder.build()); |
||||
|
||||
builder.addStatement("$1L = $1L.withOptions($2L)", aggregationVariableName, |
||||
context.localVariable("aggregationOptions")); |
||||
} |
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock aggregationStages(String stageListVariableName, Iterable<String> stages, int stageCount, |
||||
List<CodeBlock> arguments) { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
builder.addStatement("$T<$T> $L = new $T($L)", List.class, Object.class, stageListVariableName, ArrayList.class, |
||||
stageCount); |
||||
int stageCounter = 0; |
||||
|
||||
for (String stage : stages) { |
||||
String stageName = context.localVariable("stage_%s".formatted(stageCounter++)); |
||||
builder.add(MongoCodeBlocks.renderExpressionToDocument(stage, stageName, arguments)); |
||||
builder.addStatement("$L.add($L)", context.localVariable("stages"), stageName); |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock sortingStage(String sortProvider) { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.beginControlFlow("if ($L.isSorted())", sortProvider); |
||||
builder.addStatement("$1T $2L = new $1T()", Document.class, context.localVariable("sortDocument")); |
||||
builder.beginControlFlow("for ($T $L : $L)", Order.class, context.localVariable("order"), sortProvider); |
||||
builder.addStatement("$1L.append($2L.getProperty(), $2L.isAscending() ? 1 : -1);", |
||||
context.localVariable("sortDocument"), context.localVariable("order")); |
||||
builder.endControlFlow(); |
||||
builder.addStatement("stages.add(new $T($S, $L))", Document.class, "$sort", |
||||
context.localVariable("sortDocument")); |
||||
builder.endControlFlow(); |
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock pagingStage(String pageableProvider, boolean slice) { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.add(sortingStage(pageableProvider + ".getSort()")); |
||||
|
||||
builder.beginControlFlow("if ($L.isPaged())", pageableProvider); |
||||
builder.beginControlFlow("if ($L.getOffset() > 0)", pageableProvider); |
||||
builder.addStatement("$L.add($T.skip($L.getOffset()))", context.localVariable("stages"), Aggregation.class, |
||||
pageableProvider); |
||||
builder.endControlFlow(); |
||||
if (slice) { |
||||
builder.addStatement("$L.add($T.limit($L.getPageSize() + 1))", context.localVariable("stages"), |
||||
Aggregation.class, pageableProvider); |
||||
} else { |
||||
builder.addStatement("$L.add($T.limit($L.getPageSize()))", context.localVariable("stages"), Aggregation.class, |
||||
pageableProvider); |
||||
} |
||||
builder.endControlFlow(); |
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock limitingStage(String limitProvider) { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.beginControlFlow("if ($L.isLimited())", limitProvider); |
||||
builder.addStatement("$L.add($T.limit($L.max()))", context.localVariable("stages"), Aggregation.class, |
||||
limitProvider); |
||||
builder.endControlFlow(); |
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
} |
||||
} |
||||
@ -0,0 +1,100 @@
@@ -0,0 +1,100 @@
|
||||
/* |
||||
* 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.repository.aot; |
||||
|
||||
import java.util.Optional; |
||||
|
||||
import org.jspecify.annotations.NullUnmarked; |
||||
import org.springframework.data.mongodb.core.ExecutableRemoveOperation.ExecutableRemove; |
||||
import org.springframework.data.mongodb.core.MongoOperations; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryExecution.DeleteExecution; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryMethod; |
||||
import org.springframework.data.repository.aot.generate.AotQueryMethodGenerationContext; |
||||
import org.springframework.javapoet.CodeBlock; |
||||
import org.springframework.javapoet.CodeBlock.Builder; |
||||
import org.springframework.javapoet.TypeName; |
||||
import org.springframework.util.ClassUtils; |
||||
import org.springframework.util.ObjectUtils; |
||||
|
||||
/** |
||||
* @author Christoph Strobl |
||||
* @since 5.0 |
||||
*/ |
||||
class DeleteBlocks { |
||||
|
||||
@NullUnmarked |
||||
static class DeleteExecutionCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
private String queryVariableName; |
||||
|
||||
DeleteExecutionCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
DeleteExecutionCodeBlockBuilder referencing(String queryVariableName) { |
||||
|
||||
this.queryVariableName = queryVariableName; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
String mongoOpsRef = context.fieldNameOf(MongoOperations.class); |
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
Class<?> domainType = context.getRepositoryInformation().getDomainType(); |
||||
boolean isProjecting = context.getActualReturnType() != null |
||||
&& !ObjectUtils.nullSafeEquals(TypeName.get(domainType), context.getActualReturnType()); |
||||
|
||||
Object actualReturnType = isProjecting ? context.getActualReturnType().getType() : domainType; |
||||
|
||||
builder.add("\n"); |
||||
builder.addStatement("$1T<$2T> $3L = $4L.remove($2T.class)", ExecutableRemove.class, domainType, |
||||
context.localVariable("remover"), mongoOpsRef); |
||||
|
||||
DeleteExecution.Type type = DeleteExecution.Type.FIND_AND_REMOVE_ALL; |
||||
if (!queryMethod.isCollectionQuery()) { |
||||
if (!ClassUtils.isPrimitiveOrWrapper(context.getMethod().getReturnType())) { |
||||
type = DeleteExecution.Type.FIND_AND_REMOVE_ONE; |
||||
} else { |
||||
type = DeleteExecution.Type.ALL; |
||||
} |
||||
} |
||||
|
||||
actualReturnType = ClassUtils.isPrimitiveOrWrapper(context.getMethod().getReturnType()) |
||||
? TypeName.get(context.getMethod().getReturnType()) |
||||
: queryMethod.isCollectionQuery() ? context.getReturnTypeName() : actualReturnType; |
||||
|
||||
if (ClassUtils.isVoidType(context.getMethod().getReturnType())) { |
||||
builder.addStatement("new $T($L, $T.$L).execute($L)", DeleteExecution.class, context.localVariable("remover"), |
||||
DeleteExecution.Type.class, type.name(), queryVariableName); |
||||
} else if (context.getMethod().getReturnType() == Optional.class) { |
||||
builder.addStatement("return $T.ofNullable(($T) new $T($L, $T.$L).execute($L))", Optional.class, |
||||
actualReturnType, DeleteExecution.class, context.localVariable("remover"), DeleteExecution.Type.class, |
||||
type.name(), queryVariableName); |
||||
} else { |
||||
builder.addStatement("return ($T) new $T($L, $T.$L).execute($L)", actualReturnType, DeleteExecution.class, |
||||
context.localVariable("remover"), DeleteExecution.Type.class, type.name(), queryVariableName); |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,145 @@
@@ -0,0 +1,145 @@
|
||||
/* |
||||
* 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.repository.aot; |
||||
|
||||
import org.springframework.data.geo.Distance; |
||||
import org.springframework.data.geo.GeoPage; |
||||
import org.springframework.data.geo.GeoResults; |
||||
import org.springframework.data.mongodb.core.MongoOperations; |
||||
import org.springframework.data.mongodb.core.query.NearQuery; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryMethod; |
||||
import org.springframework.data.repository.aot.generate.AotQueryMethodGenerationContext; |
||||
import org.springframework.data.support.PageableExecutionUtils; |
||||
import org.springframework.javapoet.CodeBlock; |
||||
import org.springframework.util.ClassUtils; |
||||
|
||||
/** |
||||
* @author Christoph Strobl |
||||
* @since 5.0 |
||||
*/ |
||||
class GeoBlocks { |
||||
|
||||
static class GeoNearCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
|
||||
private String variableName; |
||||
|
||||
GeoNearCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
CodeBlock.Builder builder = CodeBlock.builder(); |
||||
builder.add("\n"); |
||||
|
||||
String locationParameterName = context.getParameterName(queryMethod.getParameters().getNearIndex()); |
||||
|
||||
builder.addStatement("$1T $2L = $1T.near($3L)", NearQuery.class, variableName, locationParameterName); |
||||
|
||||
if (queryMethod.getParameters().getRangeIndex() != -1) { |
||||
|
||||
String rangeParametername = context.getParameterName(queryMethod.getParameters().getRangeIndex()); |
||||
String minVarName = context.localVariable("min"); |
||||
String maxVarName = context.localVariable("max"); |
||||
|
||||
builder.beginControlFlow("if($L.getLowerBound().isBounded())", rangeParametername); |
||||
builder.addStatement("$1T $2L = $3L.getLowerBound().getValue().get()", Distance.class, minVarName, |
||||
rangeParametername); |
||||
builder.addStatement("$1L.minDistance($2L).in($2L.getMetric())", variableName, minVarName); |
||||
builder.endControlFlow(); |
||||
|
||||
builder.beginControlFlow("if($L.getUpperBound().isBounded())", rangeParametername); |
||||
builder.addStatement("$1T $2L = $3L.getUpperBound().getValue().get()", Distance.class, maxVarName, |
||||
rangeParametername); |
||||
builder.addStatement("$1L.maxDistance($2L).in($2L.getMetric())", variableName, maxVarName); |
||||
builder.endControlFlow(); |
||||
} else { |
||||
|
||||
String distanceParametername = context.getParameterName(queryMethod.getParameters().getMaxDistanceIndex()); |
||||
builder.addStatement("$1L.maxDistance($2L).in($2L.getMetric())", variableName, distanceParametername); |
||||
} |
||||
|
||||
if (context.getPageableParameterName() != null) { |
||||
builder.addStatement("$L.with($L)", variableName, context.getPageableParameterName()); |
||||
} |
||||
|
||||
MongoCodeBlocks.appendReadPreference(context, builder, variableName); |
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
public GeoNearCodeBlockBuilder usingQueryVariableName(String variableName) { |
||||
this.variableName = variableName; |
||||
return this; |
||||
} |
||||
} |
||||
|
||||
static class GeoNearExecutionCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
private String queryVariableName; |
||||
|
||||
GeoNearExecutionCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
GeoNearExecutionCodeBlockBuilder referencing(String queryVariableName) { |
||||
|
||||
this.queryVariableName = queryVariableName; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
CodeBlock.Builder builder = CodeBlock.builder(); |
||||
builder.add("\n"); |
||||
|
||||
String executorVar = context.localVariable("nearFinder"); |
||||
builder.addStatement("var $L = $L.query($T.class).near($L)", executorVar, |
||||
context.fieldNameOf(MongoOperations.class), context.getRepositoryInformation().getDomainType(), |
||||
queryVariableName); |
||||
|
||||
if (ClassUtils.isAssignable(GeoPage.class, context.getReturnType().getRawClass())) { |
||||
|
||||
String geoResultVar = context.localVariable("geoResult"); |
||||
builder.addStatement("var $L = $L.all()", geoResultVar, executorVar); |
||||
|
||||
builder.beginControlFlow("if($L.isUnpaged())", context.getPageableParameterName()); |
||||
builder.addStatement("return new $T<>($L)", GeoPage.class, geoResultVar); |
||||
builder.endControlFlow(); |
||||
|
||||
String pageVar = context.localVariable("resultPage"); |
||||
builder.addStatement("var $L = $T.getPage($L.getContent(), $L, () -> $L.count())", pageVar, |
||||
PageableExecutionUtils.class, geoResultVar, context.getPageableParameterName(), executorVar); |
||||
builder.addStatement("return new $T<>($L, $L, $L.getTotalElements())", GeoPage.class, geoResultVar, |
||||
context.getPageableParameterName(), pageVar); |
||||
} else if (ClassUtils.isAssignable(GeoResults.class, context.getReturnType().getRawClass())) { |
||||
builder.addStatement("return $L.all()", executorVar); |
||||
} else { |
||||
builder.addStatement("return $L.all().getContent()", executorVar); |
||||
} |
||||
return builder.build(); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,68 @@
@@ -0,0 +1,68 @@
|
||||
/* |
||||
* 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.repository.aot; |
||||
|
||||
import java.util.LinkedHashMap; |
||||
import java.util.Map; |
||||
|
||||
import org.springframework.data.mongodb.repository.query.MongoParameters; |
||||
import org.springframework.data.repository.aot.generate.QueryMetadata; |
||||
|
||||
/** |
||||
* An {@link MongoInteraction} to execute a query. |
||||
* |
||||
* @author Christoph Strobl |
||||
* @since 5.0 |
||||
*/ |
||||
class NearQueryInteraction extends MongoInteraction implements QueryMetadata { |
||||
|
||||
private final InteractionType interactionType; |
||||
private final QueryInteraction query; |
||||
private final MongoParameters parameters; |
||||
|
||||
NearQueryInteraction(QueryInteraction query, MongoParameters parameters) { |
||||
interactionType = InteractionType.QUERY; |
||||
this.query = query; |
||||
this.parameters = parameters; |
||||
} |
||||
|
||||
@Override |
||||
InteractionType getExecutionType() { |
||||
return interactionType; |
||||
} |
||||
|
||||
public QueryInteraction getQuery() { |
||||
return query; |
||||
} |
||||
|
||||
@Override |
||||
public Map<String, Object> serialize() { |
||||
|
||||
Map<String, Object> serialized = new LinkedHashMap<>(); |
||||
serialized.put("near", "?%s".formatted(parameters.getNearIndex())); |
||||
if (parameters.getRangeIndex() != -1) { |
||||
serialized.put("minDistance", "?%s".formatted(parameters.getRangeIndex())); |
||||
serialized.put("maxDistance", "?%s".formatted(parameters.getRangeIndex())); |
||||
} else if (parameters.getMaxDistanceIndex() != -1) { |
||||
serialized.put("minDistance", "?%s".formatted(parameters.getMaxDistanceIndex())); |
||||
} |
||||
Object filter = query.serialize().get("filter"); // TODO: filter position index can be off due to bindable params
|
||||
if (filter != null) { |
||||
serialized.put("filter", filter); |
||||
} |
||||
return serialized; |
||||
} |
||||
} |
||||
@ -0,0 +1,303 @@
@@ -0,0 +1,303 @@
|
||||
/* |
||||
* 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.repository.aot; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.Iterator; |
||||
import java.util.List; |
||||
import java.util.Optional; |
||||
|
||||
import org.bson.Document; |
||||
import org.jspecify.annotations.NullUnmarked; |
||||
import org.jspecify.annotations.Nullable; |
||||
import org.springframework.core.annotation.MergedAnnotation; |
||||
import org.springframework.data.geo.Box; |
||||
import org.springframework.data.geo.Circle; |
||||
import org.springframework.data.geo.Polygon; |
||||
import org.springframework.data.mongodb.core.ExecutableFindOperation.FindWithQuery; |
||||
import org.springframework.data.mongodb.core.MongoOperations; |
||||
import org.springframework.data.mongodb.core.geo.GeoJson; |
||||
import org.springframework.data.mongodb.core.geo.Sphere; |
||||
import org.springframework.data.mongodb.core.query.BasicQuery; |
||||
import org.springframework.data.mongodb.repository.Hint; |
||||
import org.springframework.data.mongodb.repository.Meta; |
||||
import org.springframework.data.mongodb.repository.query.MongoParameters.MongoParameter; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryExecution.PagedExecution; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryExecution.SlicedExecution; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryMethod; |
||||
import org.springframework.data.repository.aot.generate.AotQueryMethodGenerationContext; |
||||
import org.springframework.javapoet.CodeBlock; |
||||
import org.springframework.javapoet.CodeBlock.Builder; |
||||
import org.springframework.javapoet.TypeName; |
||||
import org.springframework.util.ClassUtils; |
||||
import org.springframework.util.NumberUtils; |
||||
import org.springframework.util.StringUtils; |
||||
|
||||
/** |
||||
* @author Christoph Strobl |
||||
* @since 5.0 |
||||
*/ |
||||
class QueryBlocks { |
||||
|
||||
@NullUnmarked |
||||
static class QueryExecutionCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
private QueryInteraction query; |
||||
|
||||
QueryExecutionCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
QueryExecutionCodeBlockBuilder forQuery(QueryInteraction query) { |
||||
|
||||
this.query = query; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
String mongoOpsRef = context.fieldNameOf(MongoOperations.class); |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
boolean isProjecting = context.getReturnedType().isProjecting(); |
||||
Class<?> domainType = context.getRepositoryInformation().getDomainType(); |
||||
Object actualReturnType = queryMethod.getParameters().hasDynamicProjection() || isProjecting |
||||
? TypeName.get(context.getActualReturnType().getType()) |
||||
: domainType; |
||||
|
||||
builder.add("\n"); |
||||
|
||||
if (queryMethod.getParameters().hasDynamicProjection()) { |
||||
builder.addStatement("$T<$T> $L = $L.query($T.class).as($L)", FindWithQuery.class, actualReturnType, |
||||
context.localVariable("finder"), mongoOpsRef, domainType, context.getDynamicProjectionParameterName()); |
||||
} else if (isProjecting) { |
||||
builder.addStatement("$T<$T> $L = $L.query($T.class).as($T.class)", FindWithQuery.class, actualReturnType, |
||||
context.localVariable("finder"), mongoOpsRef, domainType, actualReturnType); |
||||
} else { |
||||
|
||||
builder.addStatement("$T<$T> $L = $L.query($T.class)", FindWithQuery.class, actualReturnType, |
||||
context.localVariable("finder"), mongoOpsRef, domainType); |
||||
} |
||||
|
||||
String terminatingMethod; |
||||
|
||||
if (queryMethod.isCollectionQuery() || queryMethod.isPageQuery() || queryMethod.isSliceQuery()) { |
||||
terminatingMethod = "all()"; |
||||
} else if (query.isCount()) { |
||||
terminatingMethod = "count()"; |
||||
} else if (query.isExists()) { |
||||
terminatingMethod = "exists()"; |
||||
} else if (queryMethod.isStreamQuery()) { |
||||
terminatingMethod = "stream()"; |
||||
} else { |
||||
terminatingMethod = Optional.class.isAssignableFrom(context.getReturnType().toClass()) ? "one()" : "oneValue()"; |
||||
} |
||||
|
||||
if (queryMethod.isPageQuery()) { |
||||
builder.addStatement("return new $T($L, $L).execute($L)", PagedExecution.class, context.localVariable("finder"), |
||||
context.getPageableParameterName(), query.name()); |
||||
} else if (queryMethod.isSliceQuery()) { |
||||
builder.addStatement("return new $T($L, $L).execute($L)", SlicedExecution.class, |
||||
context.localVariable("finder"), context.getPageableParameterName(), query.name()); |
||||
} else if (queryMethod.isScrollQuery()) { |
||||
|
||||
String scrollPositionParameterName = context.getScrollPositionParameterName(); |
||||
|
||||
builder.addStatement("return $L.matching($L).scroll($L)", context.localVariable("finder"), query.name(), |
||||
scrollPositionParameterName); |
||||
} else { |
||||
if (query.isCount() && !ClassUtils.isAssignable(Long.class, context.getActualReturnType().getRawClass())) { |
||||
|
||||
Class<?> returnType = ClassUtils.resolvePrimitiveIfNecessary(queryMethod.getReturnedObjectType()); |
||||
builder.addStatement("return $T.convertNumberToTargetClass($L.matching($L).$L, $T.class)", NumberUtils.class, |
||||
context.localVariable("finder"), query.name(), terminatingMethod, returnType); |
||||
|
||||
} else { |
||||
builder.addStatement("return $L.matching($L).$L", context.localVariable("finder"), query.name(), |
||||
terminatingMethod); |
||||
} |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
} |
||||
|
||||
@NullUnmarked |
||||
static class QueryCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
|
||||
private QueryInteraction source; |
||||
private final List<CodeBlock> arguments; |
||||
private String queryVariableName; |
||||
|
||||
QueryCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
|
||||
this.arguments = new ArrayList<>(); |
||||
this.queryMethod = queryMethod; |
||||
collectArguments(context); |
||||
|
||||
} |
||||
|
||||
private void collectArguments(AotQueryMethodGenerationContext context) { |
||||
|
||||
for (MongoParameter parameter : queryMethod.getParameters().getBindableParameters()) { |
||||
String parameterName = context.getParameterName(parameter.getIndex()); |
||||
if (ClassUtils.isAssignable(GeoJson.class, parameter.getType())) { |
||||
|
||||
// renders as generic $geometry, thus can be handled by the converter when parsing
|
||||
arguments.add(CodeBlock.of(parameterName)); |
||||
} else if (ClassUtils.isAssignable(Circle.class, parameter.getType()) |
||||
|| ClassUtils.isAssignable(Sphere.class, parameter.getType())) { |
||||
|
||||
// $center | $centerSphere : [ [ <x>, <y> ], <radius> ]
|
||||
arguments.add(CodeBlock.builder().add( |
||||
"$1T.of($1T.of($2L.getCenter().getX(), $2L.getCenter().getY()), $2L.getRadius().getNormalizedValue())", |
||||
List.class, parameterName).build()); |
||||
} else if (ClassUtils.isAssignable(Box.class, parameter.getType())) { |
||||
|
||||
// $box: [ [ <x1>, <y1> ], [ <x2>, <y2> ] ]
|
||||
arguments.add(CodeBlock.builder().add( |
||||
"$1T.of($1T.of($2L.getFirst().getX(), $2L.getFirst().getY()), $1T.of($2L.getSecond().getX(), $2L.getSecond().getY()))", |
||||
List.class, parameterName).build()); |
||||
} else if (ClassUtils.isAssignable(Polygon.class, parameter.getType())) { |
||||
|
||||
// $polygon: [ [ <x1> , <y1> ], [ <x2> , <y2> ], [ <x3> , <y3> ], ... ]
|
||||
String localVar = context.localVariable("_p"); |
||||
arguments.add( |
||||
CodeBlock.builder().add("$1L.getPoints().stream().map($2L -> $3T.of($2L.getX(), $2L.getY())).toList()", |
||||
parameterName, localVar, List.class).build()); |
||||
} else { |
||||
arguments.add(CodeBlock.of(parameterName)); |
||||
} |
||||
} |
||||
} |
||||
|
||||
QueryCodeBlockBuilder filter(QueryInteraction query) { |
||||
|
||||
this.source = query; |
||||
return this; |
||||
} |
||||
|
||||
QueryCodeBlockBuilder usingQueryVariableName(String queryVariableName) { |
||||
|
||||
this.queryVariableName = queryVariableName; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.add("\n"); |
||||
builder.add(renderExpressionToQuery(source.getQuery().getQueryString(), queryVariableName)); |
||||
|
||||
if (StringUtils.hasText(source.getQuery().getFieldsString())) { |
||||
|
||||
builder |
||||
.add(MongoCodeBlocks.renderExpressionToDocument(source.getQuery().getFieldsString(), "fields", arguments)); |
||||
builder.addStatement("$L.setFieldsObject(fields)", queryVariableName); |
||||
} |
||||
|
||||
String sortParameter = context.getSortParameterName(); |
||||
if (StringUtils.hasText(sortParameter)) { |
||||
builder.addStatement("$L.with($L)", queryVariableName, sortParameter); |
||||
} else if (StringUtils.hasText(source.getQuery().getSortString())) { |
||||
|
||||
builder.add(MongoCodeBlocks.renderExpressionToDocument(source.getQuery().getSortString(), "sort", arguments)); |
||||
builder.addStatement("$L.setSortObject(sort)", queryVariableName); |
||||
} |
||||
|
||||
String limitParameter = context.getLimitParameterName(); |
||||
if (StringUtils.hasText(limitParameter)) { |
||||
builder.addStatement("$L.limit($L)", queryVariableName, limitParameter); |
||||
} else if (context.getPageableParameterName() == null && source.getQuery().isLimited()) { |
||||
builder.addStatement("$L.limit($L)", queryVariableName, source.getQuery().getLimit()); |
||||
} |
||||
|
||||
String pageableParameter = context.getPageableParameterName(); |
||||
if (StringUtils.hasText(pageableParameter) && !queryMethod.isPageQuery() && !queryMethod.isSliceQuery()) { |
||||
builder.addStatement("$L.with($L)", queryVariableName, pageableParameter); |
||||
} |
||||
|
||||
MergedAnnotation<Hint> hintAnnotation = context.getAnnotation(Hint.class); |
||||
String hint = hintAnnotation.isPresent() ? hintAnnotation.getString("value") : null; |
||||
|
||||
if (StringUtils.hasText(hint)) { |
||||
builder.addStatement("$L.withHint($S)", queryVariableName, hint); |
||||
} |
||||
|
||||
MongoCodeBlocks.appendReadPreference(context, builder, queryVariableName); |
||||
|
||||
MergedAnnotation<Meta> metaAnnotation = context.getAnnotation(Meta.class); |
||||
if (metaAnnotation.isPresent()) { |
||||
|
||||
long maxExecutionTimeMs = metaAnnotation.getLong("maxExecutionTimeMs"); |
||||
if (maxExecutionTimeMs != -1) { |
||||
builder.addStatement("$L.maxTimeMsec($L)", queryVariableName, maxExecutionTimeMs); |
||||
} |
||||
|
||||
int cursorBatchSize = metaAnnotation.getInt("cursorBatchSize"); |
||||
if (cursorBatchSize != 0) { |
||||
builder.addStatement("$L.cursorBatchSize($L)", queryVariableName, cursorBatchSize); |
||||
} |
||||
|
||||
String comment = metaAnnotation.getString("comment"); |
||||
if (StringUtils.hasText("comment")) { |
||||
builder.addStatement("$L.comment($S)", queryVariableName, comment); |
||||
} |
||||
} |
||||
|
||||
// TODO: Meta annotation: Disk usage
|
||||
|
||||
return builder.build(); |
||||
} |
||||
|
||||
private CodeBlock renderExpressionToQuery(@Nullable String source, String variableName) { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
if (!StringUtils.hasText(source)) { |
||||
|
||||
builder.addStatement("$1T $2L = new $1T(new $3T())", BasicQuery.class, variableName, Document.class); |
||||
} else if (!MongoCodeBlocks.containsPlaceholder(source)) { |
||||
builder.addStatement("$1T $2L = new $1T($3T.parse($4S))", BasicQuery.class, variableName, Document.class, |
||||
source); |
||||
} else { |
||||
builder.add("$T $L = createQuery($S, new $T[]{ ", BasicQuery.class, variableName, source, Object.class); |
||||
Iterator<CodeBlock> iterator = arguments.iterator(); |
||||
while (iterator.hasNext()) { |
||||
builder.add(iterator.next()); |
||||
if (iterator.hasNext()) { |
||||
builder.add(", "); |
||||
} else { |
||||
builder.add(" "); |
||||
} |
||||
} |
||||
builder.add("});\n"); |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,145 @@
@@ -0,0 +1,145 @@
|
||||
/* |
||||
* 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 |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
/* |
||||
* 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 |
||||
* |
||||
* 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.data.mongodb.repository.aot; |
||||
|
||||
import java.util.List; |
||||
import java.util.stream.Collectors; |
||||
|
||||
import org.jspecify.annotations.NullUnmarked; |
||||
import org.springframework.data.mongodb.core.ExecutableUpdateOperation.ExecutableUpdate; |
||||
import org.springframework.data.mongodb.core.MongoOperations; |
||||
import org.springframework.data.mongodb.core.query.BasicUpdate; |
||||
import org.springframework.data.mongodb.repository.query.MongoQueryMethod; |
||||
import org.springframework.data.repository.aot.generate.AotQueryMethodGenerationContext; |
||||
import org.springframework.data.util.ReflectionUtils; |
||||
import org.springframework.javapoet.CodeBlock; |
||||
import org.springframework.javapoet.CodeBlock.Builder; |
||||
import org.springframework.util.ClassUtils; |
||||
import org.springframework.util.NumberUtils; |
||||
|
||||
/** |
||||
* @author Christoph Strobl |
||||
* @since 2025/06 |
||||
*/ |
||||
class UpdateBlocks { |
||||
|
||||
@NullUnmarked |
||||
static class UpdateExecutionCodeBlockBuilder { |
||||
|
||||
private final AotQueryMethodGenerationContext context; |
||||
private final MongoQueryMethod queryMethod; |
||||
private String queryVariableName; |
||||
private String updateVariableName; |
||||
|
||||
UpdateExecutionCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
|
||||
this.context = context; |
||||
this.queryMethod = queryMethod; |
||||
} |
||||
|
||||
UpdateExecutionCodeBlockBuilder withFilter(String queryVariableName) { |
||||
|
||||
this.queryVariableName = queryVariableName; |
||||
return this; |
||||
} |
||||
|
||||
UpdateExecutionCodeBlockBuilder referencingUpdate(String updateVariableName) { |
||||
|
||||
this.updateVariableName = updateVariableName; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
String mongoOpsRef = context.fieldNameOf(MongoOperations.class); |
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.add("\n"); |
||||
|
||||
String updateReference = updateVariableName; |
||||
Class<?> domainType = context.getRepositoryInformation().getDomainType(); |
||||
builder.addStatement("$1T<$2T> $3L = $4L.update($2T.class)", ExecutableUpdate.class, domainType, |
||||
context.localVariable("updater"), mongoOpsRef); |
||||
|
||||
Class<?> returnType = ClassUtils.resolvePrimitiveIfNecessary(queryMethod.getReturnedObjectType()); |
||||
if (ReflectionUtils.isVoid(returnType)) { |
||||
builder.addStatement("$L.matching($L).apply($L).all()", context.localVariable("updater"), queryVariableName, |
||||
updateReference); |
||||
} else if (ClassUtils.isAssignable(Long.class, returnType)) { |
||||
builder.addStatement("return $L.matching($L).apply($L).all().getModifiedCount()", |
||||
context.localVariable("updater"), queryVariableName, updateReference); |
||||
} else { |
||||
builder.addStatement("$T $L = $L.matching($L).apply($L).all().getModifiedCount()", Long.class, |
||||
context.localVariable("modifiedCount"), context.localVariable("updater"), queryVariableName, |
||||
updateReference); |
||||
builder.addStatement("return $T.convertNumberToTargetClass($L, $T.class)", NumberUtils.class, |
||||
context.localVariable("modifiedCount"), returnType); |
||||
} |
||||
|
||||
return builder.build(); |
||||
} |
||||
} |
||||
|
||||
@NullUnmarked |
||||
static class UpdateCodeBlockBuilder { |
||||
|
||||
private UpdateInteraction source; |
||||
private List<CodeBlock> arguments; |
||||
private String updateVariableName; |
||||
|
||||
public UpdateCodeBlockBuilder(AotQueryMethodGenerationContext context, MongoQueryMethod queryMethod) { |
||||
this.arguments = context.getBindableParameterNames().stream().map(CodeBlock::of).collect(Collectors.toList()); |
||||
} |
||||
|
||||
public UpdateCodeBlockBuilder update(UpdateInteraction update) { |
||||
this.source = update; |
||||
return this; |
||||
} |
||||
|
||||
public UpdateCodeBlockBuilder usingUpdateVariableName(String updateVariableName) { |
||||
this.updateVariableName = updateVariableName; |
||||
return this; |
||||
} |
||||
|
||||
CodeBlock build() { |
||||
|
||||
Builder builder = CodeBlock.builder(); |
||||
|
||||
builder.add("\n"); |
||||
String tmpVariableName = updateVariableName + "Document"; |
||||
builder.add(MongoCodeBlocks.renderExpressionToDocument(source.getUpdate().getUpdateString(), tmpVariableName, arguments)); |
||||
builder.addStatement("$1T $2L = new $1T($3L)", BasicUpdate.class, updateVariableName, tmpVariableName); |
||||
|
||||
return builder.build(); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,26 @@
@@ -0,0 +1,26 @@
|
||||
/* |
||||
* 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 example.aot; |
||||
|
||||
import org.springframework.data.geo.Point; |
||||
|
||||
/** |
||||
* @param planet |
||||
* @param coordinates |
||||
* @author Christoph Strobl |
||||
*/ |
||||
public record Location(String planet, Point coordinates) { |
||||
} |
||||
@ -0,0 +1,222 @@
@@ -0,0 +1,222 @@
|
||||
/* |
||||
* 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.repository.aot; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
import example.aot.User; |
||||
import example.aot.UserRepository; |
||||
|
||||
import java.lang.reflect.Method; |
||||
import java.util.Arrays; |
||||
|
||||
import javax.lang.model.element.Modifier; |
||||
|
||||
import org.junit.jupiter.api.Assertions; |
||||
import org.junit.jupiter.api.Test; |
||||
import org.springframework.data.domain.Pageable; |
||||
import org.springframework.data.domain.Range; |
||||
import org.springframework.data.geo.Box; |
||||
import org.springframework.data.geo.Circle; |
||||
import org.springframework.data.geo.Distance; |
||||
import org.springframework.data.geo.GeoResults; |
||||
import org.springframework.data.geo.Point; |
||||
import org.springframework.data.geo.Polygon; |
||||
import org.springframework.data.mongodb.core.MongoOperations; |
||||
import org.springframework.data.mongodb.core.geo.GeoJsonPolygon; |
||||
import org.springframework.data.mongodb.core.geo.Sphere; |
||||
import org.springframework.data.mongodb.repository.ReadPreference; |
||||
import org.springframework.data.repository.Repository; |
||||
import org.springframework.data.repository.aot.generate.AotQueryMethodGenerationContext; |
||||
import org.springframework.data.repository.aot.generate.AotRepositoryFragmentMetadata; |
||||
import org.springframework.data.repository.aot.generate.MethodContributor; |
||||
import org.springframework.data.repository.core.RepositoryInformation; |
||||
import org.springframework.data.repository.query.QueryMethod; |
||||
import org.springframework.javapoet.ClassName; |
||||
import org.springframework.javapoet.FieldSpec; |
||||
import org.springframework.javapoet.MethodSpec; |
||||
|
||||
/** |
||||
* @author Christoph Strobl |
||||
*/ |
||||
public class QueryMethodContributionUnitTests { |
||||
|
||||
@Test // GH-5004
|
||||
void rendersQueryForNearUsingPoint() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesNear", Point.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("{'location.coordinates':{'$near':?0}}") //
|
||||
.contains("Object[]{ location }") //
|
||||
.contains("return finder.matching(filterQuery).all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersQueryForWithinUsingCircle() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesWithin", Circle.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("{'location.coordinates':{'$geoWithin':{'$center':?0}}") //
|
||||
.contains( |
||||
"List.of(circle.getCenter().getX(), circle.getCenter().getY()), circle.getRadius().getNormalizedValue())") //
|
||||
.contains("return finder.matching(filterQuery).all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersQueryForWithinUsingSphere() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesWithin", Sphere.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("{'location.coordinates':{'$geoWithin':{'$centerSphere':?0}}") //
|
||||
.contains( |
||||
"List.of(circle.getCenter().getX(), circle.getCenter().getY()), circle.getRadius().getNormalizedValue())") //
|
||||
.contains("return finder.matching(filterQuery).all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersQueryForWithinUsingBox() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesWithin", Box.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("{'location.coordinates':{'$geoWithin':{'$box':?0}}") //
|
||||
.contains("List.of(box.getFirst().getX(), box.getFirst().getY())") //
|
||||
.contains("List.of(box.getSecond().getX(), box.getSecond().getY())") //
|
||||
.contains("return finder.matching(filterQuery).all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersQueryForWithinUsingPolygon() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesWithin", Polygon.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("{'location.coordinates':{'$geoWithin':{'$polygon':?0}}") //
|
||||
.contains("polygon.getPoints().stream().map(_p ->") //
|
||||
.contains("List.of(_p.getX(), _p.getY())") //
|
||||
.contains("return finder.matching(filterQuery).all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersQueryForWithinUsingGeoJsonPolygon() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesWithin", GeoJsonPolygon.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("{'location.coordinates':{'$geoWithin':{'$geometry':?0}}") //
|
||||
.contains("Object[]{ polygon }") //
|
||||
.contains("return finder.matching(filterQuery).all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersNearQueryForGeoResults() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepoWithMeta.class, "findByLocationCoordinatesNear", Point.class, |
||||
Distance.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("NearQuery.near(point)") //
|
||||
.contains("nearQuery.maxDistance(maxDistance).in(maxDistance.getMetric())") //
|
||||
.contains(".withReadPreference(com.mongodb.ReadPreference.valueOf(\"NEAREST\")") //
|
||||
.doesNotContain("nearQuery.query(") //
|
||||
.contains(".near(nearQuery)") //
|
||||
.contains("return nearFinder.all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersNearQueryWithDistanceRangeForGeoResults() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesNear", Point.class, Range.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("NearQuery.near(point)") //
|
||||
.contains("if(distance.getLowerBound().isBounded())") //
|
||||
.contains("nearQuery.minDistance(min).in(min.getMetric())") //
|
||||
.contains("if(distance.getUpperBound().isBounded())") //
|
||||
.contains("nearQuery.maxDistance(max).in(max.getMetric())") //
|
||||
.contains(".near(nearQuery)") //
|
||||
.contains("return nearFinder.all()"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersNearQueryReturningGeoPage() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesNear", Point.class, Distance.class, |
||||
Pageable.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("NearQuery.near(point)") //
|
||||
.contains("nearQuery.maxDistance(maxDistance).in(maxDistance.getMetric())") //
|
||||
.doesNotContain("nearQuery.query(") //
|
||||
.contains("var geoResult = nearFinder.all()") //
|
||||
.contains("PageableExecutionUtils.getPage(geoResult.getContent(), pageable, () -> nearFinder.count())") |
||||
.contains("GeoPage<>(geoResult, pageable, resultPage.getTotalElements())"); |
||||
} |
||||
|
||||
@Test // GH-5004
|
||||
void rendersNearQueryWithFilterForGeoResults() throws NoSuchMethodException { |
||||
|
||||
MethodSpec methodSpec = codeOf(UserRepository.class, "findByLocationCoordinatesNearAndLastname", Point.class, |
||||
Distance.class, String.class); |
||||
|
||||
assertThat(methodSpec.toString()) //
|
||||
.contains("NearQuery.near(point)") //
|
||||
.contains("nearQuery.maxDistance(maxDistance).in(maxDistance.getMetric())") //
|
||||
.contains("filterQuery = createQuery(\"{'lastname':?0}\", new java.lang.Object[]{ lastname })") //
|
||||
.contains("nearQuery.query(filterQuery)") //
|
||||
.contains(".near(nearQuery)") //
|
||||
.contains("return nearFinder.all()"); |
||||
} |
||||
|
||||
private static MethodSpec codeOf(Class<?> repository, String methodName, Class<?>... args) |
||||
throws NoSuchMethodException { |
||||
|
||||
Method method = repository.getMethod(methodName, args); |
||||
|
||||
TestMongoAotRepositoryContext repoContext = new TestMongoAotRepositoryContext(repository, null); |
||||
MongoRepositoryContributor contributor = new MongoRepositoryContributor(repoContext); |
||||
MethodContributor<? extends QueryMethod> methodContributor = contributor.contributeQueryMethod(method); |
||||
|
||||
if (methodContributor == null) { |
||||
Assertions.fail("No contribution for method %s.%s(%s)".formatted(repository.getSimpleName(), methodName, |
||||
Arrays.stream(args).map(Class::getSimpleName).toList())); |
||||
} |
||||
AotRepositoryFragmentMetadata metadata = new AotRepositoryFragmentMetadata(ClassName.get(UserRepository.class)); |
||||
metadata.addField( |
||||
FieldSpec.builder(MongoOperations.class, "mongoOperations", Modifier.PRIVATE, Modifier.FINAL).build()); |
||||
|
||||
TestQueryMethodGenerationContext methodContext = new TestQueryMethodGenerationContext( |
||||
repoContext.getRepositoryInformation(), method, methodContributor.getQueryMethod(), metadata); |
||||
return methodContributor.contribute(methodContext); |
||||
} |
||||
|
||||
static class TestQueryMethodGenerationContext extends AotQueryMethodGenerationContext { |
||||
|
||||
protected TestQueryMethodGenerationContext(RepositoryInformation repositoryInformation, Method method, |
||||
QueryMethod queryMethod, AotRepositoryFragmentMetadata targetTypeMetadata) { |
||||
super(repositoryInformation, method, queryMethod, targetTypeMetadata); |
||||
} |
||||
} |
||||
|
||||
interface UserRepoWithMeta extends Repository<User, String> { |
||||
|
||||
@ReadPreference("NEAREST") |
||||
GeoResults<User> findByLocationCoordinatesNear(Point point, Distance maxDistance); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue