|
|
|
@ -1,24 +1,26 @@
|
|
|
|
|
package com.baeldung.completablefuture;
|
|
|
|
|
|
|
|
|
|
import java.util.concurrent.*;
|
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
|
|
|
|
|
|
import java.util.concurrent.CancellationException;
|
|
|
|
|
import java.util.concurrent.CompletableFuture;
|
|
|
|
|
import java.util.concurrent.ExecutionException;
|
|
|
|
|
import java.util.concurrent.Executors;
|
|
|
|
|
import java.util.concurrent.Future;
|
|
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
import java.util.stream.Stream;
|
|
|
|
|
|
|
|
|
|
import org.junit.Test;
|
|
|
|
|
|
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
|
|
|
|
|
|
public class CompletableFutureTest {
|
|
|
|
|
public class CompletableFutureLongRunningUnitTest {
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenRunningCompletableFutureAsynchronously_thenGetMethodWaitsForResult() throws InterruptedException, ExecutionException {
|
|
|
|
|
|
|
|
|
|
Future<String> completableFuture = calculateAsync();
|
|
|
|
|
|
|
|
|
|
String result = completableFuture.get();
|
|
|
|
|
assertEquals("Hello", result);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Future<String> calculateAsync() throws InterruptedException {
|
|
|
|
@ -35,12 +37,10 @@ public class CompletableFutureTest {
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenRunningCompletableFutureWithResult_thenGetMethodReturnsImmediately() throws InterruptedException, ExecutionException {
|
|
|
|
|
|
|
|
|
|
Future<String> completableFuture = CompletableFuture.completedFuture("Hello");
|
|
|
|
|
|
|
|
|
|
String result = completableFuture.get();
|
|
|
|
|
assertEquals("Hello", result);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Future<String> calculateAsyncWithCancellation() throws InterruptedException {
|
|
|
|
@ -57,82 +57,65 @@ public class CompletableFutureTest {
|
|
|
|
|
|
|
|
|
|
@Test(expected = CancellationException.class)
|
|
|
|
|
public void whenCancelingTheFuture_thenThrowsCancellationException() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
Future<String> future = calculateAsyncWithCancellation();
|
|
|
|
|
future.get();
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenCreatingCompletableFutureWithSupplyAsync_thenFutureReturnsValue() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> future = CompletableFuture.supplyAsync(() -> "Hello");
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello", future.get());
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenAddingThenAcceptToFuture_thenFunctionExecutesAfterComputationIsFinished() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> "Hello");
|
|
|
|
|
|
|
|
|
|
CompletableFuture<Void> future = completableFuture.thenAccept(s -> System.out.println("Computation returned: " + s));
|
|
|
|
|
|
|
|
|
|
future.get();
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenAddingThenRunToFuture_thenFunctionExecutesAfterComputationIsFinished() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> "Hello");
|
|
|
|
|
|
|
|
|
|
CompletableFuture<Void> future = completableFuture.thenRun(() -> System.out.println("Computation finished."));
|
|
|
|
|
|
|
|
|
|
future.get();
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenAddingThenApplyToFuture_thenFunctionExecutesAfterComputationIsFinished() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> "Hello");
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> future = completableFuture.thenApply(s -> s + " World");
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello World", future.get());
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenUsingThenCompose_thenFuturesExecuteSequentially() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> "Hello").thenCompose(s -> CompletableFuture.supplyAsync(() -> s + " World"));
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello World", completableFuture.get());
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenUsingThenCombine_thenWaitForExecutionOfBothFutures() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> "Hello").thenCombine(CompletableFuture.supplyAsync(() -> " World"), (s1, s2) -> s1 + s2);
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello World", completableFuture.get());
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenUsingThenAcceptBoth_thenWaitForExecutionOfBothFutures() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture.supplyAsync(() -> "Hello").thenAcceptBoth(CompletableFuture.supplyAsync(() -> " World"), (s1, s2) -> System.out.println(s1 + s2));
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenFutureCombinedWithAllOfCompletes_thenAllFuturesAreDone() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> future1 = CompletableFuture.supplyAsync(() -> "Hello");
|
|
|
|
|
CompletableFuture<String> future2 = CompletableFuture.supplyAsync(() -> "Beautiful");
|
|
|
|
|
CompletableFuture<String> future3 = CompletableFuture.supplyAsync(() -> "World");
|
|
|
|
@ -150,12 +133,10 @@ public class CompletableFutureTest {
|
|
|
|
|
String combined = Stream.of(future1, future2, future3).map(CompletableFuture::join).collect(Collectors.joining(" "));
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello Beautiful World", combined);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenFutureThrows_thenHandleMethodReceivesException() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
String name = null;
|
|
|
|
|
|
|
|
|
|
// ...
|
|
|
|
@ -168,12 +149,10 @@ public class CompletableFutureTest {
|
|
|
|
|
}).handle((s, t) -> s != null ? s : "Hello, Stranger!");
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello, Stranger!", completableFuture.get());
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test(expected = ExecutionException.class)
|
|
|
|
|
public void whenCompletingFutureExceptionally_thenGetMethodThrows() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = new CompletableFuture<>();
|
|
|
|
|
|
|
|
|
|
// ...
|
|
|
|
@ -183,18 +162,15 @@ public class CompletableFutureTest {
|
|
|
|
|
// ...
|
|
|
|
|
|
|
|
|
|
completableFuture.get();
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Test
|
|
|
|
|
public void whenAddingThenApplyAsyncToFuture_thenFunctionExecutesAfterComputationIsFinished() throws ExecutionException, InterruptedException {
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> "Hello");
|
|
|
|
|
|
|
|
|
|
CompletableFuture<String> future = completableFuture.thenApplyAsync(s -> s + " World");
|
|
|
|
|
|
|
|
|
|
assertEquals("Hello World", future.get());
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|