HHH-16120 Error advancing (next) ResultSet position

Author: dreab8

hibernate-core/src/main/java/org/hibernate/collection/spi/AbstractPersistentCollection.java

@@ -286,7 +286,11 @@ else if ( !session.isConnected() ) {
 				 Whenever the collection lazy loading is triggered during the loading process,
 				 closing the connection will cause an error when RowProcessingStateStandardImpl#next() will be called.
 				 */
-				if ( !session.isTransactionInProgress() && session.getPersistenceContext().isLoadFinished() ) {
+				final PersistenceContext persistenceContext = session.getPersistenceContext();
+				if ( !session.isTransactionInProgress()
+						&& ( !persistenceContext.hasLoadContext()
+							|| ( persistenceContext.hasLoadContext()
+								&& persistenceContext.getLoadContexts().isLoadingFinished() ) ) ) {
 					session.getJdbcCoordinator().afterTransaction();
 				}
 			}

hibernate-core/src/main/java/org/hibernate/engine/internal/StatefulPersistenceContext.java

@@ -201,6 +201,11 @@ public LoadContexts getLoadContexts() {
 		return loadContexts;
 	}
 
+	@Override
+	public boolean hasLoadContext() {
+		return loadContexts != null;
+	}
+
 //	@Override
 //	public void addUnownedCollection(CollectionKey key, PersistentCollection collection) {
 //		if ( unownedCollections == null ) {

hibernate-core/src/main/java/org/hibernate/engine/spi/PersistenceContext.java

@@ -58,6 +58,11 @@ public interface PersistenceContext {
 	 */
 	LoadContexts getLoadContexts();
 
+	default boolean hasLoadContext() {
+		getLoadContexts();
+		return true;
+	}
+
 //	/**
 //	 * Add a collection which has no owner loaded
 //	 *

hibernate-core/src/main/java/org/hibernate/sql/results/spi/LoadContexts.java

@@ -48,6 +48,10 @@ public void deregister(JdbcValuesSourceProcessingState state) {
 		}
 	}
 
+	public boolean isLoadingFinished() {
+		return jdbcValuesSourceProcessingStateStack.depth() == 0;
+	}
+
 	public LoadingEntityEntry findLoadingEntityEntry(EntityKey entityKey) {
 		return jdbcValuesSourceProcessingStateStack.findCurrentFirst(
 				state -> state.findLoadingEntityLocally( entityKey )