Commit c32ae0d

committed

Add MethodParameter accessor to AotQueryMethodGenerationContext.

Closes #3334

1 parent c21409c commit c32ae0dCopy full SHA for c32ae0d

File tree

+19

-0

lines changed

src/main/java/org/springframework/data/repository/aot/generate
- AotQueryMethodGenerationContext.java
- MethodMetadata.java

+19

-0

lines changed

Lines changed: 13 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,8 @@`
`21`	`21`	`import java.util.List;`
`22`	`22`
`23`	`23`	`import org.jspecify.annotations.Nullable;`
	`24`	`+`
	`25`	`+import org.springframework.core.MethodParameter;`
`24`	`26`	`import org.springframework.core.ResolvableType;`
`25`	`27`	`import org.springframework.core.annotation.MergedAnnotation;`
`26`	`28`	`import org.springframework.core.annotation.MergedAnnotationSelectors;`
`@@ -65,6 +67,17 @@ MethodMetadata getTargetMethodMetadata() {`
`65`	`67`	`return targetMethodMetadata;`
`66`	`68`	`}`
`67`	`69`
	`70`	`+ /**`
	`71`	`+ * Lookup the {@link MethodParameter} by its {@link MethodParameter#getParameterName()}.`
	`72`	`+ *`
	`73`	`+ * @param name the name of the parameter to look up.`
	`74`	`+ * @return the found method parameter or {@literal null} if no parameter with the given name exists.`
	`75`	`+ */`
	`76`	`+ @Nullable`
	`77`	`+ public MethodParameter getMethodParameter(String name) {`
	`78`	`+ return getTargetMethodMetadata().getMethodParameters().get(name);`
	`79`	`+ }`
	`80`	`+`
`68`	`81`	`public RepositoryInformation getRepositoryInformation() {`
`69`	`82`	`return repositoryInformation;`
`70`	`83`	`}`

Lines changed: 6 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,7 @@`
`41`	`41`	`class MethodMetadata {`
`42`	`42`
`43`	`43`	`private final Map<String, ParameterSpec> methodArguments = new LinkedHashMap<>();`
	`44`	`+ private final Map<String, MethodParameter> methodParameters = new LinkedHashMap<>();`
`44`	`45`	`private final Map<String, String> localVariables = new LinkedHashMap<>();`
`45`	`46`	`private final ResolvableType actualReturnType;`
`46`	`47`	`private final ResolvableType returnType;`
`@@ -66,6 +67,7 @@ private void initParameters(RepositoryInformation repositoryInformation, Method`
`66`	`67`	`TypeName parameterType = TypeName.get(resolvableParameterType.getType());`
`67`	`68`
`68`	`69`	`addParameter(ParameterSpec.builder(parameterType, methodParameter.getParameterName()).build());`
	`70`	`+ methodParameters.put(methodParameter.getParameterName(), methodParameter);`
`69`	`71`	`}`
`70`	`72`	`}`
`71`	`73`
`@@ -85,6 +87,10 @@ Map<String, ParameterSpec> getMethodArguments() {`
`85`	`87`	`return methodArguments;`
`86`	`88`	`}`
`87`	`89`
	`90`	`+ Map<String, MethodParameter> getMethodParameters() {`
	`91`	`+ return methodParameters;`
	`92`	`+ }`
	`93`	`+`
`88`	`94`	`@Nullable`
`89`	`95`	`String getParameterName(int position) {`
`90`	`96`

Comments

(0)