当前位置：首页 > article >正文

Java Stream reduce 函数，聚合数据

article 2025/2/7 14:27:18

Stream.reduce() 是 Stream 的一个聚合方法，它可以把一个 Stream 的所有元素按照自定义聚合逻辑，聚合成一个结果。

先看一个简单数字求和：

public class Main {
	public static void main(String[] args）{
	int sum = Stream.of(1, 2, 3, 4, 5, 6, 7, 8, 9).reduce(0,(acc, n) -> acc + n);
	System.out.println(sum); // 45
}

如下方法签名所示，带有三个参数的 reduce 是单线程流和多线程并行流都可以使用的。

<U> U reduce(U identity,
             BiFunction<U, ? super T, U> accumulator,
             BinaryOperator<U> combiner);

如下是一个逻辑处理收集复杂一些的示例，综合演示了 map、flatMap 和 reduce 的使用。

// 处理逻辑：将一组纯数字字符串，先使用map分别为每个元素追加一个1，
// 再使用flatMap将数字字符串转换为一个只包含单个Integer数字的新stream，
// 然后对新的单数字流分别乘以10之后在前面拼接字符串Num-，最后将结果收集到集合中。

public static void main(String[] args) {
    List<String> resultList = Stream.of("12", "34", "56")
            .map(item -> item.concat("1"))
            .flatMap((Function<String, Stream<Integer>>) str -> Stream.of(str.split("")).map(Integer::parseInt))
            .map(item -> "Num-".concat(String.valueOf(item * 10)))
            .reduce(new ArrayList<>(), (list, item) -> {
                list.add(item);
                return list;
            }, (list, item) -> {
                list.addAll(item);
                return list;
            });
    resultList.forEach(System.out::println);
}