[#2518] Add createSession and createStatelessSession

In Mutiny and Stage classes

Author: DavideD

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

@@ -5,6 +5,11 @@
  */
 package org.hibernate.reactive.mutiny;
 
+import java.lang.invoke.MethodHandles;
+import java.util.List;
+import java.util.function.BiFunction;
+import java.util.function.Function;
+
 import org.hibernate.Cache;
 import org.hibernate.CacheMode;
 import org.hibernate.Filter;
@@ -44,10 +49,6 @@
 import jakarta.persistence.criteria.CriteriaUpdate;
 import jakarta.persistence.metamodel.Attribute;
 import jakarta.persistence.metamodel.Metamodel;
-import java.lang.invoke.MethodHandles;
-import java.util.List;
-import java.util.function.BiFunction;
-import java.util.function.Function;
 
 import static org.hibernate.engine.internal.ManagedTypeHelper.asPersistentAttributeInterceptable;
 import static org.hibernate.engine.internal.ManagedTypeHelper.isPersistentAttributeInterceptable;
@@ -1971,6 +1972,59 @@ interface Transaction {
 	 */
 	interface SessionFactory extends AutoCloseable {
 
+		/**
+		 * Obtain a new {@link Session reactive session}, the main
+		 * interaction point between the user's program and Hibernate
+		 * Reactive.
+		 * <p>
+		 * The underlying database connection is obtained lazily
+		 * when the returned {@link Session} needs to access the
+		 * database.
+		 * <p>
+		 * The client must close the session using {@link Session#close()}.
+		 */
+		@Incubating
+		Session createSession();
+
+		/**
+		 * Obtain a new {@link Session reactive session}.
+		 * <p>
+		 * The underlying database connection is obtained lazily
+		 * when the returned {@link Session} needs to access the
+		 * database.
+		 * <p>
+		 * The client must close the session using {@link Session#close()}.
+		 * @param tenantId the id of the tenant
+		 */
+		@Incubating
+		Session createSession(String tenantId);
+
+		/**
+		 * Obtain a new {@link StatelessSession reactive stateless session}.
+		 * <p>
+		 * The underlying database connection is obtained lazily
+		 * when the returned {@link StatelessSession} needs to access the
+		 * database.
+		 * <p>
+		 * The client must close the session using {@link Session#close()}.
+		 */
+		@Incubating
+		StatelessSession createStatelessSession();
+
+		/**
+		 * Obtain a new {@link StatelessSession reactive stateless session}.
+		 * <p>
+		 * The underlying database connection is obtained lazily
+		 * when the returned {@link StatelessSession} needs to access the
+		 * database.
+		 * <p>
+		 * The client must close the session using {@link Session#close()}.
+		 *
+		 * @param tenantId the id of the tenant
+		 */
+		@Incubating
+		StatelessSession createStatelessSession(String tenantId);
+
 		/**
 		 * Obtain a new {@link Session reactive session} {@link Uni}, the main
 		 * interaction point between the user's program and Hibernate

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

@@ -28,6 +28,8 @@
 import org.hibernate.reactive.mutiny.Mutiny;
 import org.hibernate.reactive.pool.ReactiveConnection;
 import org.hibernate.reactive.pool.ReactiveConnectionPool;
+import org.hibernate.reactive.session.ReactiveSession;
+import org.hibernate.reactive.session.ReactiveStatelessSession;
 import org.hibernate.reactive.session.impl.ReactiveSessionImpl;
 import org.hibernate.reactive.session.impl.ReactiveStatelessSessionImpl;
 import org.hibernate.service.ServiceRegistry;
@@ -88,6 +90,32 @@ public Context getContext() {
 		return context;
 	}
 
+	@Override
+	public Mutiny.Session createSession() {
+		return createSession( getTenantIdentifier( options() ) );
+	}
+
+	@Override
+	public Mutiny.Session createSession(String tenantId) {
+		final SessionCreationOptions options = options();
+		ReactiveConnectionPool pool = delegate.getServiceRegistry().getService( ReactiveConnectionPool.class );
+		ReactiveSession sessionImpl = new ReactiveSessionImpl( delegate, options, pool.getProxyConnection( tenantId ) );
+		return new MutinySessionImpl( sessionImpl, this );
+	}
+
+	@Override
+	public Mutiny.StatelessSession createStatelessSession() {
+		return createStatelessSession( getTenantIdentifier( options() ) );
+	}
+
+	@Override
+	public Mutiny.StatelessSession createStatelessSession(String tenantId) {
+		final SessionCreationOptions options = options();
+		ReactiveConnectionPool pool = delegate.getServiceRegistry().getService( ReactiveConnectionPool.class );
+		ReactiveStatelessSession sessionImpl = new ReactiveStatelessSessionImpl( delegate, options, pool.getProxyConnection( tenantId ) );
+		return new MutinyStatelessSessionImpl( sessionImpl, this );
+	}
+
 	@Override
 	public Uni<Mutiny.Session> openSession() {
 		SessionCreationOptions options = options();

hibernate-reactive-core/src/main/java/org/hibernate/reactive/pool/ReactiveConnectionPool.java

@@ -37,12 +37,25 @@
  */
 @Incubating
 public interface ReactiveConnectionPool extends Service {
+	/**
+	 * Obtain a lazily-initializing reactive connection. The
+	 * actual connection might be made when the returned
+	 * instance if {@link ReactiveConnection} is first used.
+	 */
+	ReactiveConnection getProxyConnection();
 
 	/**
 	 * Obtain a reactive connection, returning the connection
-	 * via a {@link CompletionStage}.
+	 * via a {@link CompletionStage} and overriding the default
+	 * {@link SqlExceptionHelper} for the pool.
 	 */
-	CompletionStage<ReactiveConnection> getConnection();
+	ReactiveConnection getProxyConnection(SqlExceptionHelper sqlExceptionHelper);
+
+	/**
+	 * Obtain a reactive connection for the given tenant id,
+	 * returning the connection via a {@link CompletionStage}.
+	 */
+	ReactiveConnection getProxyConnection(String tenantId);
 
 	/**
 	 * Obtain a reactive connection, returning the connection
@@ -57,6 +70,12 @@ public interface ReactiveConnectionPool extends Service {
 	 */
 	CompletionStage<ReactiveConnection> getConnection(String tenantId);
 
+	/**
+	 * Obtain a reactive connection, returning the connection
+	 * via a {@link CompletionStage}.
+	 */
+	CompletionStage<ReactiveConnection> getConnection();
+
 	/**
 	 * Obtain a reactive connection for the given tenant id,
 	 * returning the connection via a {@link CompletionStage}