Leverage HHH-16227 to plug custom SqmMultiTableMutationStrategy and SqmMultiTableInsertStrategy implementations

Author: Sanne

hibernate-reactive-core/src/main/java/org/hibernate/reactive/dialect/ReactiveDialectWrapper.java

@@ -13,8 +13,6 @@
 import org.hibernate.query.sqm.mutation.spi.SqmMultiTableInsertStrategy;
 import org.hibernate.query.sqm.mutation.spi.SqmMultiTableMutationStrategy;
 import org.hibernate.reactive.dialect.identity.ReactiveIdentityColumnSupportAdapter;
-import org.hibernate.reactive.query.sqm.mutation.internal.cte.ReactiveCteInsertStrategy;
-import org.hibernate.reactive.query.sqm.mutation.internal.cte.ReactiveCteMutationStrategy;
 
 /**
  * Wraps the given dialect to make some internal components reactive;
@@ -36,14 +34,17 @@ public IdentityColumnSupport getIdentityColumnSupport() {
 	public SqmMultiTableMutationStrategy getFallbackSqmMutationStrategy(
 			EntityMappingType rootEntityDescriptor,
 			RuntimeModelCreationContext runtimeModelCreationContext) {
-		return new ReactiveCteMutationStrategy( rootEntityDescriptor, runtimeModelCreationContext );
+		throw new IllegalStateException(
+				"Should never be called - see ReactiveSqmMultiTableMutationStrategyProvider" );
 	}
 
 	@Override
 	public SqmMultiTableInsertStrategy getFallbackSqmInsertStrategy(
 			EntityMappingType rootEntityDescriptor,
 			RuntimeModelCreationContext runtimeModelCreationContext) {
-		return new ReactiveCteInsertStrategy( rootEntityDescriptor, runtimeModelCreationContext );
+		throw new IllegalStateException(
+				"Should never be called - see ReactiveSqmMultiTableMutationStrategyProvider" );
+
 	}
 
 }

hibernate-reactive-core/src/main/java/org/hibernate/reactive/provider/impl/ReactiveServiceInitiators.java

@@ -36,6 +36,7 @@
 import org.hibernate.reactive.provider.service.ReactivePersisterClassResolverInitiator;
 import org.hibernate.reactive.provider.service.ReactiveSchemaManagementToolInitiator;
 import org.hibernate.reactive.provider.service.ReactiveSessionFactoryBuilderInitiator;
+import org.hibernate.reactive.provider.service.ReactiveSqmMultiTableMutationStrategyProviderInitiator;
 import org.hibernate.reactive.provider.service.ReactiveValuesMappingProducerProviderInitiator;
 import org.hibernate.reactive.vertx.impl.VertxInstanceInitiator;
 import org.hibernate.resource.beans.spi.ManagedBeanRegistryInitiator;
@@ -129,6 +130,9 @@ private static List<StandardServiceInitiator<?>> buildInitialServiceInitiatorLis
 		//Custom for Hibernate Reactive:
 		serviceInitiators.add( ReactiveValuesMappingProducerProviderInitiator.INSTANCE );
 
+		//Custom for Hibernate Reactive:
+		serviceInitiators.add( ReactiveSqmMultiTableMutationStrategyProviderInitiator.INSTANCE );
+
 		serviceInitiators.trimToSize();
 
 		return unmodifiableList( serviceInitiators );

hibernate-reactive-core/src/main/java/org/hibernate/reactive/provider/service/ReactiveSqmMultiTableMutationStrategyProvider.java

@@ -0,0 +1,39 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive.provider.service;
+
+import org.hibernate.metamodel.mapping.EntityMappingType;
+import org.hibernate.metamodel.mapping.internal.MappingModelCreationProcess;
+import org.hibernate.metamodel.spi.RuntimeModelCreationContext;
+import org.hibernate.query.sqm.mutation.spi.SqmMultiTableInsertStrategy;
+import org.hibernate.query.sqm.mutation.spi.SqmMultiTableMutationStrategy;
+import org.hibernate.query.sqm.mutation.spi.SqmMultiTableMutationStrategyProvider;
+import org.hibernate.reactive.query.sqm.mutation.internal.cte.ReactiveCteInsertStrategy;
+import org.hibernate.reactive.query.sqm.mutation.internal.cte.ReactiveCteMutationStrategy;
+
+public class ReactiveSqmMultiTableMutationStrategyProvider implements SqmMultiTableMutationStrategyProvider {
+
+	@Override
+	public SqmMultiTableMutationStrategy createMutationStrategy(
+			EntityMappingType rootEntityDescriptor,
+			MappingModelCreationProcess creationProcess) {
+		final RuntimeModelCreationContext creationContext = creationProcess.getCreationContext();
+
+		//TODO there's more flavours in ORM ? And what do we do about explicitly configured instances & Dialect produced implementations?
+		return new ReactiveCteMutationStrategy( rootEntityDescriptor, creationContext );
+	}
+
+	@Override
+	public SqmMultiTableInsertStrategy createInsertStrategy(
+			EntityMappingType rootEntityDescriptor,
+			MappingModelCreationProcess creationProcess) {
+		final RuntimeModelCreationContext creationContext = creationProcess.getCreationContext();
+
+		//TODO there's more flavours in ORM ? And what do we do about explicitly configured instances & Dialect produced implementations?
+		return new ReactiveCteInsertStrategy( rootEntityDescriptor, creationContext );
+	}
+
+}

hibernate-reactive-core/src/main/java/org/hibernate/reactive/provider/service/ReactiveSqmMultiTableMutationStrategyProviderInitiator.java

@@ -0,0 +1,35 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive.provider.service;
+
+import java.util.Map;
+
+import org.hibernate.boot.registry.StandardServiceInitiator;
+import org.hibernate.query.sqm.mutation.spi.SqmMultiTableMutationStrategyProvider;
+import org.hibernate.service.spi.ServiceRegistryImplementor;
+
+/**
+ * Custom variations for SqmMultiTableMutationStrategyProvider
+ */
+public class ReactiveSqmMultiTableMutationStrategyProviderInitiator
+		implements StandardServiceInitiator<SqmMultiTableMutationStrategyProvider> {
+	/**
+	 * Singleton access
+	 */
+	public static final ReactiveSqmMultiTableMutationStrategyProviderInitiator INSTANCE = new ReactiveSqmMultiTableMutationStrategyProviderInitiator();
+
+	@Override
+	public SqmMultiTableMutationStrategyProvider initiateService(
+			Map<String, Object> configurationValues,
+			ServiceRegistryImplementor registry) {
+		return new ReactiveSqmMultiTableMutationStrategyProvider();
+	}
+
+	@Override
+	public Class<SqmMultiTableMutationStrategyProvider> getServiceInitiated() {
+		return SqmMultiTableMutationStrategyProvider.class;
+	}
+}