Support for simple natural id loader

Author: DavideD

hibernate-reactive-core/src/main/java/org/hibernate/reactive/loader/ast/internal/ReactiveCompoundNaturalIdLoader.java

@@ -0,0 +1,69 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive.loader.ast.internal;
+
+import java.util.concurrent.CompletionStage;
+import java.util.function.BiConsumer;
+import java.util.function.Consumer;
+
+import org.hibernate.engine.spi.SharedSessionContractImplementor;
+import org.hibernate.loader.ast.internal.CompoundNaturalIdLoader;
+import org.hibernate.loader.ast.internal.LoaderSqlAstCreationState;
+import org.hibernate.loader.ast.spi.NaturalIdLoadOptions;
+import org.hibernate.metamodel.mapping.EntityMappingType;
+import org.hibernate.metamodel.mapping.internal.CompoundNaturalIdMapping;
+import org.hibernate.reactive.loader.ast.spi.ReactiveNaturalIdLoader;
+import org.hibernate.sql.ast.tree.from.TableGroup;
+
+public class ReactiveCompoundNaturalIdLoader<T> extends CompoundNaturalIdLoader implements ReactiveNaturalIdLoader {
+
+	private final ReactiveNaturalIdLoaderDelegate<T> delegate;
+
+	public ReactiveCompoundNaturalIdLoader(
+			CompoundNaturalIdMapping naturalIdMapping,
+			EntityMappingType entityDescriptor) {
+		super( naturalIdMapping, entityDescriptor );
+		delegate = new ReactiveNaturalIdLoaderDelegate<>( naturalIdMapping, entityDescriptor ) {
+			@Override
+			protected void applyNaturalIdRestriction(
+					Object bindValue,
+					TableGroup rootTableGroup,
+					Consumer consumer,
+					BiConsumer jdbcParameterConsumer,
+					LoaderSqlAstCreationState sqlAstCreationState,
+					SharedSessionContractImplementor session) {
+				ReactiveCompoundNaturalIdLoader.this.applyNaturalIdRestriction(
+						bindValue,
+						rootTableGroup,
+						consumer,
+						jdbcParameterConsumer,
+						sqlAstCreationState,
+						session
+				);
+			}
+		};
+	}
+
+	@Override
+	public CompletionStage<Object> resolveNaturalIdToId(
+			Object naturalIdValue,
+			SharedSessionContractImplementor session) {
+		return delegate.resolveNaturalIdToId( naturalIdValue, session );
+	}
+
+	@Override
+	public CompletionStage<Object> resolveIdToNaturalId(Object id, SharedSessionContractImplementor session) {
+		return delegate.resolveIdToNaturalId( id, session );
+	}
+
+	@Override
+	public CompletionStage<T> load(
+			Object naturalIdValue,
+			NaturalIdLoadOptions options,
+			SharedSessionContractImplementor session) {
+		return delegate.load( naturalIdValue, options, session );
+	}
+}