From fcdba49c352e5ec522608bd2b832944313a307d1 Mon Sep 17 00:00:00 2001 From: Oliver Gierke Date: Thu, 31 Jul 2014 20:10:52 +0200 Subject: [PATCH] DATACMNS-293 - Setting QueryAugementationEngine on queries. --- .../repository/core/support/RepositoryFactorySupport.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/main/java/org/springframework/data/repository/core/support/RepositoryFactorySupport.java b/src/main/java/org/springframework/data/repository/core/support/RepositoryFactorySupport.java index 92ad9d2b6..eff6a0530 100644 --- a/src/main/java/org/springframework/data/repository/core/support/RepositoryFactorySupport.java +++ b/src/main/java/org/springframework/data/repository/core/support/RepositoryFactorySupport.java @@ -84,8 +84,8 @@ public abstract class RepositoryFactorySupport implements BeanClassLoaderAware { private ClassLoader classLoader = org.springframework.util.ClassUtils.getDefaultClassLoader(); private EvaluationContextProvider evaluationContextProvider = DefaultEvaluationContextProvider.INSTANCE; - private QueryCollectingQueryCreationListener collectingListener = new QueryCollectingQueryCreationListener(); private QueryAugmentationEngine augmentationEngine = QueryAugmentationEngine.NONE; + private QueryCollectingQueryCreationListener collectingListener = new QueryCollectingQueryCreationListener(); public RepositoryFactorySupport() { this.queryPostProcessors.add(collectingListener); @@ -441,6 +441,11 @@ public abstract class RepositoryFactorySupport implements BeanClassLoaderAware { for (Method method : queryMethods) { RepositoryQuery query = lookupStrategy.resolveQuery(method, repositoryInformation, namedQueries); + + if (query instanceof QueryAugmentationEngineAware) { + ((QueryAugmentationEngineAware) query).setQueryAugmentationEngine(augmentationEngine); + } + invokeListeners(query); queries.put(method, query); }