[hibernate#929] Adapting H2SqlClientPool

blafond · Mar 29, 2022 · 0702ecd · 0702ecd
1 parent c2eb3f7
commit 0702ecd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/...-core/src/main/java/org/hibernate/reactive/pool/impl/ReactiveConnectionPoolInitiator.java b/...-core/src/main/java/org/hibernate/reactive/pool/impl/ReactiveConnectionPoolInitiator.java
@@ -38,7 +38,13 @@ public ReactiveConnectionPoolInitiator() {}
 	@Override
 	public ReactiveConnectionPool initiateService(Map configurationValues, ServiceRegistryImplementor registry) {
 		Object configValue = configurationValues.get( Settings.SQL_CLIENT_POOL );
-		if (configValue==null) {
+
+		String dbType = (String) configurationValues.get( "db" );
+		if ( dbType.equalsIgnoreCase( "h2" ) ) {
+			configValue = "org.hibernate.reactive.pool.impl.H2SqlClientPool";
+		}
+
+		if ( configValue == null ) {
 			return new DefaultSqlClientPool();
 		}
 

diff --git a/...ore/src/main/java/org/hibernate/reactive/provider/service/NoJdbcEnvironmentInitiator.java b/...ore/src/main/java/org/hibernate/reactive/provider/service/NoJdbcEnvironmentInitiator.java
@@ -9,6 +9,7 @@
 import org.hibernate.dialect.CockroachDB201Dialect;
 import org.hibernate.dialect.DB297Dialect;
 import org.hibernate.dialect.Dialect;
+import org.hibernate.dialect.H2Dialect;
 import org.hibernate.dialect.MariaDB103Dialect;
 import org.hibernate.dialect.MySQL8Dialect;
 import org.hibernate.dialect.Oracle12cDialect;
@@ -35,7 +36,7 @@
 /**
  * A Hibernate {@link StandardServiceInitiator service initiator} that
  * provides an implementation of {@link JdbcEnvironment} that infers
- * the Hibernate {@link org.hibernate.dialect.Dialect} from the JDBC URL.
+ * the Hibernate {@link Dialect} from the JDBC URL.
  */
 public class NoJdbcEnvironmentInitiator extends JdbcEnvironmentInitiator {
 	private static final Log LOG = LoggerFactory.make( Log.class, MethodHandles.lookup() );
@@ -147,6 +148,9 @@ else if ( url.startsWith( "sqlserver:" ) ) {
 		else if ( url.startsWith( "oracle:" ) ) {
 			return Oracle12cDialect.class;
 		}
+		else if ( url.startsWith( "h2:" ) ) {
+			return H2Dialect.class;
+		}
 		else {
 			return null;
 		}