如何从第一个元素获取另一个集合中集合的第一个元素？

Question

我想找到最有效的方法来查找集合的第一个元素，该元素位于另一个集合中，我也采用第一个元素。

我认为这是一个“丑陋”的解决方案：

public class UnitLine {}

public class Unit {

    private Collection<UnitLine> unitLines;

    public Collection<UnitLine> getUnitLines() {
        return unitLines;
    }
}

public class Wrapper {

    Collection<Unit> units;

    public Collection<Unit> getUnits() {
        return units;
    }
}

public static void main(String[] args) {

    Wrapper wrapper = new Wrapper();

    Unit unit = wrapper.getUnits().stream().findFirst().orElseGet(null);
    if (unit != null) {
        UnitLine unitLine = unit.getUnitLines().stream().findFirst().orElseGet(null);
        if (unitLine != null) {
            // do something
        }
    }
}

Answer 1

将Optional 变为null 意味着您无法利用Optional 的良好功能。这是等效于使用可选

map

和

orElse

的代码：

UnitLine unitLine = wrapper.getUnits().stream()
  .findFirst()
  .map(Unit::getLines)
  .orElse(Collections.emptySet()).stream()
  .findFirst()
  .orElse(null);

该代码获取第一个集合的第一个元素，或者

null

（如果第一个集合为空）。要获取第一个非空集合的第一个元素，可以使用

flatMap

:

wrapper.getUnits().stream()
  .map(Unit::getUnitLines)
  .flatMap(Collection::stream)
  .findFirst()

Answer 2

假设集合都不为 null 或没有 null 元素，则以下内容等效：

wrapper.getUnits().stream().findFirst()
        .flatMap(unit -> unit.getUnitLines().stream().findFirst())
        .ifPresent(unitLine -> {
            // do something
        });

这使用

Optional.flatMap

将包含

Optional<Unit>

第一个元素的

wrapper.getUnits()

转换为包含其中第一个元素的

Optional<UnitLine>

。 如果任一集合为空，则生成的可选值也为空。

Optional.ifPresent

用于对结果元素执行操作，如果该元素不存在则不执行任何操作。