使用 java 8 流将集合转换为 Map<String,Collection<String>> [关闭]

Question

我希望使用 java 流将 Collection<Placement> 转换为 Map<String, Collection<String>>，其中 Placement 类如下

public class Placement {

   private String id;

   /* Links two placements, may be same for atleast 2 placements. */
   private String futureLinkId;
}

地图的键是未来链接 ID，值是放置 ID。例如，如果有 2 个展示位置 p1 { id=1, futureLinkId=f1 }, p2 { id=2, futureLinkId=f1 } ，输出应该是Map { f1 - { id1, id2 } }

Answer 1

这实际上很简单，尽管其他人提供了复杂的解决方案：

Map<String, List<String>> map = placements.stream()
        .collect(groupingBy(Placement::getFutureLinkId, mapping(Placement::getId, toList())));

Answer 2

使用three-argument toMap 是完成您需要的一种方式。 （groupingBy 似乎是另一种方法，但我不太熟悉这种方法。）

对于toMap，您需要一个将Placement 映射到键的函数；将您的 Placement 映射到一个值的函数，在这种情况下，该值需要是一个 set 值；最后是一个函数，如果它们具有相同的键，它将把两个值合并在一起。类似的东西

Map<String,HashSet<String>> result =
    placements.stream()
        .collect(Collectors.toMap
             (placement -> placement.futureLinkId,
              placement -> new HashSet(Collections.singleton(placement.id)),
              (s1, s2) -> { HashSet<String> result = new HashSet<>(s1);
                            result.add(s2);
                            return result; }));

注意：我没有对此进行测试。如果流是顺序流，这可能无需复制 s1 即可工作。

Answer 3

您可以简单地按futureLinkId 分组，然后将值映射到id。一种写法：

Map<String, List<String>> result = Stream.of(new Placement("id1", "f1"), new Placement("id2", "f1"))
        .collect(Collectors.groupingBy(Placement::getFutureLinkId))
        .entrySet()
        .stream()
        .collect(Collectors.toMap(
                Map.Entry::getKey,
                entry -> entry.getValue().stream().map(Placement::getId).collect(Collectors.toList())));
System.out.println(result);

输出：

{f1=[id1, id2]}

为了让它看起来更好看，我建议写两个助手：

public static <K, V, N> Map<K, N> mapValues(Map<K, V> map, Function<V, N> mapper) {
    return map.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, mapper.compose(Map.Entry::getValue)));
}

public static <T, R> List<R> map(List<T> list, Function<T, R> mapper) {
    return list.stream().map(mapper).collect(toList());
}

然后

Map<String, List<String>> result = Stream.of(new Placement("id1", "f1"), new Placement("id2", "f1")).collect(
    collectingAndThen(groupingBy(Placement::getFutureLinkId),
                      res -> mapValues(res, list -> map(list, Placement::getId))));

System.out.println(result);