Seal CompletionStrategy interface

pivovarit · Oct 3, 2023 · 35cdffb · 35cdffb
1 parent 8d63b02
commit 35cdffb
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/src/main/java/com/pivovarit/collectors/CompletionStrategy.java b/src/main/java/com/pivovarit/collectors/CompletionStrategy.java
@@ -6,13 +6,31 @@
 import java.util.stream.Stream;
 import java.util.stream.StreamSupport;
 
-interface CompletionStrategy<T> extends Function<List<CompletableFuture<T>>, Stream<T>> {
+sealed interface CompletionStrategy<T>
+  extends Function<List<CompletableFuture<T>>, Stream<T>>
+  permits CompletionStrategy.Unordered, CompletionStrategy.Ordered {
 
     static <R> CompletionStrategy<R> unordered() {
-        return futures -> StreamSupport.stream(new CompletionOrderSpliterator<>(futures), false);
+        return new Unordered<>();
     }
 
     static <R> CompletionStrategy<R> ordered() {
-        return futures -> futures.stream().map(CompletableFuture::join);
+        return new Ordered<>();
+    }
+
+    final class Unordered<R> implements CompletionStrategy<R> {
+
+        @Override
+        public Stream<R> apply(List<CompletableFuture<R>> futures) {
+            return StreamSupport.stream(new CompletionOrderSpliterator<>(futures), false);
+        }
+    }
+
+    final class Ordered<R> implements CompletionStrategy<R> {
+
+        @Override
+        public Stream<R> apply(List<CompletableFuture<R>> futures) {
+            return futures.stream().map(CompletableFuture::join);
+        }
     }
 }