Use greedy Integrator for MoreGatherers.sample() (#20)

pivovarit · Oct 9, 2024 · 94de294 · 94de294
1 parent 31afeee
commit 94de294
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/main/java/com/pivovarit/gatherers/MoreGatherers.java b/src/main/java/com/pivovarit/gatherers/MoreGatherers.java
@@ -9,6 +9,7 @@
 import java.util.function.BiFunction;
 import java.util.function.Function;
 import java.util.stream.Gatherer;
+import java.util.stream.Gatherer.Integrator;
 import java.util.stream.Stream;
 
 import static java.util.stream.Gatherer.Integrator.ofGreedy;
@@ -27,13 +28,13 @@ private MoreGatherers() {
           ? noop()
           : Gatherer.ofSequential(
           () -> new AtomicLong(),
-          (state, element, downstream) -> {
-              if (state.getAndIncrement() % n == 0) {
-                  downstream.push(element);
-              }
-              return true;
-          }
-        );
+          Integrator.ofGreedy((state, element, downstream) -> {
+                if (state.getAndIncrement() % n == 0) {
+                    downstream.push(element);
+                }
+                return true;
+            }
+          ));
     }
 
     public static <T, U> Gatherer<T, ?, T> distinctBy(Function<? super T, ? extends U> keyExtractor) {