Java：Group By Then Map

Question

我有一個Event流

public class Event {
    Location location;
    double turnout;
    //... other fields & getters
}

還有一個統計類EventStatistics

public class EventStatistics {
    // Stats properties e.g. turnout standard deviation/median

    public EventStatistics(List<Event> events) {
        // Generate stats
    }
}

我需要按位置對所有事件進行分組，並創建位置和事件統計信息的Map<Location, EventStatistics>

小組是：

Map<Location, List<Event>> byLocation = events.stream().collect(groupingBy(Event::getLocation));

我知道有一個重載的groupingBy(function, collector)收集器。 我可以用某種方式在單個流中生成Map<Location, EventStatistics>嗎？

Answer 1

所有你需要的是收集和然后 ：

Map<Location, EventStatistics> result = 
    events.stream()
          .collect(Collectors.groupingBy(Event::getLocation,
                                         Collectors.collectingAndThen(
                                             Collectors.toList(), 
                                             EventStatistics::new)));

Answer 2

如果您的EventStatistics能夠接受單個Events而不是完整列表，以及合並兩個統計信息的方法，如

EventStatistics {
    public EventStatistics() {}
    public void addEvent(Event e);
    public EventStatistics merge(EventStatistics toMerge);
}

那么你可以做到

groupingBy(Event::getLocation, Collector.of(EventStatistics::new, EventStatistics::accept, EventStatistics::merge));

這里，無參數構造函數是Supplier ， accept是accumulator ， merge是combiner 。

Answer 3

您可以使用Collector.of(...)構建自己的Collector Collector.of(...) ，如下所示：

Map<Location, EventStatistics> collect = events.stream().collect(groupingBy(Event::getLocation,
        Collector.of(ArrayList::new,
                     List::add,
                     (left, right) -> { left.addAll(right); return left; },
                     EventStatistics::new)
));