Zipping streams using JDK8 with lambda (java.util.stream.Streams.zip)

by

I needed this as well so I just took the source code from b93 and put it in a “util” class. I had to modify it slightly to work with the current API.

For reference here’s the working code (take it at your own risk…):

public static<A, B, C> Stream<C> zip(Stream<? extends A> a,
                                     Stream<? extends B> b,
                                     BiFunction<? super A, ? super B, ? extends C> zipper) {
    Objects.requireNonNull(zipper);
    Spliterator<? extends A> aSpliterator = Objects.requireNonNull(a).spliterator();
    Spliterator<? extends B> bSpliterator = Objects.requireNonNull(b).spliterator();

    // Zipping looses DISTINCT and SORTED characteristics
    int characteristics = aSpliterator.characteristics() & bSpliterator.characteristics() &
            ~(Spliterator.DISTINCT | Spliterator.SORTED);

    long zipSize = ((characteristics & Spliterator.SIZED) != 0)
            ? Math.min(aSpliterator.getExactSizeIfKnown(), bSpliterator.getExactSizeIfKnown())
            : -1;

    Iterator<A> aIterator = Spliterators.iterator(aSpliterator);
    Iterator<B> bIterator = Spliterators.iterator(bSpliterator);
    Iterator<C> cIterator = new Iterator<C>() {
        @Override
        public boolean hasNext() {
            return aIterator.hasNext() && bIterator.hasNext();
        }

        @Override
        public C next() {
            return zipper.apply(aIterator.next(), bIterator.next());
        }
    };

    Spliterator<C> split = Spliterators.spliterator(cIterator, zipSize, characteristics);
    return (a.isParallel() || b.isParallel())
           ? StreamSupport.stream(split, true)
           : StreamSupport.stream(split, false);
}

More Related Contents:

  1. Why doesn’t java.util.Collection implement the new Stream interface?
  2. Does Java SE 8 have Pairs or Tuples?
  3. Convert String array to Map using Java 8 Lambda expressions
  4. Java 8 List into Map
  5. Why filter() after flatMap() is “not completely” lazy in Java streams?
  6. How can I throw CHECKED exceptions from inside Java 8 streams?
  7. How to force max to return ALL maximum values in a Java Stream?
  8. Java 8 lambdas, Function.identity() or t->t
  9. Copy a stream to avoid “stream has already been operated upon or closed”
  10. Using Java 8’s Optional with Stream::flatMap
  11. Java 8 Streams: multiple filters vs. complex condition
  12. Partition a Java 8 Stream
  13. Retrieving a List from a java.util.stream.Stream in Java 8
  14. Java 8, Streams to find the duplicate elements
  15. How to use a Java8 lambda to sort a stream in reverse order?
  16. Java 8 Supplier with arguments in the constructor
  17. Java 8 Collectors.toMap SortedMap
  18. Why does Stream.allMatch() return true for an empty stream?
  19. Java 8 grouping using custom collector?
  20. How can I throw CHECKED exceptions from inside Java 8 lambdas/streams?
  21. How to apply multiple predicates to a java.util.Stream?
  22. Finding enum value with Java 8 Stream API
  23. Parallel stream from a HashSet doesn’t run in parallel
  24. Do Java8 lambdas maintain a reference to their enclosing instance like anonymous classes?
  25. Picking elements of a list until condition is met with Java 8 Lambdas
  26. Non-interference exact meaning in Java 8 streams
  27. In Java 8 how do I transform a Map to another Map using a lambda?
  28. Is it possible to get next element in the Stream?
  29. Group sequences of values
  30. Java 8 stream map to list of keys sorted by values

Leave a Comment