From 4c918cd4e02abff236f2986817d1205d6e75b568 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Timo=20Westk=C3=A4mper?= <timo.westkamper@mysema.com>
Date: Wed, 17 Jul 2013 17:39:59 +0300
Subject: [PATCH] #457 Add QueryHandler usage

---
 .../mysema/query/jpa/sql/AbstractJPASQLQuery.java | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/querydsl-jpa/src/main/java/com/mysema/query/jpa/sql/AbstractJPASQLQuery.java b/querydsl-jpa/src/main/java/com/mysema/query/jpa/sql/AbstractJPASQLQuery.java
index 03c40e6e7..a98714fb8 100644
--- a/querydsl-jpa/src/main/java/com/mysema/query/jpa/sql/AbstractJPASQLQuery.java
+++ b/querydsl-jpa/src/main/java/com/mysema/query/jpa/sql/AbstractJPASQLQuery.java
@@ -152,14 +152,15 @@ public abstract class AbstractJPASQLQuery<Q extends AbstractJPASQLQuery<Q> & com
 
         FactoryExpression<?> wrapped = projection.size() > 1 ? FactoryExpressionUtils.wrap(projection) : null;
         if ((projection.size() == 1 && projection.get(0) instanceof FactoryExpression) || wrapped != null) {
+            Expression<?> expr = wrapped != null ? wrapped : projection.get(0);
 
-            // TODO : add conversion logic like in AbstractJPAQuery
-
-            this.projection = (FactoryExpression)projection.get(0);
-            if (wrapped != null) {
-                this.projection = wrapped;
-                getMetadata().clearProjection();
-                getMetadata().addProjection(wrapped);
+            if (!queryHandler.transform(query, (FactoryExpression<?>)expr)) {
+                this.projection = (FactoryExpression)projection.get(0);
+                if (wrapped != null) {
+                    this.projection = wrapped;
+                    getMetadata().clearProjection();
+                    getMetadata().addProjection(wrapped);
+                }
             }
         }