java - 将 Java Map> 转换为 Map>

Question

什么是转换的好习惯 Map<Obj1, Collection<Obj2>>至Map<Obj2, Collection<Obj1>> ？

我尝试用 MultiMap 来做这件事.

    Map<Obj1, Collection<Obj2>> originalMap = ...;
    Multimap<Obj1, Obj2> multiMap = ArrayListMultimap.create();
    originalMap.forEach(multiMap::putAll);
    Map<Obj2, Collection<Obj1>> convertedMap = Multimaps.invertFrom(multiMap, ArrayListMultimap.create()).asMap();

有更好的方法吗？

Answer 1

您可以使用 Java 8 Streams 实现相同的目标；

Map<Obj1, List<Obj2>> originalMap = ...

Map<Obj2, List<Obj1>> reversedMap = originalMap.entrySet().stream()
        .flatMap(obj2s -> obj2s.getValue().stream()
                .map(obj2 -> Map.entry(obj2, obj2s.getKey())))
        .collect(
                Collectors.groupingBy(
                        Map.Entry::getKey, 
                        Collectors.mapping(Map.Entry::getValue, Collectors.toList())
                )
        );

fltMap隐蔽Map<Obj1, List<Obj2>>变成像 <Obj2, Obj1> 这样的元组，然后收集按Obj2分组的结果.