chore(utils): Refactor FutureUtils with KIP-707

streamthoughts · Oct 14, 2021 · 81b413b · 81b413b
1 parent 18edfdc
commit 81b413b
Showing 1 changed file with 4 additions and 17 deletions.
diff --git a/src/main/java/io/streamthoughts/kafka/specs/internal/FutureUtils.java b/src/main/java/io/streamthoughts/kafka/specs/internal/FutureUtils.java
@@ -31,25 +31,12 @@
 
 public class FutureUtils {
 
-    public static <T> CompletableFuture<T> toCompletableFuture(final KafkaFuture<T> listings) {
-        return CompletableFuture.supplyAsync(() -> {
-            try {
-                return listings.get();
-            } catch (InterruptedException | ExecutionException e) {
-                throw new RuntimeException(e);
-            }
-        });
+    public static <T> CompletableFuture<T> toCompletableFuture(final KafkaFuture<T> future) {
+        return future.toCompletionStage().toCompletableFuture();
     }
 
-    public static CompletableFuture<Void> toVoidCompletableFuture(final KafkaFuture<?> listings) {
-        return CompletableFuture.supplyAsync(() -> {
-            try {
-                listings.get();
-                return null;
-            } catch (InterruptedException | ExecutionException e) {
-                throw new RuntimeException(e);
-            }
-        });
+    public static CompletableFuture<Void> toVoidCompletableFuture(final KafkaFuture<?> future) {
+        return toCompletableFuture(future).thenApply(p -> null);
     }
 
     public static <T extends Change<T>> CompletableFuture<OperationResult<T>> makeCompletableFuture(