Don't change result type of criteria query when changing select clause

Author: beikov

hibernate-core/src/main/java/org/hibernate/query/sqm/internal/ConcreteSqmSelectQueryPlan.java

@@ -176,7 +176,7 @@ protected static <T> RowTransformer<T> determineRowTransformer(
 			return makeRowTransformerTupleTransformerAdapter( sqm, queryOptions );
 		}
 
-		if ( resultType == null ) {
+		if ( resultType == null || resultType == Object.class ) {
 			return RowTransformerStandardImpl.instance();
 		}
 

hibernate-core/src/main/java/org/hibernate/query/sqm/tree/select/AbstractSqmSelectQuery.java

@@ -46,7 +46,7 @@ public abstract class AbstractSqmSelectQuery<T>
 		implements SqmSelectQuery<T> {
 	private final Map<String, SqmCteStatement<?>> cteStatements;
 	private SqmQueryPart<T> sqmQueryPart;
-	private Class<T> resultType;
+	private final Class<T> resultType;
 
 	public AbstractSqmSelectQuery(Class<T> resultType, NodeBuilder builder) {
 		this( new SqmQuerySpec<>( builder ), resultType, builder );
@@ -189,7 +189,7 @@ public Class<T> getResultType() {
 	}
 
 	protected void setResultType(Class<T> resultType) {
-		this.resultType = resultType;
+		// No-op
 	}
 
 	@Override
@@ -392,7 +392,6 @@ protected Selection<? extends T> getResultSelection(Selection<?>[] selections) {
 					break;
 				}
 				default: {
-					setResultType( (Class<T>) Object[].class );
 					resultSelection = ( Selection<? extends T> ) nodeBuilder().array( selections );
 				}
 			}

hibernate-core/src/main/java/org/hibernate/query/sqm/tree/select/SqmSelectStatement.java

@@ -108,6 +108,10 @@ public SqmSelectStatement<T> copy(SqmCopyContext context) {
 		if ( existing != null ) {
 			return existing;
 		}
+		return createCopy( context, getResultType() );
+	}
+
+	private <X> SqmSelectStatement<X> createCopy(SqmCopyContext context, Class<X> resultType) {
 		final Set<SqmParameter<?>> parameters;
 		if ( this.parameters == null ) {
 			parameters = null;
@@ -118,17 +122,19 @@ public SqmSelectStatement<T> copy(SqmCopyContext context) {
 				parameters.add( parameter.copy( context ) );
 			}
 		}
-		final SqmSelectStatement<T> statement = context.registerCopy(
+		//noinspection unchecked
+		final SqmSelectStatement<X> statement = (SqmSelectStatement<X>) context.registerCopy(
 				this,
 				new SqmSelectStatement<>(
 						nodeBuilder(),
 						copyCteStatements( context ),
-						getResultType(),
+						resultType,
 						getQuerySource(),
 						parameters
 				)
 		);
-		statement.setQueryPart( getQueryPart().copy( context ) );
+		//noinspection unchecked
+		statement.setQueryPart( (SqmQueryPart<X>) getQueryPart().copy( context ) );
 		return statement;
 	}
 
@@ -262,9 +268,6 @@ public SqmSelectStatement<T> select(Selection<? extends T> selection) {
 			checkSelectionIsJpaCompliant( selection );
 		}
 		getQuerySpec().setSelection( (JpaSelection<T>) selection );
-		if ( getResultType() == Object.class ) {
-			setResultType( (Class<T>) selection.getJavaType() );
-		}
 		return this;
 	}
 
@@ -306,7 +309,6 @@ public SqmSelectStatement<T> multiselect(List<Selection<?>> selectionList) {
 					break;
 				}
 				default: {
-					setResultType( (Class<T>) Object[].class );
 					resultSelection = ( Selection<? extends T> ) nodeBuilder().array( selections );
 				}
 			}

hibernate-core/src/main/java/org/hibernate/query/sqm/tree/select/SqmSubQuery.java

@@ -241,7 +241,6 @@ public SqmSubQuery<T> multiselect(List<Selection<?>> selectionList) {
 					break;
 				}
 				default: {
-					setResultType( (Class<T>) Object[].class );
 					resultSelection = ( Selection<? extends T> ) nodeBuilder().array( selections );
 				}
 			}
@@ -580,7 +579,6 @@ public SqmExpressible<T> getNodeType() {
 	public void applyInferableType(SqmExpressible<?> type) {
 		//noinspection unchecked
 		this.expressibleType = (SqmExpressible<T>) type;
-		setResultType( type == null ? null : expressibleType.getExpressibleJavaType().getJavaTypeClass() );
 	}
 
 	private void applyInferableType(Class<T> type) {