使用Java Streams迭代两个相同的Set

问题描述

我有两个具有相同对象类型的HashSet。我的搜索条件类似于,在第一组中搜索,如果不存在,则在另一组中搜索。我已经尝试过使用Stream层,并按照以下步骤进行操作

Set<MyObject> firstSet = new HashSet<>();
Set<MyObject> secondSet = new HashSet<>();

这两个集合都有一些值。

Predicate<MyObject> match = myObject -> StringUtils.equals(myValue,myObject.getMyValue());

firstSet.values().stream().filter(match).findFirst()
.orElse(secondSet.values().stream().filter(match)
.findFirst().orElseThrow(()-> new MyException()));

我的匹配对象在“第一个集合”中,我尝试通过手动方式获取它,但是得到了...但是使用上述迭代,即使第一个集合具有匹配的对象,我也总是会遇到异常。请纠正我。谢谢。

解决方法

您的问题是您没有按预期使用Optional.orElse

使用Optional.orElse时,其参数将被急切地求值。这意味着您将首先搜索第二组,以解析第一组Optional.orElse的参数。

相反,请使用Optional.orElseGet,它会收到一个{懒惰地评估的Supplier

firstSet.stream()
    .filter(match)
    .findFirst()
    .orElseGet(() -> secondSet.stream()
        .filter(match)
        .findFirst()
        .orElseThrow(()-> new MyException()));

编辑:正如Holger在评论中所建议的,有一种更简单的方法:

Stream.of(firstSet,secondSet)
    .flatMap(Set::stream)
    .filter(match)
    .findFirst()
    .orElseThrow(MyException::new);

先流传输集,然后调用flatMap,以确保第一个集的元素都将出现在第二个集的元素之前。