HHH-18610 ColumnReferenceKey for discriminator

Author: dreab8

hibernate-core/src/main/java/org/hibernate/query/results/ResultsHelper.java

@@ -4,6 +4,7 @@
  */
 package org.hibernate.query.results;
 
+import org.hibernate.metamodel.mapping.EntityDiscriminatorMapping;
 import org.hibernate.metamodel.mapping.EntityIdentifierMapping;
 import org.hibernate.metamodel.mapping.ModelPart;
 import org.hibernate.metamodel.mapping.SelectableMapping;
@@ -19,6 +20,7 @@
 import org.hibernate.sql.results.jdbc.spi.JdbcValuesMetadata;
 
 import static org.hibernate.sql.ast.spi.SqlExpressionResolver.createColumnReferenceKey;
+import static org.hibernate.sql.ast.spi.SqlExpressionResolver.createDiscriminatorColumnReferenceKey;
 
 /**
  * @author Steve Ebersole
@@ -65,6 +67,25 @@ public static Expression resolveSqlExpression(
 		);
 	}
 
+	public static Expression resolveSqlExpression(
+			DomainResultCreationStateImpl resolver,
+			JdbcValuesMetadata jdbcValuesMetadata,
+			TableReference tableReference,
+			EntityDiscriminatorMapping discriminatorMapping,
+			String columnAlias) {
+		return resolver.resolveSqlExpression(
+				createDiscriminatorColumnReferenceKey(
+						tableReference,
+						discriminatorMapping
+				),
+				processingState -> {
+					final int jdbcPosition = jdbcValuesMetadata.resolveColumnPosition( columnAlias );
+					final int valuesArrayPosition = jdbcPositionToValuesArrayPosition( jdbcPosition );
+					return new ResultSetMappingSqlSelection( valuesArrayPosition, discriminatorMapping.getJdbcMapping() );
+				}
+		);
+	}
+
 	public static Expression resolveSqlExpression(
 			DomainResultCreationStateImpl resolver,
 			TableReference tableReference,

hibernate-core/src/main/java/org/hibernate/query/results/dynamic/DynamicResultBuilderEntityStandard.java

@@ -14,6 +14,7 @@
 import org.hibernate.LockMode;
 import org.hibernate.engine.FetchTiming;
 import org.hibernate.metamodel.mapping.CollectionPart;
+import org.hibernate.metamodel.mapping.EntityDiscriminatorMapping;
 import org.hibernate.metamodel.mapping.EntityMappingType;
 import org.hibernate.metamodel.mapping.ModelPart;
 import org.hibernate.metamodel.mapping.PluralAttributeMapping;
@@ -262,7 +263,7 @@ else if ( idFetchBuilder != null ) {
 		}
 
 		if ( discriminatorColumnName != null ) {
-			resolveSqlSelection(
+			resolveDiscriminatorSqlSelection(
 					discriminatorColumnName,
 					tableReference,
 					entityMapping.getDiscriminatorMapping(),
@@ -287,6 +288,25 @@ else if ( idFetchBuilder != null ) {
 		}
 	}
 
+	private static void resolveDiscriminatorSqlSelection(String columnAlias, TableReference tableReference, EntityDiscriminatorMapping discriminatorMapping, JdbcValuesMetadata jdbcResultsMetadata, DomainResultCreationState domainResultCreationState) {
+		final DomainResultCreationStateImpl creationStateImpl = impl( domainResultCreationState );
+		creationStateImpl.resolveSqlSelection(
+				ResultsHelper.resolveSqlExpression(
+						creationStateImpl,
+						jdbcResultsMetadata,
+						tableReference,
+						discriminatorMapping,
+						columnAlias
+				),
+				discriminatorMapping.getJdbcMapping().getJdbcJavaType(),
+				null,
+				domainResultCreationState.getSqlAstCreationState()
+						.getCreationContext()
+						.getSessionFactory()
+						.getTypeConfiguration()
+		);
+	}
+
 	private FetchBuilder findIdFetchBuilder() {
 		return findFetchBuilder( entityMapping.getIdentifierMapping() );
 	}

hibernate-core/src/main/java/org/hibernate/sql/ast/spi/SqlExpressionResolver.java

@@ -6,10 +6,10 @@
 
 import java.util.function.Function;
 
+import org.hibernate.metamodel.mapping.EntityDiscriminatorMapping;
 import org.hibernate.metamodel.mapping.JdbcMapping;
 import org.hibernate.metamodel.mapping.SelectableMapping;
 import org.hibernate.metamodel.mapping.SelectablePath;
-import org.hibernate.metamodel.mapping.internal.DiscriminatorTypeImpl;
 import org.hibernate.sql.ast.tree.expression.ColumnReference;
 import org.hibernate.sql.ast.tree.expression.Expression;
 import org.hibernate.sql.ast.tree.expression.NestedColumnReference;
@@ -91,6 +91,15 @@ static ColumnReferenceKey createColumnReferenceKey(TableReference tableReference
 		return createColumnReferenceKey( tableReference, selectable.getSelectablePath(), selectable.getJdbcMapping() );
 	}
 
+	/**
+	 * Convenience form for creating a key from TableReference and EntityDiscriminatorMapping
+	 */
+	static ColumnReferenceKey createDiscriminatorColumnReferenceKey(TableReference tableReference, EntityDiscriminatorMapping discriminatorMapping) {
+		assert tableReference.containsAffectedTableName( discriminatorMapping.getContainingTableExpression() )
+				: String.format( ROOT, "Expecting tables to match between TableReference (%s) and SelectableMapping (%s)", tableReference.getTableId(), discriminatorMapping.getContainingTableExpression() );
+		return createColumnReferenceKey( tableReference, discriminatorMapping.getSelectablePath(), discriminatorMapping.getUnderlyingJdbcMapping() );
+	}
+
 	default Expression resolveSqlExpression(TableReference tableReference, SelectableMapping selectableMapping) {
 		return resolveSqlExpression(
 				createColumnReferenceKey( tableReference, selectableMapping ),
@@ -127,12 +136,7 @@ final class ColumnReferenceKey {
 		public ColumnReferenceKey(String tableQualifier, SelectablePath selectablePath, JdbcMapping jdbcMapping) {
 			this.tableQualifier = tableQualifier;
 			this.selectablePath = selectablePath;
-			if ( jdbcMapping instanceof DiscriminatorTypeImpl discriminatorType ) {
-				this.jdbcMapping = discriminatorType.getUnderlyingJdbcMapping();
-			}
-			else {
-				this.jdbcMapping = jdbcMapping;
-			}
+			this.jdbcMapping = jdbcMapping;
 		}
 
 		@Override