add new methods from H6.3 to control batch/subselect fetching

Author: gavinking

hibernate-reactive-core/src/main/java/org/hibernate/reactive/mutiny/Mutiny.java

@@ -1336,6 +1336,53 @@ default Session setCacheRetrieveMode(CacheRetrieveMode cacheRetrieveMode) {
 		 */
 		Filter getEnabledFilter(String filterName);
 
+		/**
+		 * Get the maximum batch size for batch fetching associations by
+		 * id in this session.
+		 *
+		 * @since 2.1
+		 */
+		int getFetchBatchSize();
+
+		/**
+		 * Set the maximum batch size for batch fetching associations by
+		 * id in this session.
+		 * Override the default controlled by the configuration property
+		 * {@code hibernate.default_batch_fetch_size}.
+		 * <p>
+		 * <ul>
+		 * <li>If {@code batchSize>1}, then batch fetching is enabled.
+		 * <li>If {@code batchSize<0}, the batch size is inherited from
+		 *     the factory-level setting.
+		 * <li>Otherwise, batch fetching is disabled.
+		 * </ul>
+		 *
+		 * @param batchSize the maximum batch size for batch fetching
+		 *
+		 * @since 2.1
+		 */
+		Session setFetchBatchSize(int batchSize);
+
+		/**
+		 * Determine if subselect fetching is enabled in this session.
+		 *
+		 * @return {@code true} if subselect fetching is enabled
+		 *
+		 * @since 2.1
+		 */
+		boolean isSubselectFetchingEnabled();
+
+		/**
+		 * Enable or disable subselect fetching in this session.
+		 * Override the default controlled by the configuration property
+		 * {@code hibernate.use_subselect_fetch}.
+		 *
+		 * @param enabled {@code true} to enable subselect fetching
+		 *
+		 * @since 2.1
+		 */
+		Session setSubselectFetchingEnabled(boolean enabled);
+
 		/**
 		 * Performs the given work within the scope of a database transaction,
 		 * automatically flushing the session. The transaction will be rolled

hibernate-reactive-core/src/main/java/org/hibernate/reactive/mutiny/impl/MutinySessionImpl.java

@@ -437,6 +437,28 @@ public Filter getEnabledFilter(String filterName) {
 		return delegate.getEnabledFilter( filterName );
 	}
 
+	@Override
+	public int getFetchBatchSize() {
+		return delegate.getFetchBatchSize();
+	}
+
+	@Override
+	public Mutiny.Session setFetchBatchSize(int batchSize) {
+		delegate.setFetchBatchSize( batchSize );
+		return this;
+	}
+
+	@Override
+	public boolean isSubselectFetchingEnabled() {
+		return delegate.isSubselectFetchingEnabled();
+	}
+
+	@Override
+	public Mutiny.Session setSubselectFetchingEnabled(boolean enabled) {
+		delegate.setSubselectFetchingEnabled( enabled );
+		return this;
+	}
+
 	@Override
 	public <T> Uni<T> withTransaction(Function<Mutiny.Transaction, Uni<T>> work) {
 		return currentTransaction == null

hibernate-reactive-core/src/main/java/org/hibernate/reactive/session/ReactiveSession.java

@@ -128,6 +128,11 @@ public interface ReactiveSession extends ReactiveQueryProducer, ReactiveSharedSe
 	void enableFetchProfile(String name) throws UnknownProfileException;
 	void disableFetchProfile(String name) throws UnknownProfileException;
 
+	int getFetchBatchSize();
+	void setFetchBatchSize(int batchSize);
+	boolean isSubselectFetchingEnabled();
+	void setSubselectFetchingEnabled(boolean enabled);
+
 	void clear();
 
 	boolean isDirty();

hibernate-reactive-core/src/main/java/org/hibernate/reactive/stage/Stage.java

@@ -5,7 +5,12 @@
  */
 package org.hibernate.reactive.stage;
 
-import jakarta.persistence.*;
+import jakarta.persistence.CacheRetrieveMode;
+import jakarta.persistence.CacheStoreMode;
+import jakarta.persistence.EntityGraph;
+import jakarta.persistence.FlushModeType;
+import jakarta.persistence.LockModeType;
+import jakarta.persistence.Parameter;
 import jakarta.persistence.criteria.CriteriaBuilder;
 import jakarta.persistence.criteria.CriteriaDelete;
 import jakarta.persistence.criteria.CriteriaQuery;
@@ -1333,6 +1338,53 @@ default Session setCacheRetrieveMode(CacheRetrieveMode cacheRetrieveMode) {
 		 */
 		Integer getBatchSize();
 
+		/**
+		 * Get the maximum batch size for batch fetching associations by
+		 * id in this session.
+		 *
+		 * @since 2.1
+		 */
+		int getFetchBatchSize();
+
+		/**
+		 * Set the maximum batch size for batch fetching associations by
+		 * id in this session.
+		 * Override the default controlled by the configuration property
+		 * {@code hibernate.default_batch_fetch_size}.
+		 * <p>
+		 * <ul>
+		 * <li>If {@code batchSize>1}, then batch fetching is enabled.
+		 * <li>If {@code batchSize<0}, the batch size is inherited from
+		 *     the factory-level setting.
+		 * <li>Otherwise, batch fetching is disabled.
+		 * </ul>
+		 *
+		 * @param batchSize the maximum batch size for batch fetching
+		 *
+		 * @since 2.1
+		 */
+		Stage.Session setFetchBatchSize(int batchSize);
+
+		/**
+		 * Determine if subselect fetching is enabled in this session.
+		 *
+		 * @return {@code true} if subselect fetching is enabled
+		 *
+		 * @since 2.1
+		 */
+		boolean isSubselectFetchingEnabled();
+
+		/**
+		 * Enable or disable subselect fetching in this session.
+		 * Override the default controlled by the configuration property
+		 * {@code hibernate.use_subselect_fetch}.
+		 *
+		 * @param enabled {@code true} to enable subselect fetching
+		 *
+		 * @since 2.1
+		 */
+		Stage.Session setSubselectFetchingEnabled(boolean enabled);
+
 		/**
 		 * Enable the named filter for this session.
 		 *

hibernate-reactive-core/src/main/java/org/hibernate/reactive/stage/impl/StageSessionImpl.java

@@ -362,6 +362,28 @@ public Filter getEnabledFilter(String filterName) {
 		return delegate.getEnabledFilter(filterName);
 	}
 
+	@Override
+	public int getFetchBatchSize() {
+		return delegate.getFetchBatchSize();
+	}
+
+	@Override
+	public Stage.Session setFetchBatchSize(int batchSize) {
+		delegate.setFetchBatchSize( batchSize );
+		return this;
+	}
+
+	@Override
+	public boolean isSubselectFetchingEnabled() {
+		return delegate.isSubselectFetchingEnabled();
+	}
+
+	@Override
+	public Stage.Session setSubselectFetchingEnabled(boolean enabled) {
+		delegate.setSubselectFetchingEnabled( enabled );
+		return this;
+	}
+
 	@Override
 	public <T> CompletionStage<T> withTransaction(Function<Stage.Transaction, CompletionStage<T>> work) {
 		return currentTransaction==null ? new Transaction<T>().execute(work) : work.apply(currentTransaction);