在Java Stream API中，我们经常需要对对象集合进行过滤。通常情况下，我们可以直接基于对象的某个属性进行过滤，例如：
// 过滤所有会游泳的动物 StreamswimmingAnimalStream = animalStream .filter(Animal::canSwim);
当过滤条件稍微复杂时，我们通常会使用Lambda表达式：
// 过滤所有绿色的动物 StreamgreenAnimals = animalStream .filter(animal -> animal.getColor().equals(Colors.GREEN));
然而，有时我们面临一个特殊的需求：我们希望根据对象某个“派生”属性（即通过某种映射函数获取的属性）进行过滤，但过滤完成后，Stream中保留的仍然是原始的完整对象，而不是被映射后的属性值。例如，我们想根据动物的颜色过滤，但最终得到的仍然是Animal对象，而不是Color对象。直接使用map后再filter会导致Stream的元素类型改变，这并非我们所期望的：
// 错误示例：此操作将导致Stream中只剩下Color信息
animalStream
.map(Animal::getColor)
.filter(Colors.GREEN::equals); // 此时Stream是Stream我们理想中的过滤方式，是能够在一个操作中同时指定映射函数和过滤谓词，例如：
立即学习“Java免费学习笔记（深入）”；
// 设想中的理想API：在过滤时指定映射函数 // animalStream.filter(Colors.GREEN::equals, Animal::getColor);
甚至更进一步，我们可能希望能够处理更复杂的映射或多个映射函数。下面将介绍两种实现这种高级过滤模式的有效方法，分别适用于Java 16及更高版本和更早的Java版本。
Java 16+ 解决方案：使用 Stream#mapMulti
从Java 16开始，Stream接口引入了一个强大的新方法mapMulti，它提供了一种更灵活的方式来转换和过滤Stream元素。mapMulti接受一个BiConsumer，允许我们在处理每个元素时，根据条件选择性地将零个、一个或多个元素放入下游Stream。这完美契合了我们“基于映射值过滤但保留原始对象”的需求。
工作原理：mapMulti的BiConsumer参数签名为(T t, Consumer
示例代码： 要实现根据颜色过滤动物，但保留原始Animal对象，我们可以这样做：
import java.util.List;
import java.util.stream.Stream;
// 假设存在Animal类和Colors枚举
class Animal {
enum Colors { RED, GREEN, BLUE }
private String name;
private Colors color;
private Animal mother; // 示例：可能存在母亲对象
public Animal(String name, Colors color) {
this.name = name;
this.color = color;
}
public Animal(String name, Colors color, Animal mother) {
this.name = name;
this.color = color;
this.mother = mother;
}
public Colors getColor() { return color; }
public String getName() { return name; }
public Animal getMother() { return mother; }
public boolean canSwim() { return name.contains("Fish") || name.contains("Duck"); }
@Override
public String toString() {
return "Animal{" + "name='" + name + '\'' + ", color=" + color + '}';
}
}
public class AdvancedStreamFilter {
public static void main(String[] args) {
List animals = List.of(
new Animal("Lion", Animal.Colors.RED),
new Animal("Frog", Animal.Colors.GREEN),
new Animal("Parrot", Animal.Colors.BLUE),
new Animal("Snake", Animal.Colors.GREEN, new Animal("Mama Snake", Animal.Colors.GREEN)),
new Animal("Fish", Animal.Colors.BLUE)
);
Stream animalStream = animals.stream();
// 使用mapMulti过滤绿色动物，保留原始Animal对象
Stream greenAnimalsStream = animalStream
.mapMulti((animal, consumer) -> {
// 条件判断：如果动物颜色是绿色
if (Animal.Colors.GREEN.equals(animal.getColor())) {
// 满足条件，则将原始Animal对象放入下游Stream
consumer.accept(animal);
}
});
greenAnimalsStream.forEach(System.out::println);
// 输出:
// Animal{name='Frog', color=GREEN}
// Animal{name='Snake', color=GREEN}
// 另一个示例：过滤有母亲的动物（假设getMother返回Optional或null，这里简化为null判断）
animalStream = animals.stream(); // 重置Stream
Stream animalsWithMother = animalStream
.mapMulti((animal, consumer) -> {
if (animal.getMother() != null) { // 检查是否存在母亲
consumer.accept(animal);
}
});
System.out.println("\nAnimals with a mother:");
animalsWithMother.forEach(System.out::println);
// 输出:
// Animals with a mother:
// Animal{name='Snake', color=GREEN}
}
} 优点：
- 性能优化： mapMulti 可以在一次遍历中完成映射和过滤，避免了 map 和 filter 链式调用可能带来的两次迭代开销（尽管JVM通常会优化）。
- 高度灵活性： consumer.accept() 可以被调用任意次。这意味着你可以根据一个输入元素生成零个、一个或多个输出元素。例如，如果一个动物有多个子颜色属性，你可以将它分解成多个具有单一颜色属性的动物。
- 复杂逻辑处理： 适合需要复杂条件判断或一对多转换的场景。
注意事项：
- 声明性降低： 相比于简洁的 filter 操作，mapMulti 的 Lambda 表达式内部更偏向于命令式编程风格，可读性在简单场景下可能不如直接 filter。
- 适用场景： 对于简单的过滤条件，其优势可能不明显。只有当需要 mapMulti 提供的独特灵活性（如性能优化或一对多转换）时，才建议使用。
Java 8+ (通用) 解决方案：直接使用 Stream#filter 结合 Lambda 表达式
在大多数情况下，以及对于Java 8及更高版本，最简洁、最符合Stream API设计哲学的方法，仍然是直接在Stream#filter中使用一个Lambda表达式。这个Lambda表达式会访问原始对象，并在其中执行必要的映射和判断逻辑。
示例代码： 要实现根据颜色过滤动物，并保留原始Animal对象，我们可以简单地这样做：
import java.util.List;
import java.util.stream.Stream;
// ... (Animal类定义同上)
public class SimpleStreamFilter {
public static void main(String[] args) {
List animals = List.of(
new Animal("Lion", Animal.Colors.RED),
new Animal("Frog", Animal.Colors.GREEN),
new Animal("Parrot", Animal.Colors.BLUE),
new Animal("Snake", Animal.Colors.GREEN, new Animal("Mama Snake", Animal.Colors.GREEN)),
new Animal("Fish", Animal.Colors.BLUE)
);
Stream animalStream = animals.stream();
// 直接在filter中判断动物颜色
Stream greenAnimalsStream = animalStream
.filter(animal -> Animal.Colors.GREEN.equals(animal.getColor()));
greenAnimalsStream.forEach(System.out::println);
// 输出:
// Animal{name='Frog', color=GREEN}
// Animal{name='Snake', color=GREEN}
// 另一个示例：过滤有母亲的动物
animalStream = animals.stream(); // 重置Stream
Stream animalsWithMother = animalStream
.filter(animal -> animal.getMother() != null);
System.out.println("\nAnimals with a mother:");
animalsWithMother.forEach(System.out::println);
// 输出:
// Animals with a mother:
// Animal{name='Snake', color=GREEN}
}
} 优点：
- 简洁性与可读性： 这种方式非常直观，易于理解，符合Stream API的声明式风格。
- 标准实践： 这是Java Stream API中最常见和推荐的过滤方式。
- 无需引入新API： 适用于所有支持Lambda表达式的Java版本（Java 8及更高）。
总结：
在Java Stream中，如果你的目标是根据对象的某个属性进行过滤，并且过滤后仍然需要原始对象，那么最直接、最推荐的方法是直接在Stream#filter操作中使用一个Lambda表达式。这个Lambda表达式可以访问并操作原始对象，执行任何必要的判断逻辑。
只有当你的过滤逻辑非常复杂，或者需要在一个操作中实现一对多（或多对零）的转换，并且追求极致的性能优化时，Java 16及更高版本提供的Stream#mapMulti才是一个更强大、更灵活的选择。它允许你在一个更命令式的回调中精细控制哪些元素进入下游Stream，以及如何进入。在实际开发中，应根据具体需求和代码的清晰度来选择最合适的方案。
