Java 8 引入了 Stream API,使得处理集合数据变得更加简洁和高效。Stream API 允许开发者以声明式编程风格操作数据集合,而不是使用传统的迭代和条件语句。
一、基本概念
1.1 什么是 Stream
Stream 是 Java 8 中的一个新抽象,它允许对集合数据执行各种复杂的操作,例如过滤、映射、规约、收集等。Stream 不存储数据,而是从集合或其他数据源(如数组、I/O channel 等)中获取数据并进行操作。
Stream 的主要特点包括:
- 无存储:Stream 不存储数据,只是对数据进行操作。
- 函数式编程:使用 lambda 表达式进行操作,使代码更简洁。
- 延迟执行:Stream 操作是懒加载的,只有在需要结果时才会执行。
- 可组合性:多个 Stream 操作可以连成一串操作链,形成一系列的转换。
1.2 Stream 的生命周期
Stream 的操作可以分为三类:
- 源:创建 Stream 的数据源,例如集合、数组或 I/O channel。
- 中间操作:返回新的 Stream 的操作,例如过滤、映射。
- 终端操作:产生结果或副作用的操作,例如收集、计算。
一个 Stream 的生命周期可以简单描述为:
- 创建 Stream。
- 中间操作。
- 终端操作。
二、Stream API 的基本操作
2.1 创建 Stream
Stream 可以通过以下几种方式创建:
- 从集合:
List<String> list = Arrays.asList("a", "b", "c");
Stream<String> stream = list.stream();
- 从数组:
String[] array = {"a", "b", "c"};
Stream<String> stream = Arrays.stream(array);
- 从值:
Stream<String> stream = Stream.of("a", "b", "c");
- 从文件:
Stream<String> stream = Files.lines(Paths.get("path/to/file.txt"));
2.2 中间操作
中间操作返回一个新的 Stream,它们是延迟执行的,只有在终端操作执行时才会实际进行计算。常用的中间操作包括:
2.2.1 filter
filter
用于对 Stream 中的元素进行过滤,只保留满足条件的元素。
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
Stream<Integer> evenNumbers = numbers.stream().filter(n -> n % 2 == 0);
2.2.2 map
map
用于将 Stream 中的每个元素映射到另一个元素。
List<String> words = Arrays.asList("Java", "Stream", "API");
Stream<Integer> wordLengths = words.stream().map(String::length);
2.2.3 flatMap
flatMap
用于将 Stream 中的每个元素映射到一个新的 Stream,并将这些新 Stream 合并成一个 Stream。
List<List<String>> listOfLists = Arrays.asList(Arrays.asList("a", "b"), Arrays.asList("c", "d"));
Stream<String> flatStream = listOfLists.stream().flatMap(Collection::stream);
2.2.4 distinct
distinct
用于去除 Stream 中的重复元素。
List<Integer> numbers = Arrays.asList(1, 2, 2, 3, 4, 4, 5);
Stream<Integer> distinctNumbers = numbers.stream().distinct();
2.2.5 sorted
sorted
用于对 Stream 中的元素进行排序,可以传递一个比较器。
List<String> words = Arrays.asList("Java", "Stream", "API");
Stream<String> sortedWords = words.stream().sorted();
2.3 终端操作
终端操作会触发 Stream 的计算,并生成结果或副作用。常用的终端操作包括:
2.3.1 forEach
forEach
用于对 Stream 中的每个元素执行一个动作。
List<String> words = Arrays.asList("Java", "Stream", "API");
words.stream().forEach(System.out::println);
2.3.2 toArray
toArray
用于将 Stream 中的元素收集到一个数组中。
List<String> words = Arrays.asList("Java", "Stream", "API");
String[] array = words.stream().toArray(String[]::new);
2.3.3 reduce
reduce
用于将 Stream 中的元素通过一个关联函数组合起来,生成一个值。
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
int sum = numbers.stream().reduce(0, Integer::sum);
2.3.4 collect
collect
用于将 Stream 中的元素收集到一个容器中,例如 List、Set 或 Map。
List<String> words = Arrays.asList("Java", "Stream", "API");
List<String> upperCaseWords = words.stream().map(String::toUpperCase).collect(Collectors.toList());
2.3.5 count
count
用于返回 Stream 中的元素数量。
List<String> words = Arrays.asList("Java", "Stream", "API");
long count = words.stream().count();
2.3.6 findFirst
和 findAny
findFirst
用于返回 Stream 中的第一个元素(如果存在)。
List<String> words = Arrays.asList("Java", "Stream", "API");
Optional<String> first = words.stream().findFirst();
findAny
用于返回 Stream 中的任意一个元素(如果存在),常用于并行流。
List<String> words = Arrays.asList("Java", "Stream", "API");
Optional<String> any = words.stream().findAny();
2.3.7 anyMatch
、allMatch
和 noneMatch
这三个操作用于检查 Stream 中是否有任意、所有或没有元素满足指定的条件。
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
boolean anyEven = numbers.stream().anyMatch(n -> n % 2 == 0);
boolean allEven = numbers.stream().allMatch(n -> n % 2 == 0);
boolean noneNegative = numbers.stream().noneMatch(n -> n < 0);
三、并行流
Java 8 提供了并行流,可以充分利用多核处理器的优势。只需调用 parallelStream
方法即可创建一个并行流。
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
int sum = numbers.parallelStream().reduce(0, Integer::sum);
并行流通过将数据分成多个子流,并在不同的 CPU 核心上并行处理这些子流,然后再合并结果,来提高处理速度。需要注意的是,并行流适合于无状态和无副作用的操作,使用时需小心处理共享变量和同步问题。
四、Stream API 的最佳实践
4.1 使用 Lambda 表达式
Stream API 通常与 lambda 表达式一起使用,使代码更加简洁和易读。例如:
List<String> words = Arrays.asList("Java", "Stream", "API");
List<String> upperCaseWords = words.stream().map(word -> word.toUpperCase()).collect(Collectors.toList());
4.2 避免使用修改状态的中间操作
Stream 操作应该是无副作用的,即不应修改外部状态。以下示例展示了一个错误的用法:
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
List<Integer> results = new ArrayList<>();
numbers.stream().forEach(n -> results.add(n * 2)); // 这样做是错误的
正确的做法是使用终端操作 collect
:
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
List<Integer> results = numbers.stream().map(n -> n * 2).collect(Collectors.toList());
4.3 利用方法引用
方法引用可以使代码更加简洁。例如,使用方法引用替代 lambda 表达式:
List<String> words = Arrays.asList("Java", "Stream", "API");
List<String> upperCaseWords = words.stream().map(String::toUpperCase).collect(Collectors.toList());
4.4 避免使用并行流进行小任务
并行流在处理大量数据或复杂计算时非常高效,但对于小任务,启动并行计算的开销可能会大于收益。因此,在数据量较小或计算较简单的情况下,优先使用顺序流。
4.5 避免在终端操作之前调用 findAny
在终端操作之前调用 findAny
会导致流的中间操作链被截断,进而无法正确执行后续的操作。例如:
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
Optional<Integer> result = numbers.stream().filter(n -> n % 2 == 0).findAny(); // 这样做会中断流
应将 findAny
用作终端操作:
List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
Optional<Integer> result = numbers.stream().filter(n -> n % 2 == 0).findAny();
4.6 使用 collect
进行结果收集
collect
是一个强大的终端操作,可以将流中的元素收集到各种容器中。例如,收集到 List:
List<String> words = Arrays.asList("Java", "Stream", "API");
List<String> wordList = words.stream().collect(Collectors.toList());
4.7 使用 Collectors
进行复杂收集操作
Collectors
提供了多种收集器,可以进行复杂的结果收集。例如,收集到 Map:
List<String> words = Arrays.asList("Java", "Stream", "API");
Map<Integer, List<String>> wordLengthMap = words.stream().collect(Collectors.groupingBy(String::length));
4.8 使用 Optional
处理可能的空值
Stream API 中的某些终端操作会返回 Optional
,例如 findFirst
、findAny
。使用 Optional
可以避免空指针异常:
List<String> words = Arrays.asList("Java", "Stream", "API");
Optional<String> firstWord = words.stream().findFirst();
firstWord.ifPresent(System.out::println);
Java 8 的 Stream API 为集合数据的处理提供了一种高效、简洁的方式。通过理解和掌握 Stream 的基本概念、常用操作以及最佳实践,可以大大提高 Java 开发的生产力和代码质量。
Stream API 不仅支持顺序流,还支持并行流,使得在多核环境下处理大量数据变得更加高效。在实际开发中,合理使用 Stream API 可以显著提升代码的可读性和稳定性。