public static <T,U,A,R> Collector<T,?,R> flatMapping(Function<? super T,? extends Stream<? extends U>> mapper, Collector<? super U,A,R> downstream)
先看一个简单例子:
void test22() {
List<Integer> list = List.of(2,5,8,9,4,20,11,43,55);
list.stream().collect(Collectors.flatMapping(a -> Stream.of(a).filter(b->b>11), Collectors.toList())).stream().forEach(System.out::println);
}
有没有发现和test21中的filtering的功能类似,但是仔细看一下还是有差异的,首先flatMapping第一个参数结果为Stream,而filtering第一参数结果为一个布尔值,Stream可以继续后续操作,这就给了flatMapping扩大了很多的功能,下面给一个flatMapping复杂的示例:
void test23() {
Map<Integer, List<Integer>> map = Stream.of(List.of(1, 2, 3, 4, 5, 6), List.of(7, 8, 9, 10), List.of(5, 8, 12, 10))
.collect(Collectors.groupingBy(
Collection::size,
Collectors.flatMapping(
l -> l.stream().filter(i -> i % 2 == 0), Collectors.toList())
)
);
System.out.println(map);
}
这个例子很好的说明了flatMapping的使用,第一个参数在l.stream的时候得到一个Stream,然后还以进行后续的filter操作,该代码执行结果如下:
{4=[8, 10, 8, 12, 10], 6=[2, 4, 6]}
如果把filter去掉,直接l -> l.stream(),返回结果如下:
{4=[7, 8, 9, 10, 5, 8, 12, 10], 6=[1, 2, 3, 4, 5, 6]}
在l.stream().filter后得到的还是一个Stream,因此还可以进行后续操作,例如修改如下:
l -> l.stream().filter(i -> i % 2 == 0).filter(j -> j>4), Collectors.toList())
执行结果如下:
{4=[8, 10, 8, 12, 10], 6=[6]}
当然,第二个filter条件可以合并到第一个filter,这只是一个例子,根据实际需求第二个filter可以改为distinct,sorted等等,反正是只有你想不到的,没有我做不到的。
再看一个例子:
void test24() {
List<List<Integer>> list = List.of(List.of(List.of(2,22,43,9,4,20,11,43,55),List.of(3,77,34,33,23,5),List.of(32,34,55,76,9,22,77,88,99));
Map<Integer, List<Integer>> map = list.stream().collect(Collectors.groupingBy(Collection::size,
Collectors.flatMapping(a -> a.stream().distinct().sorted(), Collectors.toList())));
System.out.println(map);
}
执行结果如下:
{6=[3, 5, 23, 33, 34, 77], 9=[2, 4, 9, 11, 20, 22, 43, 55, 9, 22, 32, 34, 55, 76, 77, 88, 99]}从上面结果可以看到groupingBy方法是先执行flatMapping然后再groupingBy的,也就是说先对list中的元素进行去重和排序,然后再分组。