iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BAEL-6091 structured concurrency (#13297)

This commit is contained in:
Kilian Schneider 2023-01-23 21:57:00 +01:00 committed by GitHub
parent ea9bf9ebeb
commit 69f6e11088
1 changed files with 122 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

122
core-java-modules/core-java-19/src/test/java/com/baeldung/features/JEP428StructuredConcurrencyUnitTest.java Normal file
View File

@ -0,0 +1,122 @@
package com.baeldung.features;
import jdk.incubator.concurrent.StructuredTaskScope;
import org.junit.jupiter.api.Test;
import java.time.Instant;
import java.util.List;
import java.util.concurrent.*;
import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
import static org.assertj.core.api.AssertionsForClassTypes.assertThatThrownBy;
public class JEP428StructuredConcurrencyUnitTest {
private static final String ERROR_MESSAGE = "Failed to get the result";
@Test
public void givenStructuredConcurrency_whenThrowingException_thenCorrect() {
assertThatThrownBy(() -> {
try (var scope = new StructuredTaskScope.ShutdownOnFailure()) {
Future<Shelter> shelter = scope.fork(this::getShelter);
Future<List<Dog>> dogs = scope.fork(this::getDogsWithException);
scope.throwIfFailed(e -> new RuntimeException(ERROR_MESSAGE));
scope.join();
Response response = new Response(shelter.resultNow(), dogs.resultNow());
assertThat(response).isNotNull();
assertThat(response.shelter()).isNotNull();
assertThat(response.dogs()).isNotNull();
assertThat(response.dogs().size()).isEqualTo(2);
}
}).isInstanceOf(RuntimeException.class)
.hasMessage(ERROR_MESSAGE);
}
@Test
public void givenStructuredConcurrency_whenSlowTasksReachesDeadline_thenCorrect() {
assertThatThrownBy(() -> {
try (var scope = new StructuredTaskScope.ShutdownOnFailure()) {
Future<Shelter> shelter = scope.fork(this::getShelter);
Future<List<Dog>> dogs = scope.fork(this::getDogsSlowly);
scope.throwIfFailed(e -> new RuntimeException(ERROR_MESSAGE));
scope.join();
scope.joinUntil(Instant.now().plusMillis(50));
Response response = new Response(shelter.resultNow(), dogs.resultNow());
assertThat(response).isNotNull();
assertThat(response.shelter()).isNotNull();
assertThat(response.dogs()).isNotNull();
assertThat(response.dogs().size()).isEqualTo(2);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}).isInstanceOf(IllegalStateException.class);
}
@Test
public void givenStructuredConcurrency_whenResultNow_thenCorrect() {
try (var scope = new StructuredTaskScope.ShutdownOnFailure()) {
Future<Shelter> shelter = scope.fork(this::getShelter);
Future<List<Dog>> dogs = scope.fork(this::getDogs);
scope.join();
Response response = new Response(shelter.resultNow(), dogs.resultNow());
assertThat(response).isNotNull();
assertThat(response.shelter()).isNotNull();
assertThat(response.dogs()).isNotNull();
assertThat(response.dogs().size()).isEqualTo(2);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
@Test
public void givenUnstructuredConcurrency_whenGet_thenCorrect() {
Future<Shelter> shelter;
Future<List<Dog>> dogs;
try (ExecutorService executorService = Executors.newFixedThreadPool(3)) {
shelter = executorService.submit(this::getShelter);
dogs = executorService.submit(this::getDogs);
Shelter theShelter = shelter.get(); // Join the shelter
List<Dog> theDogs = dogs.get(); // Join the dogs
Response response = new Response(theShelter, theDogs);
assertThat(response).isNotNull();
assertThat(response.shelter()).isNotNull();
assertThat(response.dogs()).isNotNull();
assertThat(response.dogs().size()).isEqualTo(2);
} catch (ExecutionException | InterruptedException e) {
throw new RuntimeException(e);
}
}
private Shelter getShelter() {
return new Shelter("Shelter");
}
private List<Dog> getDogs() {
return List.of(new Dog("Buddy"), new Dog("Simba"));
}
private List<Dog> getDogsWithException() {
throw new RuntimeException(ERROR_MESSAGE);
}
private List<Dog> getDogsSlowly() throws InterruptedException {
Thread.sleep(1500);
throw new RuntimeException(ERROR_MESSAGE);
}
record Shelter(String name) {
}
record Dog(String name) {
}
record Response(Shelter shelter, List<Dog> dogs) {
}
}