Hibernate/hibernate-orm
1
0
Fork 0
mirror of https://github.com/hibernate/hibernate-orm synced 2025-02-17 16:44:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

HHH-12982 - Generify Hibernate#unproxy

Browse Source
This commit is contained in:
Vlad Mihalcea 2018-09-19 10:26:16 +03:00
parent 340618c9a0
commit 01bf67e4b8
2 changed files with 195 additions and 122 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
hibernate-core/src/main/java/org/hibernate/Hibernate.java
View File

@ -212,4 +212,16 @@ public static Object unproxy(Object proxy) {
return proxy; return proxy;
} }
} }
/**
* Unproxies a {@link HibernateProxy}. If the proxy is uninitialized, it automatically triggers an initialization.
* In case the supplied object is null or not a proxy, the object will be returned as-is.
*
* @param proxy the {@link HibernateProxy} to be unproxied
* @param entityClass the entity type
* @return the proxy's underlying implementation object, or the supplied object otherwise
*/
public static <T> T unproxy(T proxy, Class<T> entityClass) {
return entityClass.cast( unproxy( proxy ) );
}
} }

305
hibernate-core/src/test/java/org/hibernate/test/proxy/HibernateUnproxyTest.java
View File

@ -6,152 +6,213 @@
*/ */
package org.hibernate.test.proxy; package org.hibernate.test.proxy;
import java.util.Objects;
import javax.persistence.CascadeType;
import javax.persistence.Entity;
import javax.persistence.FetchType;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import javax.persistence.OneToOne;
import org.hibernate.Hibernate; import org.hibernate.Hibernate;
import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase; import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase;
import org.jboss.logging.Logger;
import org.junit.Test; import org.junit.Test;
import javax.persistence.*;
import java.util.Objects;
import static org.hibernate.testing.transaction.TransactionUtil.doInJPA; import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;
import static org.junit.Assert.*; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
public class HibernateUnproxyTest extends BaseEntityManagerFunctionalTestCase { public class HibernateUnproxyTest extends BaseEntityManagerFunctionalTestCase {
@Override @Override
protected Class<?>[] getAnnotatedClasses() { protected Class<?>[] getAnnotatedClasses() {
return new Class<?>[]{Parent.class, Child.class}; return new Class<?>[] { Parent.class, Child.class };
} }
@Test @Test
public void testInitializedProxyCanBeUnproxied() { public void testInitializedProxyCanBeUnproxied() {
Parent p = new Parent(); Parent p = new Parent();
Child c = new Child(); Child c = new Child();
p.setChild(c); p.setChild( c );
doInJPA(this::entityManagerFactory, (entityManager -> {
entityManager.persist(p);
}));
doInJPA(this::entityManagerFactory, (entityManager -> {
Parent parent = entityManager.find(Parent.class, p.getId());
Child child = parent.getChild();
assertFalse(Hibernate.isInitialized(child));
Hibernate.initialize(child);
Child unproxiedChild = (Child) Hibernate.unproxy(child);
assertEquals(Child.class, unproxiedChild.getClass());
}));
}
@Test doInJPA( this::entityManagerFactory, ( entityManager -> {
public void testNotInitializedProxyCanBeUnproxiedWithInitialization() { entityManager.persist( p );
Parent p = new Parent(); } ) );
Child c = new Child();
p.setChild(c);
doInJPA(this::entityManagerFactory, (entityManager -> {
entityManager.persist(p);
}));
doInJPA(this::entityManagerFactory, (entityManager -> {
Parent parent = entityManager.find(Parent.class, p.getId());
Child child = parent.getChild();
assertFalse(Hibernate.isInitialized(child));
Child unproxiedChild = (Child) Hibernate.unproxy(child);
assertTrue(Hibernate.isInitialized(child));
assertEquals(Child.class, unproxiedChild.getClass());
}));
}
@Test doInJPA( this::entityManagerFactory, ( entityManager -> {
public void testNotHibernateProxyShouldThrowException() { Parent parent = entityManager.find( Parent.class, p.getId() );
Parent p = new Parent(); Child child = parent.getChild();
Child c = new Child();
p.setChild(c);
doInJPA(this::entityManagerFactory, (entityManager -> {
entityManager.persist(p);
}));
doInJPA(this::entityManagerFactory, (entityManager -> {
Parent parent = entityManager.find(Parent.class, p.getId());
assertSame(parent, Hibernate.unproxy(parent));
}));
}
@Test assertFalse( Hibernate.isInitialized( child ) );
public void testNullUnproxyReturnsNull() { Hibernate.initialize( child );
assertNull(Hibernate.unproxy(null));
}
@Test Child unproxiedChild = (Child) Hibernate.unproxy( child );
public void testProxyEquality() { assertEquals( Child.class, unproxiedChild.getClass() );
Parent parent = doInJPA(this::entityManagerFactory, (entityManager -> { } ) );
Parent p = new Parent();
p.name = "John Doe";
entityManager.persist(p);
return p;
}));
doInJPA(this::entityManagerFactory, (entityManager -> {
Parent p = entityManager.getReference(Parent.class, parent.getId());
assertFalse(parent.equals(p));
assertTrue(parent.equals(Hibernate.unproxy(p)));
}));
}
@Entity(name = "Parent") doInJPA( this::entityManagerFactory, ( entityManager -> {
public static class Parent { Parent parent = entityManager.find( Parent.class, p.getId() );
@Id Child child = parent.getChild();
@GeneratedValue
private Integer id;
private String name; assertFalse( Hibernate.isInitialized( child ) );
Hibernate.initialize( child );
@OneToOne(cascade = CascadeType.ALL, fetch = FetchType.LAZY) Child unproxiedChild = Hibernate.unproxy( child, Child.class );
private Child child;
public Integer getId() { assertEquals( Child.class, unproxiedChild.getClass() );
return id; } ) );
} }
public void setChild(Child child) { @Test
this.child = child; public void testNotInitializedProxyCanBeUnproxiedWithInitialization() {
child.setParent(this); Parent p = new Parent();
} Child c = new Child();
p.setChild( c );
public Child getChild() { doInJPA( this::entityManagerFactory, ( entityManager -> {
return child; entityManager.persist( p );
} } ) );
@Override doInJPA( this::entityManagerFactory, ( entityManager -> {
public boolean equals(Object o) { Parent parent = entityManager.find( Parent.class, p.getId() );
if (this == o) return true; Child child = parent.getChild();
if (o == null || getClass() != o.getClass()) return false;
Parent parent = (Parent) o;
return Objects.equals(name, parent.name);
}
@Override assertFalse( Hibernate.isInitialized( child ) );
public int hashCode() {
return Objects.hash(name);
}
}
@Entity(name = "Child") Child unproxiedChild = (Child) Hibernate.unproxy( child );
public static class Child {
@Id
@GeneratedValue
private Integer id;
@OneToOne(fetch = FetchType.LAZY) assertTrue( Hibernate.isInitialized( child ) );
private Parent parent; assertEquals( Child.class, unproxiedChild.getClass() );
} ) );
public Integer getId() { doInJPA( this::entityManagerFactory, ( entityManager -> {
return id; Parent parent = entityManager.find( Parent.class, p.getId() );
} Child child = parent.getChild();
public void setParent(Parent parent) { assertFalse( Hibernate.isInitialized( child ) );
this.parent = parent;
}
public Parent getParent() { Child unproxiedChild = Hibernate.unproxy( child, Child.class );
return parent;
} assertTrue( Hibernate.isInitialized( child ) );
} assertEquals( Child.class, unproxiedChild.getClass() );
} ) );
}
@Test
public void testNotHibernateProxyShouldThrowException() {
Parent p = new Parent();
Child c = new Child();
p.setChild( c );
doInJPA( this::entityManagerFactory, ( entityManager -> {
entityManager.persist( p );
} ) );
doInJPA( this::entityManagerFactory, ( entityManager -> {
Parent parent = entityManager.find( Parent.class, p.getId() );
assertSame( parent, Hibernate.unproxy( parent ) );
} ) );
doInJPA( this::entityManagerFactory, ( entityManager -> {
Parent parent = entityManager.find( Parent.class, p.getId() );
assertSame( parent, Hibernate.unproxy( parent, Parent.class ) );
} ) );
}
@Test
public void testNullUnproxyReturnsNull() {
assertNull( Hibernate.unproxy( null ) );
assertNull( Hibernate.unproxy( null, Parent.class ) );
}
@Test
public void testProxyEquality() {
Parent parent = doInJPA( this::entityManagerFactory, ( entityManager -> {
Parent p = new Parent();
p.name = "John Doe";
entityManager.persist( p );
return p;
} ) );
doInJPA( this::entityManagerFactory, ( entityManager -> {
Parent p = entityManager.getReference( Parent.class, parent.getId() );
assertFalse( parent.equals( p ) );
assertTrue( parent.equals( Hibernate.unproxy( p ) ) );
} ) );
doInJPA( this::entityManagerFactory, ( entityManager -> {
Parent p = entityManager.getReference( Parent.class, parent.getId() );
assertFalse( parent.equals( p ) );
assertTrue( parent.equals( Hibernate.unproxy( p, Parent.class ) ) );
} ) );
}
@Entity(name = "Parent")
public static class Parent {
@Id
@GeneratedValue
private Integer id;
private String name;
@OneToOne(cascade = CascadeType.ALL, fetch = FetchType.LAZY)
private Child child;
public Integer getId() {
return id;
}
public void setChild(Child child) {
this.child = child;
child.setParent( this );
}
public Child getChild() {
return child;
}
@Override
public boolean equals(Object o) {
if ( this == o ) {
return true;
}
if ( o == null || getClass() != o.getClass() ) {
return false;
}
Parent parent = (Parent) o;
return Objects.equals( name, parent.name );
}
@Override
public int hashCode() {
return Objects.hash( name );
}
}
@Entity(name = "Child")
public static class Child {
@Id
@GeneratedValue
private Integer id;
@OneToOne(fetch = FetchType.LAZY)
private Parent parent;
public Integer getId() {
return id;
}
public void setParent(Parent parent) {
this.parent = parent;
}
public Parent getParent() {
return parent;
}
}
} }