From 7929feaa40f966dd38323673d84fc47780035395 Mon Sep 17 00:00:00 2001
From: Oliver Gierke <ogierke@pivotal.io>
Date: Mon, 29 Jun 2015 17:24:59 +0200
Subject: [PATCH] DATACMNS-293 - Polished AnnotationBasedQueryAugmentor.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Renamed variables to better express what they stand for. Added hook to forward augmentNativeQuery(…) to prepareNativeQuery(…).
---
 .../AnnotationBasedQueryAugmentor.java        | 24 ++++++++++++-------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/src/main/java/org/springframework/data/repository/augment/AnnotationBasedQueryAugmentor.java b/src/main/java/org/springframework/data/repository/augment/AnnotationBasedQueryAugmentor.java
index 60d3143f0..9ec1498ac 100644
--- a/src/main/java/org/springframework/data/repository/augment/AnnotationBasedQueryAugmentor.java
+++ b/src/main/java/org/springframework/data/repository/augment/AnnotationBasedQueryAugmentor.java
@@ -103,11 +103,11 @@ public abstract class AnnotationBasedQueryAugmentor<T extends Annotation, Q exte
 	 */
 	private T findAndCache(Class<?> type, Method method) {
 
-		T expression = AnnotationUtils.findAnnotation(type, annotationType);
+		T annotation = AnnotationUtils.findAnnotation(type, annotationType);
 
-		if (expression != null) {
-			cache.put(method, expression);
-			return expression;
+		if (annotation != null) {
+			cache.put(method, annotation);
+			return annotation;
 		}
 
 		return null;
@@ -118,7 +118,15 @@ public abstract class AnnotationBasedQueryAugmentor<T extends Annotation, Q exte
 	 * @see org.springframework.data.repository.augment.QueryAugmentor#augmentNativeQuery(org.springframework.data.repository.augment.QueryContext, org.springframework.data.repository.augment.MethodMetadata)
 	 */
 	public final N augmentNativeQuery(N context, MethodMetadata metadata) {
-		return null;
+
+		Method method = metadata.getMethod();
+
+		if (cache.containsKey(method)) {
+			return prepareNativeQuery(context, cache.get(method));
+		}
+
+		T expression = findAndCacheAnnotation(metadata);
+		return expression == null ? context : prepareNativeQuery(context, expression);
 	}
 
 	/*
@@ -153,7 +161,7 @@ public abstract class AnnotationBasedQueryAugmentor<T extends Annotation, Q exte
 		return expression == null ? update : prepareUpdate(update, cache.get(method));
 	}
 
-	protected N prepareNativeQuery(N context, T expression) {
+	protected N prepareNativeQuery(N context, T annotation) {
 		return context;
 	}
 
@@ -162,10 +170,10 @@ public abstract class AnnotationBasedQueryAugmentor<T extends Annotation, Q exte
 	 * returns the context as is.
 	 * 
 	 * @param context will never be {@literal null}.
-	 * @param expression will never be {@literal null}.
+	 * @param annotation will never be {@literal null}.
 	 * @return
 	 */
-	protected Q prepareQuery(Q context, T expression) {
+	protected Q prepareQuery(Q context, T annotation) {
 		return context;
 	}