Fixed class loading bug with proxies

Blazebit · Jul 22, 2014 · f343556 · f343556
1 parent bcb2d1b
commit f343556
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/entity-view/impl/src/main/java/com/blazebit/persistence/view/impl/proxy/ProxyFactory.java b/entity-view/impl/src/main/java/com/blazebit/persistence/view/impl/proxy/ProxyFactory.java
@@ -28,6 +28,8 @@
 import java.util.concurrent.ConcurrentMap;
 import java.util.concurrent.atomic.AtomicInteger;
 import javassist.CannotCompileException;
+import javassist.ClassClassPath;
+import javassist.ClassPath;
 import javassist.ClassPool;
 import javassist.CtClass;
 import javassist.CtConstructor;
@@ -73,6 +75,9 @@ private static <T> Class<? extends T> createProxyClass(ViewType<T> viewType) {
         CtClass cc = pool.makeClass(clazz.getName() + "_$$_javassist_entityview_" + classCounter.getAndIncrement());
         CtClass superCc;
 
+        ClassPath classPath = new ClassClassPath(clazz);
+        pool.insertClassPath(classPath);
+
         try {
             superCc = pool.get(clazz.getName());
 
@@ -157,6 +162,8 @@ private static <T> Class<? extends T> createProxyClass(ViewType<T> viewType) {
             return cc.toClass();
         } catch (Exception ex) {
             throw new RuntimeException("Probably we did something wrong, please contact us if you see this message.", ex);
+        } finally {
+            pool.removeClassPath(classPath);
         }
     }