use safe casts

Author: gavinking

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

@@ -18,7 +18,6 @@
 import org.hibernate.LockMode;
 import org.hibernate.bytecode.enhance.spi.interceptor.EnhancementAsProxyLazinessInterceptor;
 import org.hibernate.collection.spi.AbstractPersistentCollection;
-import org.hibernate.collection.spi.PersistentCollection;
 import org.hibernate.engine.spi.PersistentAttributeInterceptable;
 import org.hibernate.engine.spi.PersistentAttributeInterceptor;
 import org.hibernate.engine.spi.SharedSessionContractImplementor;
@@ -2452,18 +2451,18 @@ static <T> Uni<T> fetch(T association) {
 		}
 
 		final SharedSessionContractImplementor session;
-		if ( association instanceof HibernateProxy ) {
-			session = ( (HibernateProxy) association ).getHibernateLazyInitializer().getSession();
+		if ( association instanceof HibernateProxy proxy ) {
+			session = proxy.getHibernateLazyInitializer().getSession();
 		}
-		else if ( association instanceof PersistentCollection ) {
+		else if ( association instanceof AbstractPersistentCollection<?> collection ) {
 			//this unfortunately doesn't work for stateless session because the session ref gets set to null
-			session = ( (AbstractPersistentCollection<?>) association ).getSession();
+			session = collection.getSession();
 		}
 		else if ( isPersistentAttributeInterceptable( association ) ) {
 			final PersistentAttributeInterceptable interceptable = asPersistentAttributeInterceptable( association );
 			final PersistentAttributeInterceptor interceptor = interceptable.$$_hibernate_getInterceptor();
-			if ( interceptor instanceof EnhancementAsProxyLazinessInterceptor ) {
-				session = ( (EnhancementAsProxyLazinessInterceptor) interceptor ).getLinkedSession();
+			if ( interceptor instanceof EnhancementAsProxyLazinessInterceptor lazinessInterceptor ) {
+				session = lazinessInterceptor.getLinkedSession();
 			}
 			else {
 				return Uni.createFrom().item( association );

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

@@ -19,7 +19,6 @@
 import org.hibernate.LockMode;
 import org.hibernate.bytecode.enhance.spi.interceptor.EnhancementAsProxyLazinessInterceptor;
 import org.hibernate.collection.spi.AbstractPersistentCollection;
-import org.hibernate.collection.spi.PersistentCollection;
 import org.hibernate.engine.spi.PersistentAttributeInterceptable;
 import org.hibernate.engine.spi.PersistentAttributeInterceptor;
 import org.hibernate.engine.spi.SharedSessionContractImplementor;
@@ -2499,17 +2498,17 @@ static <T> CompletionStage<T> fetch(T association) {
 		}
 
 		final SharedSessionContractImplementor session;
-		if ( association instanceof HibernateProxy) {
-			session = ( (HibernateProxy) association ).getHibernateLazyInitializer().getSession();
+		if ( association instanceof HibernateProxy proxy ) {
+			session = proxy.getHibernateLazyInitializer().getSession();
 		}
-		else if ( association instanceof PersistentCollection) {
-			session = ( (AbstractPersistentCollection<?>) association ).getSession();
+		else if ( association instanceof AbstractPersistentCollection<?> collection ) {
+			session = collection.getSession();
 		}
 		else if ( isPersistentAttributeInterceptable( association ) ) {
 			final PersistentAttributeInterceptable interceptable = asPersistentAttributeInterceptable( association );
 			final PersistentAttributeInterceptor interceptor = interceptable.$$_hibernate_getInterceptor();
-			if ( interceptor instanceof EnhancementAsProxyLazinessInterceptor) {
-				session = ( (EnhancementAsProxyLazinessInterceptor) interceptor ).getLinkedSession();
+			if ( interceptor instanceof EnhancementAsProxyLazinessInterceptor lazinessInterceptor) {
+				session = lazinessInterceptor.getLinkedSession();
 			}
 			else {
 				return CompletionStages.completedFuture( association );