最近的需求里有这样一个场景,要校验一个集合中每个对象的多个Id的有效性。比如一个Customer对象,有3个Id:id1
,id2
,id3
,要把这些Id全部取出来,然后去数据库里查询它们是否存在。
@Data
public class Customer {
private String name;
private String id1;
private String id2;
private String id3;
}
通常情况下,我们都是从集合中取出对象的某一个字段,像这样:
List<String> id1s =
customerList.stream().map(Customer::getId1).filter(Objects::nonNull).collect(Collectors.toList());
现在要取3个字段,怎么做呢?
Stream.concat
Stream接口中的静态方法concat,可以把两个流合成一个,我们取3个字段可以合并两次:
Stream<String> concat = Stream.concat(
customerList.stream().map(Customer::getId1),
customerList.stream().map(Customer::getId2));
List<String> ids = Stream.concat(concat, customerList.stream().map(Customer::getId3))
.filter(Objects::nonNull)
.collect(Collectors.toList());
取4个字段,就再继续合并。但是这种不够简洁,可以使用扁平化流flatMap
。
flatMap
<R> Stream<R> flatMap(Function<? super T, ? extends Stream<? extends R>> mapper);
flatMap
方法让你把一个流中的每个值都换成另一个流,然后把所有的流连接起来成为一个流。
Stream.flatMap
方法的入参为一个Function
函数,函数返回值的泛型要求为Stream类型。对比来看,map
和flatMap
都是将流中的元素映射为我们想要的值,只是flatMap
把流中元素映射为一个新的Stream。
Stream.of(T... values)
方法将多个元素构建成一个流,相当于Arrays.stream
方法,元素本身为Stream时就可以构建一个泛型为Stream的原始流,以供flatMap
操作。
List<String> ids = Stream.of(customerList.stream().map(Customer::getId1),
customerList.stream().map(Customer::getId2),
customerList.stream().map(Customer::getId3))
.flatMap(idStream -> idStream)
.filter(Objects::nonNull)
.collect(Collectors.toList());
上面的代码就相当于,Stream.of(stream, stream, stream)
, 得到的结果就是Stream<Stream>
,而flatMap
要将元素映射为Stream,所以flatMap
中的Lambda表达式返回本身即可,然后把多个流合成一个新流。
加深印象
再举一个例子,假设有这样一个需求:有一个由逗号连接的字符串组成的数组,如{"1,2,3", "3,4,5"}
,要求合并为一个数组,并去重,如{"1", "2", "3", "4", "5"}
。文章来源:https://www.toymoban.com/news/detail-434478.html
public static void main(String[] args) {
String[] strArray = {"1,2,3", "3,4,5"};
List<String> collect = Arrays.stream(strArray) // Stream<String>
.map(str -> str.split(",")) // Stream<String[]>
.flatMap(Arrays::stream) // Stream<String>
.distinct()
.collect(Collectors.toList());
System.out.println(collect);
}
map
函数将元素映射为一个数组,flatMap
函数再将元素映射为一个Stream,并将所有Stream合并成一个新Stream,然后就能愉快操作流中的每个元素了。文章来源地址https://www.toymoban.com/news/detail-434478.html
到了这里,关于Java8 Stream流的合并的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!