HHH-11437 - Entity joins are not polymorphic

(cherry picked from commit 3d6f8eb)
hibernate · May 10, 2017 · fc514da · fc514da
1 parent 698aea1
commit fc514da
Show file tree

Hide file tree

Showing 3 changed files with 74 additions and 14 deletions.
diff --git a/hibernate-core/src/main/java/org/hibernate/engine/internal/JoinSequence.java b/hibernate-core/src/main/java/org/hibernate/engine/internal/JoinSequence.java
@@ -87,6 +87,10 @@ public void applyTreatAsDeclarations(Set<String> treatAsDeclarations) {
 		this.treatAsDeclarations.addAll( treatAsDeclarations );
 	}
 
+	protected Set<String> getTreatAsDeclarations() {
+		return treatAsDeclarations;
+	}
+
 
 	/**
 	 * Create a full, although shallow, copy.
@@ -269,7 +273,7 @@ private void addSubclassJoins(
 		);
 	}
 
-	private boolean isIncluded(String alias) {
+	protected boolean isIncluded(String alias) {
 		return selector != null && selector.includeSubclasses( alias );
 	}
 

diff --git a/hibernate-core/src/main/java/org/hibernate/hql/internal/ast/tree/EntityJoinFromElement.java b/hibernate-core/src/main/java/org/hibernate/hql/internal/ast/tree/EntityJoinFromElement.java
@@ -8,6 +8,7 @@
 
 import java.util.Collections;
 import java.util.Map;
+import java.util.Set;
 
 import org.hibernate.AssertionFailure;
 import org.hibernate.MappingException;
@@ -16,6 +17,7 @@
 import org.hibernate.hql.internal.antlr.HqlSqlTokenTypes;
 import org.hibernate.hql.internal.ast.HqlSqlWalker;
 import org.hibernate.internal.util.StringHelper;
+import org.hibernate.persister.entity.AbstractEntityPersister;
 import org.hibernate.persister.entity.EntityPersister;
 import org.hibernate.persister.entity.Joinable;
 import org.hibernate.persister.entity.Queryable;
@@ -127,11 +129,34 @@ public JoinFragment toJoinFragment(
 			}
 
 			final StringBuilder buffer = new StringBuilder();
-			buffer.append( joinString )
-					.append( entityTableText )
+			final AbstractEntityPersister joinable = (AbstractEntityPersister) entityType.getAssociatedJoinable(factory);
+
+			buffer.append( joinString );
+
+			Set<String> treatAsDeclarations = getTreatAsDeclarations();
+			final boolean include = includeAllSubclassJoins && isIncluded( entityTableAlias );
+			String fromFragment = joinable.fromJoinFragment( entityTableAlias, true, include, treatAsDeclarations );
+			String whereFragment = joinable.whereJoinFragment( entityTableAlias, true, include, treatAsDeclarations );
+
+			// We need a table group only when having e.g. a left join of a polymorphic entity
+			// fromFragment is empty if the entity is non-polymorphic
+			// Inner joined entity joins can avoid using the table grouping since the condition can be moved to the where clause
+			boolean renderTableGroup = !fromFragment.isEmpty() && joinType != JoinType.INNER_JOIN;
+
+			if ( renderTableGroup ) {
+				buffer.append( '(' );
+			}
+
+			buffer.append( entityTableText )
 					.append( ' ' )
-					.append( entityTableAlias )
-					.append( " on " );
+					.append( entityTableAlias );
+
+			if ( renderTableGroup ) {
+				buffer.append( fromFragment )
+						.append( ')' );
+			}
+
+			buffer.append( " on " );
 
 			final String filters = 	entityType.getOnCondition(
 					entityTableAlias,
@@ -140,23 +165,54 @@ public JoinFragment toJoinFragment(
 					Collections.<String>emptySet()
 			);
 
-			buffer.append( filters );
-			if ( withClauseFragment != null ) {
-				if ( StringHelper.isNotEmpty( filters ) ) {
-					buffer.append( " and " );
+			if ( fromFragment.isEmpty() || renderTableGroup ) {
+				buffer.append( filters );
+				if ( withClauseFragment != null ) {
+					if ( StringHelper.isNotEmpty( filters ) ) {
+						buffer.append( " and " );
+					}
+					buffer.append( withClauseFragment );
 				}
-				buffer.append( withClauseFragment );
 			}
+			else {
+				// We know there is a fromFragment and that we shouldn't render a table group
+				// This means the entity is polymorphic and the entity join is an inner join
+				// We move the with clause stuff to the where clause but still need to have a valid on condition
+				buffer.append( "1=1" );
+				buffer.append( fromFragment );
+
+				// Proper capacity to avoid resizing
+				StringBuilder whereBuffer = new StringBuilder(
+						10
+							+ whereFragment.length()
+							+ filters.length()
+							+ withClauseFragment.length()
+				);
+				whereBuffer.append(whereFragment);
+				if ( !filters.isEmpty() ) {
+					whereBuffer.append( " and " );
+					whereBuffer.append( filters );
+				}
+				if ( !withClauseFragment.isEmpty() ) {
+					whereBuffer.append( " and " );
+					whereBuffer.append( withClauseFragment );
+				}
 
-			return new EntityJoinJoinFragment( buffer.toString() );
+				whereFragment = whereBuffer.toString();
+			}
+
+			return new EntityJoinJoinFragment( buffer.toString(), whereFragment );
 		}
+
 	}
 
 	private static class EntityJoinJoinFragment extends JoinFragment {
 		private final String fragmentString;
+		private final String whereFragment;
 
-		public EntityJoinJoinFragment(String fragmentString) {
+		public EntityJoinJoinFragment(String fragmentString, String whereFragment) {
 			this.fragmentString = fragmentString;
+			this.whereFragment = whereFragment;
 		}
 
 		@Override
@@ -193,7 +249,7 @@ public String toFromFragmentString() {
 
 		@Override
 		public String toWhereFragmentString() {
-			return null;
+			return whereFragment;
 		}
 
 		@Override

diff --git a/hibernate-core/src/test/java/org/hibernate/test/hql/JoinOnClauseTest.java b/hibernate-core/src/test/java/org/hibernate/test/hql/JoinOnClauseTest.java
@@ -64,7 +64,7 @@ public void setUp(){
 
 	@Test
 	@TestForIssue(jiraKey = "HHH-11437")
-	public void testOnClauseUsesNonDrivingTableAlias() {
+	public void testOnClauseUsesSuperclassAttribute() {
 		Session s = openSession();
 		s.getTransaction().begin();
 		{