HHH-18313 NullPointerException because JdbcIsolationDelegate.sqlExceptionHelper() returns null

Author: dreab8

hibernate-core/src/main/java/org/hibernate/engine/jdbc/env/internal/JdbcEnvironmentInitiator.java

@@ -28,6 +28,7 @@
 import org.hibernate.engine.jdbc.internal.JdbcCoordinatorImpl;
 import org.hibernate.engine.jdbc.internal.JdbcServicesImpl;
 import org.hibernate.engine.jdbc.spi.JdbcServices;
+import org.hibernate.engine.jdbc.spi.SqlExceptionHelper;
 import org.hibernate.engine.spi.SessionFactoryImplementor;
 import org.hibernate.event.internal.EmptyEventManager;
 import org.hibernate.event.spi.EventManager;
@@ -278,6 +279,7 @@ private JdbcEnvironmentImpl getJdbcEnvironmentUsingJdbcMetadata(
 		final TemporaryJdbcSessionOwner temporaryJdbcSessionOwner = new TemporaryJdbcSessionOwner(
 				jdbcConnectionAccess,
 				jdbcServices,
+				new SqlExceptionHelper( false ),
 				registry
 		);
 		temporaryJdbcSessionOwner.transactionCoordinator = registry.requireService( TransactionCoordinatorBuilder.class )
@@ -586,16 +588,19 @@ private static class TemporaryJdbcSessionOwner implements JdbcSessionOwner, Jdbc
 		private final boolean connectionProviderDisablesAutoCommit;
 		private final PhysicalConnectionHandlingMode connectionHandlingMode;
 		private final JpaCompliance jpaCompliance;
+		private final SqlExceptionHelper sqlExceptionHelper;
 		private static final EmptyJdbcObserver EMPTY_JDBC_OBSERVER = EmptyJdbcObserver.INSTANCE;
 		TransactionCoordinator transactionCoordinator;
 		private final EmptyEventManager eventManager;
 
 		public TemporaryJdbcSessionOwner(
 				JdbcConnectionAccess jdbcConnectionAccess,
 				JdbcServices jdbcServices,
+				SqlExceptionHelper sqlExceptionHelper,
 				ServiceRegistryImplementor serviceRegistry) {
 			this.jdbcConnectionAccess = jdbcConnectionAccess;
 			this.jdbcServices = jdbcServices;
+			this.sqlExceptionHelper = sqlExceptionHelper;
 			this.serviceRegistry = serviceRegistry;
 			final ConfigurationService configuration = serviceRegistry.requireService( ConfigurationService.class );
 			this.jtaTrackByThread = configuration.getSetting( JTA_TRACK_BY_THREAD, BOOLEAN, true );
@@ -749,6 +754,11 @@ public boolean isActive() {
 			return true;
 		}
 
+		@Override
+		public SqlExceptionHelper getSqlExceptionHelper() {
+			return sqlExceptionHelper;
+		}
+
 		private static class EmptyJdbcObserver implements JdbcObserver{
 
 			public static final EmptyJdbcObserver INSTANCE = new EmptyJdbcObserver();

hibernate-core/src/main/java/org/hibernate/engine/jdbc/internal/JdbcCoordinatorImpl.java

@@ -92,8 +92,8 @@ public JdbcCoordinatorImpl(
 			this.logicalConnection = new LogicalConnectionManagedImpl(
 					owner.getJdbcConnectionAccess(),
 					owner.getJdbcSessionContext(),
-					resourceRegistry,
-					jdbcServices
+					owner.getSqlExceptionHelper(),
+					resourceRegistry
 			);
 		}
 		this.owner = owner;

hibernate-core/src/main/java/org/hibernate/engine/jdbc/internal/JdbcServicesImpl.java

@@ -84,26 +84,20 @@ public ParameterMarkerStrategy getParameterMarkerStrategy() {
 
 	@Override
 	public SqlExceptionHelper getSqlExceptionHelper() {
-		if ( jdbcEnvironment != null ) {
-			return jdbcEnvironment.getSqlExceptionHelper();
-		}
-		return null;
+		assert jdbcEnvironment != null : "JdbcEnvironment was not found";
+		return jdbcEnvironment.getSqlExceptionHelper();
 	}
 
 	@Override
 	public ExtractedDatabaseMetaData getExtractedMetaDataSupport() {
-		if ( jdbcEnvironment != null ) {
-			return jdbcEnvironment.getExtractedDatabaseMetaData();
-		}
-		return null;
+		assert jdbcEnvironment != null : "JdbcEnvironment was not found";
+		return jdbcEnvironment.getExtractedDatabaseMetaData();
 	}
 
 	@Override
 	public LobCreator getLobCreator(LobCreationContext lobCreationContext) {
-		if ( jdbcEnvironment != null ) {
-			return jdbcEnvironment.getLobCreatorBuilder().buildLobCreator( lobCreationContext );
-		}
-		return null;
+		assert jdbcEnvironment != null : "JdbcEnvironment was not found";
+		return jdbcEnvironment.getLobCreatorBuilder().buildLobCreator( lobCreationContext );
 	}
 
 }

hibernate-core/src/main/java/org/hibernate/resource/jdbc/internal/LogicalConnectionManagedImpl.java

@@ -54,8 +54,8 @@ public class LogicalConnectionManagedImpl extends AbstractLogicalConnectionImple
 	public LogicalConnectionManagedImpl(
 			JdbcConnectionAccess jdbcConnectionAccess,
 			JdbcSessionContext jdbcSessionContext,
-			ResourceRegistry resourceRegistry,
-			JdbcServices jdbcServices) {
+			SqlExceptionHelper sqlExceptionHelper,
+			ResourceRegistry resourceRegistry) {
 		this.jdbcConnectionAccess = jdbcConnectionAccess;
 		this.observer = jdbcSessionContext.getObserver();
 		this.resourceRegistry = resourceRegistry;
@@ -64,7 +64,7 @@ public LogicalConnectionManagedImpl(
 				jdbcSessionContext.getPhysicalConnectionHandlingMode(),
 				jdbcConnectionAccess );
 
-		this.sqlExceptionHelper = jdbcServices.getSqlExceptionHelper();
+		this.sqlExceptionHelper = sqlExceptionHelper;
 
 		if ( connectionHandlingMode.getAcquisitionMode() == IMMEDIATELY ) {
 			acquireConnectionIfNeeded();
@@ -82,6 +82,19 @@ public LogicalConnectionManagedImpl(
 		}
 	}
 
+	public LogicalConnectionManagedImpl(
+			JdbcConnectionAccess jdbcConnectionAccess,
+			JdbcSessionContext jdbcSessionContext,
+			ResourceRegistry resourceRegistry,
+			JdbcServices jdbcServices) {
+		this(
+				jdbcConnectionAccess,
+				jdbcSessionContext,
+				jdbcServices.getSqlExceptionHelper(),
+				resourceRegistry
+		);
+	}
+
 	private PhysicalConnectionHandlingMode determineConnectionHandlingMode(
 			PhysicalConnectionHandlingMode connectionHandlingMode,
 			JdbcConnectionAccess jdbcConnectionAccess) {

hibernate-core/src/main/java/org/hibernate/resource/jdbc/spi/JdbcSessionOwner.java

@@ -7,6 +7,7 @@
 package org.hibernate.resource.jdbc.spi;
 
 import org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess;
+import org.hibernate.engine.jdbc.spi.SqlExceptionHelper;
 import org.hibernate.event.spi.EventManager;
 import org.hibernate.resource.transaction.spi.TransactionCoordinator;
 
@@ -72,4 +73,8 @@ public interface JdbcSessionOwner {
 	Integer getJdbcBatchSize();
 
 	EventManager getEventManager();
+
+	default SqlExceptionHelper getSqlExceptionHelper() {
+		return getJdbcSessionContext().getJdbcServices().getSqlExceptionHelper();
+	}
 }

hibernate-core/src/main/java/org/hibernate/resource/transaction/backend/jdbc/internal/JdbcIsolationDelegate.java

@@ -37,7 +37,7 @@ public JdbcIsolationDelegate(TransactionCoordinatorOwner transactionCoordinatorO
 	public JdbcIsolationDelegate(JdbcSessionOwner jdbcSessionOwner) {
 		this(
 				jdbcSessionOwner.getJdbcConnectionAccess(),
-				jdbcSessionOwner.getJdbcSessionContext().getJdbcServices().getSqlExceptionHelper()
+				jdbcSessionOwner.getSqlExceptionHelper()
 		);
 	}
 

hibernate-core/src/main/java/org/hibernate/resource/transaction/backend/jta/internal/JtaIsolationDelegate.java

@@ -50,7 +50,7 @@ public JtaIsolationDelegate(TransactionCoordinatorOwner transactionCoordinatorOw
 	public JtaIsolationDelegate(JdbcSessionOwner jdbcSessionOwner, TransactionManager transactionManager) {
 		this(
 				jdbcSessionOwner.getJdbcConnectionAccess(),
-				jdbcSessionOwner.getJdbcSessionContext().getJdbcServices().getSqlExceptionHelper(),
+				jdbcSessionOwner.getSqlExceptionHelper(),
 				transactionManager
 		);
 	}

hibernate-core/src/test/java/org/hibernate/orm/test/connection/ConnectionCreatorTest.java

@@ -9,6 +9,7 @@
 import java.sql.Connection;
 import java.sql.Driver;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Properties;
@@ -17,7 +18,10 @@
 import org.hibernate.boot.registry.StandardServiceInitiator;
 import org.hibernate.boot.registry.internal.BootstrapServiceRegistryImpl;
 import org.hibernate.boot.registry.internal.StandardServiceRegistryImpl;
+import org.hibernate.dialect.H2Dialect;
 import org.hibernate.engine.jdbc.connections.internal.DriverConnectionCreator;
+import org.hibernate.engine.jdbc.env.internal.JdbcEnvironmentImpl;
+import org.hibernate.engine.jdbc.env.spi.JdbcEnvironment;
 import org.hibernate.engine.jdbc.internal.JdbcServicesImpl;
 import org.hibernate.engine.jdbc.spi.JdbcServices;
 import org.hibernate.exception.JDBCConnectionException;
@@ -75,7 +79,12 @@ private CCTStandardServiceRegistryImpl(
 		public <R extends Service> R getService(Class<R> serviceRole) {
 			if ( JdbcServices.class.equals( serviceRole ) ) {
 				// return a new, not fully initialized JdbcServicesImpl
-				return (R) new JdbcServicesImpl();
+				JdbcServicesImpl jdbcServices = new JdbcServicesImpl(this);
+				jdbcServices.configure( new HashMap<>() );
+				return (R) jdbcServices;
+			}
+			if( JdbcEnvironment.class.equals( serviceRole ) ){
+				return (R) new JdbcEnvironmentImpl( this, new H2Dialect() );
 			}
 			return super.getService( serviceRole );
 		}

hibernate-core/src/test/java/org/hibernate/orm/test/jdbc/JdbcCoordinatorTest.java

@@ -79,7 +79,7 @@ public void testConnectionClose()
 		SqlExceptionHelper sqlExceptionHelper = Mockito.mock( SqlExceptionHelper.class );
 		when( jdbcServices.getSqlExceptionHelper() ).thenReturn(
 				sqlExceptionHelper );
-
+		when(sessionOwner.getSqlExceptionHelper()).thenReturn( sqlExceptionHelper );
 		JdbcCoordinatorImpl jdbcCoordinator = new JdbcCoordinatorImpl(
 				null,
 				sessionOwner,