如何使用Java 8流和自定义列表和地图供应商将List <v>转换为Map <k,list <v =“”>&gt;?

时间:2016-11-23 19:49:20

标签: java java-stream collectors

List<V>转换为Map<K, List<V>>很容易。例如:

public Map<Integer, List<String>> getMap(List<String> strings) {
   return
      strings.stream()
             .collect(Collectors.groupingBy(String::length));
}

但我想用我自己的ListMap 供应商来做这件事。

我想出了这个:

public Map<Integer, List<String>> getMap(List<String> strings) {
   return strings.stream()
       .collect(Collectors.toMap(
             String::length,
             item -> {List<String> list = new ArrayList<>(); list.add(item); return list;},
             (list1, list2) -> {list1.addAll(list2); return list1;},
             HashMap::new));
}

问题:是否有更简单,更简洁,更有效的方法?例如,像这样的东西(不起作用):

return strings.stream()
      .collect(Collectors.toMap(
            String::length,
            ArrayList::new,                    
            HashMap::new));

如果我只需要定义List供应商而不是Map供应商,该怎么办?

3 个答案:

答案 0 :(得分:27)

您可以拥有以下内容:

public Map<Integer, List<String>> getMap(List<String> strings) {
    return strings.stream().collect(
      Collectors.groupingBy(String::length, HashMap::new, Collectors.toCollection(ArrayList::new))
    );
}

收藏家groupingBy(classifier, mapFactory, downstream)可用于指定所需的地图类型,方法是为mapFactory传递所需地图的供应商。然后,用于收集分组到同一密钥的元素的下游收集器是toCollection(collectionFactory),它可以收集到从给定供应商处获得的集合。

这可确保返回的地图为HashMap,并且每个值中的列表为ArrayList。请注意,如果要返回地图和集合的特定实现,那么您很可能希望该方法也返回这些特定类型,因此您可以使用它们的属性。

如果您只想指定一个收集供应商,并保留groupingBy默认地图,您可以在上面的代码中省略供应商并使用two arguments overload

public Map<Integer, List<String>> getMap(List<String> strings) {
    return strings.stream().collect(
      Collectors.groupingBy(String::length, Collectors.toCollection(ArrayList::new))
    );
}

作为旁注,你可以有一个通用的方法:

public <K, V, C extends Collection<V>, M extends Map<K, C>> M getMap(List<V> list,
        Function<? super V, ? extends K> classifier, Supplier<M> mapSupplier, Supplier<C> collectionSupplier) {
    return list.stream().collect(
        Collectors.groupingBy(classifier, mapSupplier, Collectors.toCollection(collectionSupplier))
    );
}

此声明的优势在于,您现在可以使用它来生成HashMap ArrayList个特定结果,或LinkedHashMap LinkedListsHashMap<Integer, ArrayList<String>> m = getMap(Arrays.asList("foo", "bar", "toto"), String::length, HashMap::new, ArrayList::new); LinkedHashMap<Integer, LinkedList<String>> m2 = getMap(Arrays.asList("foo", "bar", "toto"), String::length, LinkedHashMap::new, LinkedList::new); 祝福:

groupingBy

但是,在那时,直接在代码中使用{{1}}可能更简单......

答案 1 :(得分:0)

我有类似的情况。我解决了它:

Map<String, List<Object>> map = stringList.stream().collect(Collectors.toMap(str -> str, str -> populateList(str)));

populateList()是:

private List<Object> populateList(final String str) {
    ...
    ....
    List<Object> list = // dao.get(str);
    return list;
}

答案 2 :(得分:0)

如果您计划创建类似于Map<property_1, List<property_2>>的地图,则可以使用此解决方案:

Map<String, List<String>> ds= requestList.stream().collect(
    Collectors.groupingBy(TagRequest::getProperty_1, HashMap::new, 
    Collectors.mapping(TagRequest::getProperty_2, Collectors.toList()))
);

如果您打算创建类似于Map<property_1, Set<property_2>>的地图,则可以使用:

Map<String, List<String>> ds= requestList.stream().collect(
    Collectors.groupingBy(TagRequest::getProperty_1, HashMap::new, 
    Collectors.mapping(TagRequest::getProperty_2, Collectors.toSet()))
);