使用流API为列表中的每个对象计算对象值的累积和

Question

我试图实现列表中对象值的累积和。

对象如下所示：

public class NameValuePair {
    private String name;
    private int value;
}

我有一份清单

结果也应该是一个列表

如何使用Java流API实现它？

输入示例为：

("a", 2), ("b", 12), ("c", 15), ("d", 20)

所需输出为：

("a", 2), ("b", 14), ("c", 29), ("d", 49)

Answer 1

可以在收集器内处理累积值的过程。

在这种情况下，不需要将当前值存储在流管道之外并通过副作用进行更新，API文档不鼓励这样做。

为此，我们需要定义一个自定义收集器。它可以实现为实现Collector接口的类，或者我们可以使用静态方法Collector.of（）。

这些是收集器预期的参数。of（）：

>

累加器<代码>Bi消费者

组合器<代码>二进制运算符

分页装订器功能

特性允许提供附加信息，例如收集器。特点。本例中使用的无序表示并行执行时产生的部分缩减结果的顺序不显著。此收集器不需要任何特性。

public static List<NameValuePair> accumulateValues(List<NameValuePair> pairs) {
    return pairs.stream()
        .collect(getPairAccumulator());
}

public static Collector<NameValuePair, ?, List<NameValuePair>> getPairAccumulator() {
    
    return Collector.of(
        ArrayDeque::new, // mutable container
        (Deque<NameValuePair> deque, NameValuePair pair) -> {
            if (deque.isEmpty()) deque.add(pair);
            else deque.add(new NameValuePair(pair.name(), deque.getLast().value() + pair.value()));
        },
        (left, right) -> { throw new AssertionError("should not be executed in parallel"); }, // combiner - function responsible 
        (Deque<NameValuePair> deque) -> deque.stream().toList() // finisher function
    );
}

如果您使用的是Java 16或更高版本，则可以将NameValuePair实现为记录：

public record NameValuePair(String name, int value) {}

<代码>main（）

public static void main(String[] args) {
    List<NameValuePair> pairs =
        List.of(new NameValuePair("a", 2), new NameValuePair("b", 12),
                new NameValuePair("c", 15), new NameValuePair("d", 20));

    List<NameValuePair> result = accumulateValues(pairs);
    result.forEach(System.out::println);
}

输出：

NameValuePair[name=a, value=2]
NameValuePair[name=b, value=14]
NameValuePair[name=c, value=29]
NameValuePair[name=d, value=49]

在线演示的链接

Answer 2

可以使用AtomicInteger存储累积值：

import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;


....


List<NameValuePair> list = List.of( new NameValuePair("a", 2),
                                    new NameValuePair("b", 12),
                                    new NameValuePair("c", 15),
                                    new NameValuePair("d", 20));

AtomicInteger ai = new AtomicInteger(0);

List<NameValuePair> result = list.stream()
                                 .map(nvp -> new NameValuePair(nvp.getName(), ai.addAndGet(nvp.getValue())))
                                 .collect(Collectors.toList());