Java 8:如何优雅消除双重stream().findFirst()提取嵌套集合元素?
优化HashMap嵌套ArrayList的元素提取代码
当然可行,这里有两种更简洁优雅的写法,既能消除两次stream().findFirst()的重复逻辑,还能提升代码可读性和性能:
方法一:使用flatMap平铺流
直接将外层的List流平铺为元素流,一次获取目标元素:
map.values() .stream() .flatMap(List::stream) .findFirst() .orElse(null)
flatMap会把每个ArrayList中的元素直接"展开"到外层流中,由于每个ArrayList仅含单个元素,最终流里的第一个元素就是你要提取的目标值。
方法二:使用Optional.flatMap避免冗余对象创建
如果想避免原代码中orElse(new ArrayList<>())创建空列表的开销,可以借助Optional的flatMap方法链式处理:
map.values() .stream() .findFirst() .flatMap(list -> list.stream().findFirst()) .orElse(null)
外层findFirst()返回Optional<List<T>>,通过flatMap直接将其转换为内层元素的Optional<T>,如果外层Optional为空(即map没有值),flatMap会直接返回空Optional,无需创建空ArrayList。
两种写法都比原代码更简洁,且保留了原有的健壮性——当map为空或ArrayList意外为空时,都会返回null。
内容的提问来源于stack exchange,提问作者lando
相关产品推荐
相关产品推荐

