[#1099] Add new interfaces for reactive values generation

DavideD · DavideD · commit 81ec0e996cfc · 2022-01-14T15:40:04.000Z
diff --git a/hibernate-reactive-core/src/main/java/org/hibernate/reactive/tuple/MutinyValueGenerator.java b/hibernate-reactive-core/src/main/java/org/hibernate/reactive/tuple/MutinyValueGenerator.java
@@ -0,0 +1,31 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive.tuple;
+
+import org.hibernate.Session;
+import org.hibernate.reactive.mutiny.Mutiny;
+import org.hibernate.tuple.ValueGenerator;
+
+import io.smallrye.mutiny.Uni;
+
+/**
+ * A reactive {@link ValueGenerator} that gives access to a {@link Mutiny.Session}.
+ * <p>
+ * This interface needs to extend {@link ValueGenerator} but Hibernate Reactive
+ * ignores the method {@link MutinyValueGenerator#generateValue(Session, Object)}.
+ * </p>
+ *
+ * @see StageValueGenerator
+ */
+public interface MutinyValueGenerator<T> extends ValueGenerator<T> {
+
+	@Override
+	default T generateValue(Session session, Object owner) {
+		throw new UnsupportedOperationException( "Use generateValue(Mutiny.Session, Object) instead" );
+	}
+
+	Uni<T> generateValue(Mutiny.Session session, Object owner);
+}
diff --git a/hibernate-reactive-core/src/main/java/org/hibernate/reactive/tuple/StageValueGenerator.java b/hibernate-reactive-core/src/main/java/org/hibernate/reactive/tuple/StageValueGenerator.java
@@ -0,0 +1,32 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive.tuple;
+
+import java.util.concurrent.CompletionStage;
+
+import org.hibernate.Session;
+import org.hibernate.reactive.stage.Stage;
+import org.hibernate.tuple.ValueGenerator;
+
+/**
+ * A reactive {@link ValueGenerator} that gives access to a {@link Stage.Session}.
+ * <p>
+ * This interface needs to extend {@link ValueGenerator} but the non reactive method is ignored by
+ * Hibernate Reactive and, unless the default implementation is overridden, it will throw an exception
+ * when called.
+ * </p>
+ *
+ * @see MutinyValueGenerator
+ */
+public interface StageValueGenerator<T> extends ValueGenerator<T> {
+
+	@Override
+	default T generateValue(Session session, Object owner) {
+		throw new UnsupportedOperationException( "Use generateValue(Stage.Session, Object) instead" );
+	}
+
+	CompletionStage<T> generateValue(Stage.Session session, Object owner);
+}
