java - 将函数应用于按流分组的嵌套列表

Question

有以下类的对象列表:

class A {
   String firstName;
   String lastName;
   //.....
}

还有一个接受三个参数的方法:

List<B> someMethod(String firstName, String lastName, List<A> l);

我想根据 firstName 和 lastName 对此列表进行分组，然后将此方法应用于具有 firstName 的项目列表中的项目> 和姓氏

我尝试了以下方法:

Map<String, Map<String, List<B>>> groupedItems = l.stream()
     .collect(groupingBy(A::getFirstName, groupingBy(A::getLastName)));

List<B> result = groupedItems.keySet().stream()
     .map(firstName -> groupedItems.get(firstName).keySet().stream()
             .map(lastName -> someMethod(firstName, lastName, groupedItems.get(firstName).get(lastName))
             .collect(Collectors.toList()))
     .flatMap(Collection::stream)
     .collect(Collectors::toList);

有没有一种方法可以一次性完成此操作，而不是像现在这样？

Answer 1

您可以使用collectingAndThen来完成此操作，但在我看来，这的可读性要差得多。

List<B> result = l.stream().collect(Collectors.collectingAndThen(
    groupingBy(A::getFirstName, groupingBy(A::getLastName)),
    groupedItems -> groupedItems.keySet().stream()
            .flatMap(firstName -> 
                groupedItems.get(firstName)
                    .keySet()
                    .stream()
                    .map(lastName -> 
                        someMethod(
                                firstName, 
                                lastName, 
                                groupedItems.get(firstName).get(lastName)
                        )
                    )
            )
            .flatMap(Collection::stream)
            .collect(toList())));