diff --git a/src/main/java/org/springframework/data/repository/query/ExtensionAwareQueryMethodEvaluationContextProvider.java b/src/main/java/org/springframework/data/repository/query/ExtensionAwareQueryMethodEvaluationContextProvider.java index e2b47cd6b..6b6c56c33 100644 --- a/src/main/java/org/springframework/data/repository/query/ExtensionAwareQueryMethodEvaluationContextProvider.java +++ b/src/main/java/org/springframework/data/repository/query/ExtensionAwareQueryMethodEvaluationContextProvider.java @@ -20,6 +20,7 @@ import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Optional; +import java.util.function.Function; import java.util.stream.Collectors; import java.util.stream.Stream; @@ -145,7 +146,7 @@ public class ExtensionAwareQueryMethodEvaluationContextProvider implements Query private static final Map methodCache = new ConcurrentReferenceHashMap(); private final Object target; - private final Map> directMappings = new HashMap<>(); + private final Map> directMappings = new HashMap<>(); DelegatingMethodInterceptor(Object target) { this.target = target; @@ -158,7 +159,7 @@ public class ExtensionAwareQueryMethodEvaluationContextProvider implements Query * @param methodName * @param mapping */ - public void registerResultMapping(String methodName, java.util.function.Function mapping) { + public void registerResultMapping(String methodName, Function mapping) { this.directMappings.put(methodName, mapping); } @@ -185,7 +186,7 @@ public class ExtensionAwareQueryMethodEvaluationContextProvider implements Query return result; } - java.util.function.Function mapper = directMappings.get(targetMethod.getName()); + Function mapper = this.directMappings.get(targetMethod.getName()); return mapper != null ? mapper.apply(result) : result; }