Java Stream中利用模式匹配进行类型过滤与转换

在Java编程中，我们经常会遇到处理包含多种子类型对象的集合或流的场景。例如，一个Stream<Animal>可能包含Zebra、Lion等不同类型的Animal实例。如果我们的目标是从这个流中提取出所有Zebra类型的对象，并将其转换为Stream<Zebra>，同时利用Java 14引入的instanceof模式匹配特性，以提高代码的简洁性和安全性，就需要一些巧妙的处理方法。

传统类型过滤与转换的挑战

在Java 14之前，或者不使用模式匹配时，常见的做法是先通过filter()方法筛选出目标类型实例，再通过map()方法进行强制类型转换：

import java.util.stream.Stream;

// 假设Animal和Zebra类已定义
class Animal {}
class Zebra extends Animal {
    public int countStripes() { return 20; } // 示例方法
}

public class StreamTypeConversion {
    public static void main(String[] args) {
        Stream<Animal> animalStream = Stream.of(new Animal(), new Zebra(), new Animal(), new Zebra());

        // 传统方法：先过滤再转换
        Stream<Zebra> zebraStreamOld = animalStream
            .filter(Zebra.class::isInstance) // 过滤出Zebra实例
            .map(Zebra.class::cast);         // 强制转换为Zebra类型

        zebraStreamOld.forEach(zebra -> System.out.println("Found Zebra (old way) with " + zebra.countStripes() + " stripes"));

        // 另一种尝试（不推荐，不够优雅）
        Stream<Animal> anotherAnimalStream = Stream.of(new Animal(), new Zebra(), new Animal(), new Zebra());
        Stream<Zebra> zebraStreamUgly = anotherAnimalStream
            .map(animal -> {
                if (animal instanceof Zebra zebra) { // Java 14+ 模式匹配
                    return zebra;
                }
                return null;
            })
            .filter(java.util.Objects::nonNull); // 过滤掉null值

        zebraStreamUgly.forEach(zebra -> System.out.println("Found Zebra (ugly way) with " + zebra.countStripes() + " stripes"));
    }
}

上述“传统方法”虽然有效，但在map()中仍需显式调用cast()，而“另一种尝试”则引入了null值和额外的filter(Objects::nonNull)操作，显得不够优雅。Java 14的instanceof模式匹配提供了更简洁的语法，但如何在Stream管道中优雅地利用它，是我们需要解决的问题。

结合模式匹配与mapMulti()进行类型转换

Java 16引入的Stream.mapMulti()方法为这种“一对零或一或多”的转换场景提供了非常优雅的解决方案。它接收一个BiConsumer，其中第一个参数是当前流元素，第二个参数是一个Consumer，用于接受并发出零个或多个结果元素。

立即学习“Java免费学习笔记（深入）”；

利用mapMulti()，我们可以将instanceof模式匹配无缝集成到Stream管道中：

import java.util.stream.Stream;

// 假设Animal和Zebra类已定义
class Animal {}
class Zebra extends Animal {
    public int countStripes() { return 20; }
}

public class StreamMapMultiPatternMatching {
    public static void main(String[] args) {
        Stream<Animal> animalStream = Stream.of(new Animal(), new Zebra(), new Animal(), new Zebra());

        Stream<Zebra> zebraStream = animalStream
            .mapMulti((animal, consumer) -> {
                if (animal instanceof Zebra zebra) { // 使用模式匹配
                    consumer.accept(zebra);          // 如果是Zebra，则发出
                }
            });

        zebraStream.forEach(zebra -> System.out.println("Found Zebra (mapMulti) with " + zebra.countStripes() + " stripes"));
    }
}

代码解析：

ModelGate

一站式AI模型管理与调用工具

下载

1. animalStream.mapMulti(...)：对流中的每个animal元素执行转换操作。
2. if (animal instanceof Zebra zebra)：这是Java 14的模式匹配语法。如果animal是Zebra的实例，它不仅会返回true，还会自动将animal强制转换为Zebra类型并赋值给新的局部变量zebra。
3. consumer.accept(zebra)：如果模式匹配成功，我们将这个已经转换为Zebra类型的zebra实例传递给consumer。consumer负责将这个元素添加到新的流中。如果模式匹配失败，consumer.accept()不会被调用，也就意味着没有元素被添加到新的流中，从而实现了过滤效果。

这种方法避免了显式的filter()和map()链式调用，也避免了生成中间null值，代码更加简洁、直观。

结合模式匹配与flatMap()进行类型转换

经典的Stream.flatMap()方法也常用于一对多转换，它要求映射函数返回一个Stream。我们也可以利用flatMap()结合模式匹配来实现类型过滤和转换。

import java.util.stream.Stream;

// 假设Animal和Zebra类已定义
class Animal {}
class Zebra extends Animal {
    public int countStripes() { return 20; }
}

public class StreamFlatMapPatternMatching {
    public static void main(String[] args) {
        Stream<Animal> animalStream = Stream.of(new Animal(), new Zebra(), new Animal(), new Zebra());

        Stream<Zebra> zebraStream = animalStream
            .flatMap(animal ->
                animal instanceof Zebra zebra ? Stream.of(zebra) : null // 返回单元素流或null
            );

        zebraStream.forEach(zebra -> System.out.println("Found Zebra (flatMap) with " + zebra.countStripes() + " stripes"));
    }
}

代码解析：

1. animalStream.flatMap(...)：对流中的每个animal元素执行映射操作，期望返回一个Stream。
2. animal instanceof Zebra zebra ? Stream.of(zebra) : null：
   • 如果animal是Zebra的实例，模式匹配成功，animal被转换为Zebra类型的zebra。我们然后返回一个包含这个zebra的单元素Stream (Stream.of(zebra))。
   • 如果animal不是Zebra的实例，模式匹配失败，我们返回null。

注意事项：flatMap()与null 根据Java Stream API的文档，flatMap()方法在遇到映射函数返回null时，会将其视为一个空流 (Stream.empty())。这意味着我们可以直接返回null来表示不生成任何元素，而无需显式创建Stream.empty()，这在一定程度上简化了代码。

mapMulti()与flatMap()的选择与对比

虽然mapMulti()和flatMap()都可以实现结合模式匹配的类型过滤与转换，但它们在底层实现和性能考量上有所不同：

总结： 在进行类型过滤和转换这种“一对零或一”的特定场景下，mapMulti()通常是更推荐的选择，因为它在性能和代码简洁性上都表现更优。它避免了flatMap()在每次成功匹配时创建单元素Stream的开开销，尤其是在处理大型流时，这种性能差异可能变得显著。

总结

Java 14的instanceof模式匹配极大地提升了类型检查和转换的简洁性与安全性。结合Java 16的mapMulti()方法或经典的flatMap()方法，我们可以在Stream API中优雅地实现超类型流到子类型流的过滤与转换。对于这种“一对零或一”的转换需求，mapMulti()通常是更优的选择，它提供了更高的效率和更直观的表达方式，使得Stream管道代码更加清晰和易于维护。掌握这些现代Java特性，能够帮助开发者编写出更健壮、更高效的Stream处理逻辑。