Combine two streams with collation

Question

I need to do some Matrix work in an efficient and flexible way and hoped I could practice my Java 8 using streams and lambdas and perhaps even get free parallelism out of it. One point I am struggling with is how to perform an operation on two streams putting the result into a third.

Consider the simple mechanism:

static final List<String> a = Arrays.asList("A", "A", "A");
static final List<String> b = Arrays.asList("B", "B", "B");

public void withoutStreams() {
    // The boring old way.
    List<String> c = new ArrayList<>();
    for (Iterator<String> ai = a.iterator(), bi = b.iterator(); ai.hasNext() && bi.hasNext();) {
        c.add(ai.next() + bi.next());
    }
    System.out.println(c);
}

Works fine but I want to use Streams.

private void withStreams() {
    List<String> c = new ArrayList<>();
    combine(a.stream(), b.stream(), c, (String x, String y) -> x + y);
}

private void combine(Stream<String> a, Stream<String> b, List<String> c, BinaryOperator<String> op) {
    // What can I do here to make it happen?
}

I fully expect we will be filling c using a Consumer of some form but extra kudos for coming up with some way of referring to a specific cell of the matrix other than using (row,col) bearing in mind that the cells will be immutable.

This could be probably be achieved with a zip function, which unfortunately was removed the JDK 8 libraries at the end. But in [this other answer](http://stackoverflow.com/q/17640754/697630) you can find some interesting alternatives. — Edwin Dalorzo, Mar 25 '15 at 15:53

score 3 · Accepted Answer · edited May 23 '17 at 11:44

3

You can use the IntStream class to simulate indexing and then .mapToObj to concatenate the corresponding to the index objects from a and b:

List<String> list = IntStream.range(0, Math.max(a.size(), b.size()))
                             .mapToObj(i -> a.get(i) + b.get(i))
                             .collect(Collectors.toList());

Applied to your method, this will look like:

private void combine(List<String> a, List<String> b, List<String> c, BinaryOperator<String> op) {
    c = IntStream.range(0, Math.max(a.size(), b.size()))
                 .mapToObj(i -> op.apply(a.get(i), b.get(i)))
                 .collect(Collectors.toList());
}

However, if you don't want to change the method's signature, here is a solution which works for all possible combinations of infinite and finite streams:

private void combine(Stream<String> a, Stream<String> b, List<String> c, BinaryOperator<String> op) {
    Iterator<String> i1 = a.iterator();
    Iterator<String> i2 = b.iterator();
    Iterable<String> i = () -> new Iterator<String>() {
        public boolean hasNext() {
            return i1.hasNext() && i2.hasNext();
        }
        public String next() {
            return op.apply(i1.next(), i2.next());
        }
    };
    c = StreamSupport.stream(i.spliterator(), false).collect(Collectors.toList());
}

edited May 23 '17 at 11:44

Community

1
1

answered Mar 25 '15 at 15:35

Konstantin Yovkov

62,134
8
100
147

I can see that this would work but it kind of defeats the purpose - I was hoping to stream the data itself rather than indexes into the data. Secondly - if I apply this technique to matrices I would need a `Pair` for row/col, not as elegant. – OldCurmudgeon Mar 25 '15 at 15:42
2

Not to mention that unless those lists are random access, accessing an indexed value directly might pay a performance price. – Edwin Dalorzo Mar 25 '15 at 15:44
After your edit - the signature of my `combine` uses `Stream` not `List` - any chance of that working? – OldCurmudgeon Mar 25 '15 at 15:44
Nice - any ideas on how to hold a reference to the target cell (I will be consuming to a matrix of immutables). Could this be a lambda/closure or something? – OldCurmudgeon Mar 25 '15 at 16:04
Can you give some more details about what you're trying to achieve? :) – Konstantin Yovkov Mar 25 '15 at 19:34
@kocko Beautiful answer! – 卢声远 Shengyuan Lu Mar 26 '15 at 09:23

卢声远 Shengyuan Lu · Answer 2 · 2015-03-26T05:27:33.060

0

Functional programming style, using recursive (no loop):

static Stream<String> combine(List<String> a, List<String> b) {
    if(a.isEmpty() ||  b.isEmpty()) {
        return Stream.empty();
    }

    return Stream.concat(
            Stream.of(a.get(0) + b.get(0)),
            combine(a.stream().skip(1).collect(Collectors.toList()), 
                    b.stream().skip(1).collect(Collectors.toList()))
            );
}

plus: I vote up kocko's answer, my answer is for fun.

edited Mar 26 '15 at 05:27

answered Mar 26 '15 at 05:15

卢声远 Shengyuan Lu

31,208
22
85
130

The solution could work if the JVM does last call optimization. – Oct 26 '15 at 18:56

Combine two streams with collation

2 Answers2