使用流API分组时，多个组中的单个元素

Question

我正在查看一些旧代码，在其中对元素进行分组。看起来或多或少是这样的：

Map<Long,List<Items>> groupedItems = ...
for (long groupid : groups){
    for (Item item :items){
        if (isGroupAccepting(item.getId(),groupid) || groupid == item.getGroup()) {
            groupedItems.get(groupid).add(item);
        }
    }
}

我计划使用流API中的分组替换它，但是我遇到了麻烦。对于我的第二个条件，它工作正常，但是如何处理第一个条件，即应将项添加到接受此类项的每个组中？实际上有可能吗，还是我在这里与失落的原因作斗争？

Answer 1

您可以创建所有有效组ID和项目对，然后按组ID对它们进行分组：

Map<Long,List<Item>> groupedItems =
    groups.stream()
          .flatMap(g -> items.stream()
                             .filter(i -> isGroupAccepting(i.getId(),g) || g == i.getGroup())
                             .map(i -> new SimpleEnty<>(g,i))
          .collect(Collectors.groupingBy(Map.Entry::getKey,
                   Collectors.mapping(Map.Entry::getValue,
                                      Collectors.toList())));