add StatelessSessionBuilder.connectionHandling() to match SessionBuilder

Author: gavinking

hibernate-core/src/main/java/org/hibernate/StatelessSessionBuilder.java

@@ -33,6 +33,18 @@ public interface StatelessSessionBuilder {
 	 */
 	StatelessSessionBuilder connection(Connection connection);
 
+	/**
+	 * Specifies the connection handling modes for the session.
+	 * <p>
+	 * Note that if {@link ConnectionAcquisitionMode#IMMEDIATELY} is specified,
+	 * then the release mode must be {@link ConnectionReleaseMode#ON_CLOSE}.
+	 *
+	 * @return {@code this}, for method chaining
+	 *
+	 * @since 7.2
+	 */
+	StatelessSessionBuilder connectionHandling(ConnectionAcquisitionMode acquisitionMode, ConnectionReleaseMode releaseMode);
+
 	/**
 	 * Define the tenant identifier to be associated with the opened session.
 	 *

hibernate-core/src/main/java/org/hibernate/internal/SessionFactoryImpl.java

@@ -1396,6 +1396,7 @@ public static class StatelessSessionBuilderImpl implements StatelessSessionBuild
 		private final SessionFactoryImpl sessionFactory;
 		private StatementInspector statementInspector;
 		private Connection connection;
+		private PhysicalConnectionHandlingMode connectionHandlingMode;
 		private Object tenantIdentifier;
 		private boolean readOnly;
 		private CacheMode cacheMode;
@@ -1406,6 +1407,7 @@ public StatelessSessionBuilderImpl(SessionFactoryImpl sessionFactory) {
 			statementInspector = options.getStatementInspector();
 			cacheMode = options.getInitialSessionCacheMode();
 			tenantIdentifier = sessionFactory.resolveTenantIdentifier();
+			connectionHandlingMode = options.getPhysicalConnectionHandlingMode();
 		}
 
 		@Override
@@ -1419,6 +1421,12 @@ public StatelessSessionBuilder connection(Connection connection) {
 			return this;
 		}
 
+		@Override
+		public StatelessSessionBuilder connectionHandling(ConnectionAcquisitionMode acquisitionMode, ConnectionReleaseMode releaseMode) {
+			this.connectionHandlingMode = PhysicalConnectionHandlingMode.interpret( acquisitionMode, releaseMode);
+			return this;
+		}
+
 		@Override @Deprecated
 		public StatelessSessionBuilder tenantIdentifier(String tenantIdentifier) {
 			this.tenantIdentifier = tenantIdentifier;
@@ -1509,7 +1517,7 @@ public StatementInspector getStatementInspector() {
 
 		@Override
 		public PhysicalConnectionHandlingMode getPhysicalConnectionHandlingMode() {
-			return sessionFactory.getSessionFactoryOptions().getPhysicalConnectionHandlingMode();
+			return connectionHandlingMode;
 		}
 
 		@Override