Add Gatherers.zipWithIndex()

Author: pivovarit

pom.xml

@@ -80,6 +80,9 @@
             <plugin>
                 <artifactId>maven-surefire-plugin</artifactId>
                 <version>3.5.1</version>
+                <configuration>
+                    <argLine>--enable-preview</argLine>
+                </configuration>
             </plugin>
 
             <plugin>

src/main/java/com/pivovarit/gatherers/Gatherers.java

@@ -0,0 +1,24 @@
+package com.pivovarit.gatherers;
+
+import java.util.Map;
+import java.util.concurrent.atomic.AtomicLong;
+import java.util.stream.Gatherer;
+
+import static java.util.stream.Gatherer.Integrator.ofGreedy;
+import static java.util.stream.Gatherer.ofSequential;
+
+public final class MoreGatherers {
+
+    private MoreGatherers() {
+    }
+
+    public static <T> Gatherer<T, ?, Map.Entry<Long, T>> zipWithIndex() {
+        return ofSequential(
+          AtomicLong::new,
+          ofGreedy((state, element, downstream) -> {
+              downstream.push(Map.entry(state.getAndIncrement(), element));
+              return true;
+          })
+        );
+    }
+}

src/main/java/com/pivovarit/gatherers/MoreGatherers.java

@@ -0,0 +1,28 @@
+package com.pivovarit.gatherers;
+
+import org.junit.jupiter.api.Test;
+
+import java.util.stream.Stream;
+
+import static com.pivovarit.gatherers.MoreGatherers.zipWithIndex;
+import static java.util.Map.entry;
+import static org.assertj.core.api.Assertions.assertThat;
+
+class ZipWithIndexTest {
+
+    @Test
+    void shouldZipEmptyStream() {
+        assertThat(Stream.empty().gather(zipWithIndex())).isEmpty();
+    }
+
+    @Test
+    void shouldZipWithIndex() {
+        assertThat(Stream.of("a", "b", "c")
+          .gather(zipWithIndex()))
+          .containsExactly(
+            entry(0L, "a"),
+            entry(1L, "b"),
+            entry(2L, "c")
+          );
+    }
+}