Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Java flatMap - whats the difference stream.of() and collection.stream()

I'm trying to understand flatMap: flatMap(x->stream.of(x) ) does not flat the stream and flatMap(x->x.stream()) works and gives the desired result. Can someone explain the difference between two?

import java.util.*;
import java.util.stream.*;

class TestFlatMap{

    public static void main(String args[]){
        List<String> l1 = Arrays.asList("a","b");
        List<String> l2 = Arrays.asList("c","d");

        Stream.of(l1, l2).flatMap((x)->Stream.of(x)).forEach((x)->System.out.println(x));

        Stream.of(l1, l2).flatMap((x)->x.stream()).forEach((x)->System.out.println(x));
    }

}

Output :

[a, b]
[c, d]
a
b
c
d
like image 769
puvi Avatar asked Dec 14 '22 06:12

puvi


1 Answers

Stream.of(x) produces a Stream of a single element - x. Therefore, flatMap returns a Stream<List<String>> instead of Stream<String>.

On the other hand, x.stream() where x is a Collection<E> returns a Stream<E> whose source are the elements of the Collection, so in your case it returns a Stream<String>, which allows flatMap to produce a Stream<String> containing all the Strings in all the List<String>s of the source Stream.

You can see that in the Javadoc:

<T> Stream<T> java.util.stream.Stream.of(T t)
Returns a sequential Stream containing a single element.

vs.

Stream<E> stream()
Returns a sequential Stream with this collection as its source.

like image 110
Eran Avatar answered Apr 30 '23 09:04

Eran