forked from hibernate/hibernate-reactive
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
c96add1
commit de1ced3
Showing
3 changed files
with
163 additions
and
19 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
138 changes: 138 additions & 0 deletions
138
hibernate-reactive-core/src/test/java/org/hibernate/reactive/CancelSignalTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,138 @@ | ||
| /* Hibernate, Relational Persistence for Idiomatic Java | ||
| * | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| * Copyright: Red Hat Inc. and Hibernate Authors | ||
| */ | ||
| package org.hibernate.reactive; | ||
|
|
||
| import io.micrometer.core.instrument.Metrics; | ||
| import io.smallrye.mutiny.subscription.Cancellable; | ||
| import io.vertx.ext.unit.TestContext; | ||
| import org.assertj.core.api.Assertions; | ||
| import org.hibernate.reactive.mutiny.Mutiny; | ||
| import org.junit.Test; | ||
|
|
||
| import javax.persistence.Entity; | ||
| import javax.persistence.Id; | ||
| import javax.persistence.Table; | ||
| import java.util.ArrayList; | ||
| import java.util.Collection; | ||
| import java.util.List; | ||
| import java.util.Objects; | ||
| import java.util.concurrent.CountDownLatch; | ||
| import java.util.concurrent.ExecutorService; | ||
| import java.util.concurrent.Executors; | ||
| import java.util.concurrent.TimeUnit; | ||
|
|
||
| public class CancelSignalTest extends BaseReactiveTest { | ||
| @Override | ||
| protected Collection<Class<?>> annotatedEntities() { | ||
| return List.of(GuineaPig.class); | ||
| } | ||
|
|
||
| @Test | ||
| public void cleanupConnectionWhenCancelSignal(TestContext context) { | ||
| int executeSize = 10; | ||
| ExecutorService withSessionExecutor = Executors.newFixedThreadPool(executeSize); | ||
| ExecutorService cancelExecutor = Executors.newFixedThreadPool(executeSize); | ||
| Mutiny.SessionFactory mutinySessionFactory = getMutinySessionFactory(); | ||
|
|
||
| List<Cancellable> stack = new ArrayList<>(); | ||
| CountDownLatch poolCount = new CountDownLatch(3); | ||
|
|
||
| for (int i = 0; i < executeSize; i++) { | ||
| withSessionExecutor.execute(() -> { | ||
| CountDownLatch countDownLatch = new CountDownLatch(1); | ||
| Cancellable with = mutinySessionFactory.withSession(s -> { | ||
| poolCount.countDown(); | ||
| try { | ||
| Thread.sleep(200); | ||
| } catch (InterruptedException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
|
|
||
| return s.find(GuineaPig.class, 1); | ||
| }) | ||
| .onTermination().invoke(countDownLatch::countDown) | ||
| .subscribe() | ||
| .with(item -> System.out.println("end : " + item)); | ||
| stack.add(0, with); | ||
| try { | ||
| countDownLatch.await(); | ||
| } catch (InterruptedException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| try { | ||
| poolCount.await(); | ||
| } catch (InterruptedException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
|
|
||
| stack.forEach(cancellable -> cancelExecutor.execute(cancellable::cancel)); | ||
| try { | ||
| cancelExecutor.awaitTermination(10000, TimeUnit.MILLISECONDS); | ||
| } catch (InterruptedException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
|
|
||
| Assertions.assertThat(Metrics.globalRegistry.find("vertx.sql.queue.pending").gauge().value()) | ||
| .isEqualTo(0.0); | ||
| } | ||
|
|
||
| @Entity(name = "GuineaPig") | ||
| @Table(name = "Pig") | ||
| public static class GuineaPig { | ||
| @Id | ||
| private Integer id; | ||
| private String name; | ||
|
|
||
| public GuineaPig() { | ||
| } | ||
|
|
||
| public GuineaPig(Integer id, String name) { | ||
| this.id = id; | ||
| this.name = name; | ||
| } | ||
|
|
||
| public Integer getId() { | ||
| return id; | ||
| } | ||
|
|
||
| public void setId(Integer id) { | ||
| this.id = id; | ||
| } | ||
|
|
||
| public String getName() { | ||
| return name; | ||
| } | ||
|
|
||
| public void setName(String name) { | ||
| this.name = name; | ||
| } | ||
|
|
||
| @Override | ||
| public String toString() { | ||
| return id + ": " + name; | ||
| } | ||
|
|
||
| @Override | ||
| public boolean equals(Object o) { | ||
| if (this == o) { | ||
| return true; | ||
| } | ||
| if (o == null || getClass() != o.getClass()) { | ||
| return false; | ||
| } | ||
| GuineaPig guineaPig = (GuineaPig) o; | ||
| return Objects.equals(name, guineaPig.name); | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| return Objects.hash(name); | ||
| } | ||
| } | ||
| } |