groupingBy to Collection Java8

Question

我的方法看起来像这样：

 public static <T, E>  Map<E, List<T>> groupBy(Collection<T> collection, Function<T, E> function){
    return collection.stream().collect(Collectors.groupingBy(function::apply));
}

它的工作原理，但我希望我的方法看起来像这样：

public static <T, E>  Map<E, Collection<T>> groupBy(Collection<T> collection, Function<T, E> function){
    return collection.stream().collect(Collectors.groupingBy(function::apply));
}

然后它不起作用，因为groupingBy返回Map <E，List <T >>，而不是Map <E，Collection <T >>。

我当然可以这样做：

   public static <T, E>  Map<E, Collection<T>> groupBy(Collection<T> collection, Function<T, E> function){
    return (Map<E, Collection<T>>) (Map) collection.stream().collect(Collectors.groupingBy(function::apply));
}

但我不喜欢这个解决方案。 方法声明：

    public static <T, E>  Map<E, Collection<T>> groupBy(Collection<T> collection, Function<T, E> function)

必须保持这种方式（这是任务的一部分）。 那么有什么比铸造更好的解决方案？ 我知道有第二个版本的groupingBy作为第二个参数，它需要收集器，它可以改变groupingBy返回的地图类型。 但是我不知道如何使用它。 有Collectors.toCollection（） ，但我认为它仍将返回Collection的一些实现，而不是Collection本身。

也可以按照以下链接编写我的自定义收集器： http ： //www.nurkiewicz.com/2014/07/introduction-to-writing-custom.html

但我不知道如何将它与Collection一起使用。

我真的很感谢你的帮助，谢谢:)

Answer 1

这样的事情应该有效：

public static <T, E>  Map<E, Collection<T>> groupBy(Collection<T> collection, Function<T, E> function){
    return collection.stream()
                     .collect(Collectors.groupingBy(function,
                                                    Collectors.toCollection(ArrayList::new)));
}

这将为Map每个值生成一个ArrayList实例。 如果您愿意，可以选择其他集合（只需更改ArrayList::new ）。

Answer 2

您可以提供收集器的类型以及Map的类型：

public static <T, E> Map<E, Collection<T>> groupBy(Collection<T> collection, Function<T, E> function){
    return collection.stream().collect(Collectors.groupingBy(function::apply, HashMap::new,Collectors.toCollection(ArrayList::new)));
}