Question

使用Java 8 Streams，是否有可能以某种不会破坏流管道的方式封装和重用中间流操作？

中考虑这个例子

double average = roster
    .stream()
    .filter(p -> p.getGender() == Person.Sex.MALE)
    .mapToInt(Person::getAge)
    .average()
    .getAsDouble();

假设我需要在整个代码中的不同位置使用filter和mapToInt操作。我可能想尝试并封装该逻辑，以便可以重复使用，例如：

IntStream maleAges(Stream<Person> stream) {
    return stream
        .filter(p -> p.getGender() == Person.Sex.MALE)
        .mapToInt(Person::getAge)
}

这很好，但要使用它我可能不得不弄乱流管道。例如，如果我想要名为Bob的男性的平均年龄：

double averageBob =
    maleAges(roster
        .stream()
        .filter(p -> "Bob".equals(p.getName()))
    )
    .average()
    .getAsDouble();

有没有更好的方法呢？我正在考虑这些方面的事情：

double averageBob = roster
    .stream()
    .filter(p -> "Bob".equals(p.getName()))
    .apply(this::maleAges) // Doesn't compile
    .average()
    .getAsDouble();

Answer 1

这样做的一种方法是取一个Stream，取一个Person并发出一个IntStream，如：

 final Predicate<Person> isMale = p -> p.getGender() == Person.Sex.MALE;
 final Function<Person, IntStream> maleAges = person -> isMale.test(person)
            ? IntStream.of(person.age)
            : IntStream.empty();
 final double averageT = roster
                .stream()
                .flatMapToInt(maleAges)
                .average()
                .getAsDouble();

这样您可以在任何地方重复使用maleAges功能！

Answer 2

我的StreamEx库增强了其他功能中的标准Stream API，chain方法与您提议的apply完全相同：

double averageBob = StreamEx.of(roster)
        .filter(p -> "Bob".equals(p.getName()))
        .chain(this::maleAges) // Compiles!
        .average()
        .getAsDouble();

另一种可能的替代方法是从maleAges返回一个函数：

Function<Stream<Person>, IntStream> maleAges() {
    return stream -> stream
        .filter(p -> p.getGender() == Person.Sex.MALE)
        .mapToInt(Person::getAge);
}

并像这样使用它：

double averageBob = StreamEx.of(roster)
        .filter(p -> "Bob".equals(p.getName()))
        .chain(maleAges()) // Compiles!
        .average()
        .getAsDouble();

通过这种方式，您可以轻松地参数化封装操作。例如：

Function<Stream<Person>, IntStream> agesForSex(Person.Sex sex) {
    return stream -> stream
        .filter(p -> p.getGender() == sex)
        .mapToInt(Person::getAge);
}

double averageBob = StreamEx.of(roster)
        .filter(p -> "Bob".equals(p.getName()))
        .chain(agesForSex(Person.Sex.MALE)) // Compiles!
        .average()
        .getAsDouble();

可以封装中间流操作而不会破坏管道吗？

2 个答案: