Test

Author: DavideD

integration-tests/hibernate-reactive-panache/src/main/java/io/quarkus/it/panache/reactive/Counter.java

@@ -0,0 +1,37 @@
+package io.quarkus.it.panache.reactive;
+
+import io.quarkus.hibernate.reactive.panache.PanacheEntityBase;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+
+@Entity
+public class Counter extends PanacheEntityBase {
+
+    @Id
+    private Long id;
+    private int count = 0;
+
+    public Counter() {
+    }
+
+    public Counter(long id) {
+        this.id = id;
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public int getCount() {
+        return count;
+    }
+
+    @Override
+    public String toString() {
+        return String.valueOf(count);
+    }
+
+    public void increase() {
+        this.count++;
+    }
+}

integration-tests/hibernate-reactive-panache/src/main/java/io/quarkus/it/panache/reactive/WithTransactionCounterBean.java

@@ -0,0 +1,68 @@
+package io.quarkus.it.panache.reactive;
+
+import org.hibernate.reactive.mutiny.Mutiny;
+
+import org.junit.jupiter.api.Assertions;
+
+import io.quarkus.hibernate.reactive.panache.Panache;
+import io.quarkus.hibernate.reactive.panache.common.WithTransaction;
+import io.smallrye.mutiny.Uni;
+import jakarta.enterprise.context.ApplicationScoped;
+import jakarta.inject.Inject;
+import java.util.function.Function;
+import java.util.function.Supplier;
+
+import static jakarta.persistence.LockModeType.PESSIMISTIC_WRITE;
+
+/**
+ * The goal if this class is to test the usage of {@link WithTransaction},
+ * you should not use {@link io.quarkus.hibernate.reactive.panache.Panache#withTransaction(Supplier)}
+ * nor {@link org.hibernate.reactive.mutiny.Mutiny.SessionFactory#withTransaction(Function)}
+ */
+@ApplicationScoped
+@WithTransaction
+public class WithTransactionCounterBean {
+
+    private static final Long ID_COUNTER = 42L;
+
+    @Inject
+    Mutiny.SessionFactory sessionFactory;
+
+    public Uni<Counter> createOrResetCounter() {
+        final Counter counter = new Counter(ID_COUNTER);
+        return sessionFactory.withStatelessSession(session -> session
+                .upsert(counter)
+                .replaceWith(counter)
+        );
+    }
+
+    public Uni<Counter> increaseCounterWithHR() {
+        return sessionFactory.withSession(session -> session
+                .find(Counter.class, ID_COUNTER, PESSIMISTIC_WRITE)
+                .invoke(Counter::increase)
+        );
+    }
+
+    public Uni<Counter> increaseCounterWithPanache() {
+        return Counter
+                .<Counter> findById(ID_COUNTER, PESSIMISTIC_WRITE)
+                .invoke(Counter::increase);
+    }
+
+    public Uni<Void> assertThatSessionsAreEqual() {
+        return sessionFactory.withSession(hrSession -> Panache
+                .getSession().chain(panacheSession -> {
+                    if (panacheSession != hrSession) {
+                        return Uni.createFrom().failure(new AssertionError("Sessions are different!"));
+                    }
+                    return Uni.createFrom().voidItem();
+                })
+        );
+    }
+
+    public Uni<Counter> findCounter() {
+        return sessionFactory.withSession(session -> session
+                .find(Counter.class, ID_COUNTER)
+        );
+    }
+}

integration-tests/hibernate-reactive-panache/src/test/java/io/quarkus/it/panache/reactive/WithTransactionTest.java

@@ -0,0 +1,83 @@
+package io.quarkus.it.panache.reactive;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import io.quarkus.test.junit.QuarkusTest;
+import io.smallrye.mutiny.Uni;
+import jakarta.inject.Inject;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.Supplier;
+
+import static io.quarkus.vertx.VertxContextSupport.subscribeAndAwait;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assertions.fail;
+
+/**
+ * Hibernate Reactive and Panache store the created sessions in a local Vert.x context,
+ * and we need to make sure that they won't get lost.
+ * See issue <a href="https://github.com/quarkusio/quarkus/issues/47314">47314</a>
+ *
+ * @see io.quarkus.hibernate.reactive.panache.common.runtime.SessionOperations
+ * @see io.quarkus.hibernate.reactive.panache.common.WithTransaction
+ */
+@QuarkusTest
+public class WithTransactionTest {
+    // How many times we want to increase the counter
+    private static final int INCREASES_NUM = 10;
+
+    @Inject
+    WithTransactionCounterBean counterBean;
+
+    @BeforeEach
+    public void createOrResetCounter() throws Throwable {
+        subscribeAndAwait(counterBean::createOrResetCounter);
+    }
+
+    @Test
+    void increaseCounterWithHibernateReactive() throws Throwable {
+        subscribeAndAwait(counterBean::increaseCounterWithHR);
+        Counter counter = subscribeAndAwait(counterBean::findCounter);
+
+        assertThat(counter.getCount()).isEqualTo(1);
+    }
+
+    @Test
+    void increaseCounterWithPanache() throws Throwable {
+        subscribeAndAwait(counterBean::increaseCounterWithPanache);
+        Counter counter = subscribeAndAwait(counterBean::findCounter);
+
+        assertThat(counter.getCount()).isEqualTo(1);
+    }
+
+    @Test
+    void shouldReuseExistingSessions() throws Throwable {
+        subscribeAndAwait(counterBean::assertThatSessionsAreEqual);
+    }
+
+    @Test
+    void increaseCounter() throws Throwable {
+        List<Supplier<Uni<Counter>>> suppliers = new ArrayList<>();
+        for (int i = 0; i < INCREASES_NUM; i++) {
+            suppliers.add(() -> counterBean.increaseCounterWithHR());
+        }
+
+        final List<Counter> results = new ArrayList<>();
+        suppliers.stream()
+                .parallel()
+                .forEach(uni -> increaseCounter(uni, results));
+
+        final Counter dbCounter = subscribeAndAwait(() -> counterBean.findCounter());
+        assertThat(dbCounter.getCount()).isEqualTo(INCREASES_NUM);
+    }
+
+    private static void increaseCounter(Supplier<Uni<Counter>> func, List<Counter> counters) {
+        try {
+            final var counter = subscribeAndAwait(func);
+            counters.add(counter);
+        } catch (final Throwable e) {
+            fail(e);
+        }
+    }
+}