=== added file 'dhis-2/dhis-support/dhis-support-system/src/main/java/org/hisp/dhis/system/util/HibernateUtils.java' --- dhis-2/dhis-support/dhis-support-system/src/main/java/org/hisp/dhis/system/util/HibernateUtils.java 1970-01-01 00:00:00 +0000 +++ dhis-2/dhis-support/dhis-support-system/src/main/java/org/hisp/dhis/system/util/HibernateUtils.java 2016-01-06 07:05:53 +0000 @@ -0,0 +1,82 @@ +package org.hisp.dhis.system.util; + +/* + * Copyright (c) 2004-2016, University of Oslo + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * Neither the name of the HISP project nor the names of its contributors may + * be used to endorse or promote products derived from this software without + * specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR + * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON + * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +import org.hibernate.Hibernate; +import org.hibernate.collection.internal.PersistentSet; +import org.hibernate.collection.spi.PersistentCollection; +import org.hibernate.proxy.HibernateProxy; + +/** + * @author Morten Olav Hansen + */ +public class HibernateUtils +{ + public static boolean isProxy( Object object ) + { + return object != null && ((object instanceof HibernateProxy) || (object instanceof PersistentCollection)); + } + + /** + * If object is proxy, get unwrapped non-proxy object. + * + * @param proxy Object to check and unwrap + * @return Unwrapped object if proxyied, if not just returns same object + */ + @SuppressWarnings( "unchecked" ) + public static T unwrap( T proxy ) + { + if ( !isProxy( proxy ) ) + { + return proxy; + } + + Hibernate.initialize( proxy ); + + if ( HibernateProxy.class.isInstance( proxy ) ) + { + return (T) ((HibernateProxy) proxy).getHibernateLazyInitializer().getImplementation(); + } + + if ( PersistentCollection.class.isInstance( proxy ) ) + { + PersistentCollection persistentCollection = (PersistentCollection) proxy; + + if ( !PersistentSet.class.isInstance( persistentCollection ) ) + { + return (T) persistentCollection.getStoredSnapshot(); + } + + // for now just return same as for non-sets, but PersistentSets might require a bit of extra work. + return (T) persistentCollection.getStoredSnapshot(); + } + + return proxy; + } +}