java - 我可以内联这些流运算符序列吗？

Question

我有以下代码:

public List<PolygonStat> groupItemsByTypeAndWeight(List<Item> items)
    {
        Map<Type, List<Item>> typeToItem = items
                .stream()
                .collect(
                        Collectors.groupingBy(
                                item -> item.type,
                                Collectors.toList()
                        )
                );
        // For some reason we want to make a distinction between weighted items within type
        ArrayList<WeightedItem> weightedItems = new ArrayList<>();
        typeToItem.forEach(
                // List to list function
                (type, items) -> weightedItems.addAll(createWeightedList(type, items))
        );
        return weightedItems;
    }

我不太喜欢我的创建方式 ArrayList<WeightedItem> weightedItems = new ArrayList<>();这里。有没有机会减少到1 return运算符(即: return items.stream().(...).toList() 。我考虑过使用 flatMap 但 forEach 对于 .entrySet 应该返回 void 。

Answer 1

您可以不将中间结果保存到映射中，而只需从其entrySet 创建一个新流。然后通过使用 map() 操作，您可以将每个条目映射到新的 WeightedItem。

public List<PolygonStat> groupItemsByTypeAndWeight(List<Item> items){
    return items.stream()
        .collect(Collectors.groupingBy(item -> item.type))
        .entrySet()
        .stream()
        .map(entry -> createdWeightedList(entry.getKey(), entry.getValue()))
        .flatMap(Collection::stream)
        .collect(Collectors.toList());
}