Java groupingBy：对多个字段求和

Question

这个问题是帖子的扩展：Java 8 groupingby with returning multiple field。

对于同样的问题，如何返回Customer的列表？例如，它应该返回：

Customer("A",4500,6500)
Customer("B",3000,3500)
Customer("C",4000,4500)

Answer 1

使用以下代码：

Map<String, Customer> retObj =
    listCust.stream()
        .collect(Collectors.groupingBy(Customer::getName,
                    Collector.of(
                        Customer::new,
                        (c1, c2) -> {
                            c1.setName(c2.getName());
                            c1.setTotal(c1.getTotal() + c2.getTotal());
                            c1.setBalance(c1.getBalance() + c2.getBalance());
                        },
                        (c3, c4) -> {
                            c3.setTotal(c3.getTotal() + c4.getTotal());
                            c3.setBalance(c3.getBalance() + c4.getBalance());
                            return c3;
                        })));

System.out.println(retObj);
System.out.println(retObj.values());       //If you want only the list of all Customers

输出：

{
 A=Customer [name=A, total=4500.0, balance=6500.0], 
 B=Customer [name=B, total=3000.0, balance=3500.0], 
 C=Customer [name=C, total=4000.0, balance=4500.0]
}

Answer 2

如果您想要Map<String, Customer> 作为结果集 +1，@Pankaj Singhal 的帖子是正确的想法。但是，我会将合并逻辑提取到它自己的函数中，例如在Customer 类中，您将拥有这样的功能：

public static Customer merge(Customer first, Customer second) {
        first.setTotal(first.getTotal() + second.getTotal());
        first.setBalance(first.getBalance() + second.getBalance());
        return first;
}

那么流查询会变成：

Map<String, Customer> retObj = 
           listCust.stream()
                   .collect(Collectors.toMap(Customer::getName, Function.identity(), Customer::merge)); 

• listCust.stream() 创建一个 stream 对象，即 Stream<Customer>。
• collect 对元素执行可变归约操作 此流使用提供的Collector。
• toMap 的结果是提供的收集器，toMap 方法提取键 Customer::getName 和值 Function.identity()，如果映射的键包含重复项，则使用合并函数 Customer::merge 解决冲突。

我看到将合并逻辑提取到它自己的函数中有三个好处：

• 代码更紧凑。
• 代码更具可读性。
• 合并的复杂性与流隔离。

---

但是，如果您的意图是检索Collection<Customer>：

Collection<Customer> result = listCust.stream()
                .collect(Collectors.toMap(Customer::getName,
                        Function.identity(),
                        Customer::merge))
                .values();

或List<Customer> 作为结果集，那么您所要做的就是调用values() 并将其结果传递给ArrayList 构造函数：

List<Customer> result = new ArrayList<>(listCust.stream()
                .collect(Collectors.toMap(Customer::getName,
                        Function.identity(),
                        Customer::merge))
                .values());

更新：

如果您不想改变源中的对象，则只需修改merge 函数，如下所示：

public static Customer merge(Customer first, Customer second) {
        Customer customer = new Customer(first.getName(), first.getTotal(), first.getBalance());
        customer.setTotal(customer.getTotal() + second.getTotal());
        customer.setBalance(customer.getBalance() + second.getBalance());
        return customer;
}

其他一切都保持原样。

Answer 3

这里的other answers 很棒，但是它们会改变输入中的Customer 实例，这可能是出乎意料的。

为避免这种情况，请使用自定义 Collector。

首先，创建一个返回 Collector 的方法，该方法接受 Stream<Customer> 并将它们合并为单个 Customer：

public static Collector<Customer, Customer, Customer> customerCollector() {
    return Collector.of(Customer::new, TestBench::merge,
            (l, r) -> {
                merge(l, r);
                return l;
            });
}

public static void merge(final Customer first, final Customer second) {
    first.setName(second.getName());
    first.setTotal(first.getTotal() + second.getTotal());
    first.setBalance(first.getBalance() + second.getBalance());
}

这假设 Customer 有一个 noargs 构造函数。

那么你可以这样做：

Collection<Customer> result = listCust.stream()
        .collect(groupingBy(Customer::getName, customerCollector()))
        .values();

请注意，我使用 groupingBy 而不是 toMap - groupingBy 收集器专门设计用于对元素进行分组。