集合操作stream出了以后简化了代码和增强了可读性。
今天使用map和flatMap简单记录一下。
例:
public class Test {
public static void main(String[] args) {
Te te = new Te();
List<Integer> list = new ArrayList<>();
list.add(1);
list.add(2);
list.add(3);
list.add(4);
list.add(5);
Stream<Integer> stream = list.stream();
List<Long> rs = list.stream().map(te::tef).collect(Collectors.toList());
System.out.println(rs);
List<String> strings = new ArrayList<>();
strings.add("test");
strings.add("hesh");
List<String> list1 = strings.stream().map(te::tefList).flatMap(Collection::stream).collect(Collectors.toList());
System.out.println(list1);
}
}
public class Te {
public Long tef(Integer d){
System.out.println("输出参数:"+d);
return (d+10L);
}
public List<String> tefList(String d){
String[] strings = d.split("");
return Arrays.stream(strings).collect(Collectors.toList());
}
}
结果:
输出参数:1
输出参数:2
输出参数:3
输出参数:4
输出参数:5
[11, 12, 13, 14, 15]
[t, e, s, t, h, e, s, h]
<R> Stream<R> map(Function<? super T, ? extends R> mapper);
.map():类似于单一处理,如果方法是单参数会默认传参。
<R> Stream<R> flatMap(Function<? super T, ? extends Stream<? extends R>> mapper);
.flatMap():类似与拆分处理。flatMap入参是Function,Function中? extends Stream,所以入参必须是Stream。例子中的flatMap,它处理的是map转换之后的,tefList返回的是一个List,,而List继承自Collection,所以此处的.flatMap(Collection::stream)是一个转化的作用,将List<>转成Stream<>,然后由flatMap拆分,在由collect聚合
|