added handy unwrap proxy helper

jpos · Dec 17, 2015 · 7e6bc6f · 7e6bc6f
1 parent 2bfea2c
commit 7e6bc6f
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/modules/dbsupport/src/main/java/org/jpos/ee/DB.java b/modules/dbsupport/src/main/java/org/jpos/ee/DB.java
@@ -22,17 +22,15 @@
 import org.dom4j.DocumentException;
 import org.dom4j.Element;
 import org.dom4j.io.SAXReader;
-import org.hibernate.HibernateException;
-import org.hibernate.Session;
-import org.hibernate.SessionFactory;
-import org.hibernate.Transaction;
+import org.hibernate.*;
 import org.hibernate.boot.MetadataSources;
 import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
 import org.hibernate.boot.spi.MetadataImplementor;
 import org.hibernate.cfg.Configuration;
 import org.hibernate.engine.spi.CollectionKey;
 import org.hibernate.engine.spi.EntityKey;
 import org.hibernate.internal.util.ReflectHelper;
+import org.hibernate.proxy.HibernateProxy;
 import org.hibernate.resource.transaction.spi.TransactionStatus;
 import org.hibernate.stat.SessionStatistics;
 import org.hibernate.tool.hbm2ddl.SchemaExport;
@@ -375,6 +373,14 @@ public static Object execWithTransaction(DBAction action)
         }
     }
 
+    @SuppressWarnings("unchecked")
+    public static <T> T unwrap (T proxy) {
+        Hibernate.getClass(proxy);
+        Hibernate.initialize(proxy);
+        return (proxy instanceof HibernateProxy) ?
+          (T) ((HibernateProxy) proxy).getHibernateLazyInitializer().getImplementation() : proxy;
+    }
+
     @SuppressWarnings({"unchecked"})
     public void printStats()
     {