如何使用流重写嵌套列表的查找功能

Question

给定一个List<Unicorn> ，每个独角兽都包含一个List<Rider> ，返回该独角兽和一部分骑手的最新骑手信息。

我这样写：

static Optional<ImmutablePair<Unicorn, Rider>> findLatestExperiencedRiderInfo(
        final List<Unicorn> unicorns) {

    Rider latestExperiencedRider = null;
    Unicorn unicornOfLatestExperiencedRider = null;
    long latestRideTime = 0L;

    for (final Unicorn unicorn : unicorns) {
        for (final Rider rider : unicorn.getRiders()) {
            if (rider.getType() == Rider.Type.EXPERIENCED) {
                final long time = rider.getRideTime();
                if (time > latestRideTime) {
                    latestRideTime = time;
                    latestExperiencedRider = rider;
                    unicornOfLatestExperiencedRider = unicorn;
                }
            }
        }
    }

    return latestExperiencedRider == null
            ? Optional.empty()
            : Optional.of(new ImmutablePair<>(
                    unicornOfLatestExperiencedRider,
                    latestExperiencedRider));
}

我希望有一个更简洁的版本，可能使用流（或只是用不同的方式编写）。 您有什么建议？

Answer 1

这是流媒体版本

unicorns.stream()
        .flatMap(unicorn -> unicorn.getRiders()
                .stream()
                .filter(rider -> rider.getType() == Rider.Type.EXPERIENCED)
                .map(rider -> new AbstractMap.SimpleEntry<>(unicorn, rider)))
        .max(Comparator.comparingLong(entry -> entry.getValue().getRideTime()))
        .map(entry -> new ImmutablePair<>(entry.getKey(), entry.getValue()));

flatMap部分将所有Riders展平为AbstractMap.SimpleEntry每个都包含一个Rider和它所属的Unicorn 。 然后我们通过比较两个Rider的rideTime获得最大条目。 如果存在，我们创建ImmutablePair结果，否则为Optional.empty

Answer 2

也许像这样，您可以添加所需的null检查。

// This is your rider..
final Optional<Rider> aRider = unicorns.stream()
                                       .flatMap(unicorn -> unicorn.getRiders().stream())
                                       .filter(rider -> rider.type == Rider.Type.EXPERIENCED)
                                       .max(Comparator.comparingLong(Rider::getRideTime));

 // This is your unicorn..
final Optional<Unicorn> any = unicorns.stream()
                                      .filter(unicorn -> unicorn.getRiders().contains(aRider.get()))
                                      .findAny();

我猜您也可以在Rider保留对Unicorn的引用，因此一旦有了骑手，就无需搜索独角兽。