support for JPA 2.0 TypedQuery interface and query vendor interfaces (SPR-6733)

jhoeller · jhoeller · commit bcfef8a4e171 · 2010-01-21T12:04:26.000Z
diff --git a/org.springframework.orm/src/main/java/org/springframework/orm/jpa/SharedEntityManagerCreator.java b/org.springframework.orm/src/main/java/org/springframework/orm/jpa/SharedEntityManagerCreator.java
@@ -28,6 +28,7 @@
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 
+import org.springframework.util.ClassUtils;
 import org.springframework.util.CollectionUtils;
 
 /**
@@ -130,9 +131,17 @@ private static class SharedEntityManagerInvocationHandler implements InvocationH
 
 		private final Map properties;
 
+		private final ClassLoader proxyClassLoader;
+
 		public SharedEntityManagerInvocationHandler(EntityManagerFactory target, Map properties) {
 			this.targetFactory = target;
 			this.properties = properties;
+			if (this.targetFactory instanceof EntityManagerFactoryInfo) {
+				this.proxyClassLoader = ((EntityManagerFactoryInfo) this.targetFactory).getBeanClassLoader();
+			}
+			else {
+				this.proxyClassLoader = EntityManagerFactory.class.getClassLoader();
+			}
 		}
 
 		public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
@@ -225,7 +234,8 @@ else if (method.getName().equals("unwrap")) {
 				if (result instanceof Query) {
 					Query query = (Query) result;
 					if (isNewEm) {
-						result = Proxy.newProxyInstance(Query.class.getClassLoader(), new Class[] {Query.class},
+						Class[] ifcs = ClassUtils.getAllInterfacesForClass(query.getClass(), this.proxyClassLoader);
+						result = Proxy.newProxyInstance(this.proxyClassLoader, ifcs,
 								new DeferredQueryInvocationHandler(query, target));
 						isNewEm = false;
 					}
@@ -257,7 +267,7 @@ private static class DeferredQueryInvocationHandler implements InvocationHandler
 
 		private final EntityManager em;
 
-		private DeferredQueryInvocationHandler(Query target, EntityManager em) {
+		public DeferredQueryInvocationHandler(Query target, EntityManager em) {
 			this.target = target;
 			this.em = em;
 		}
