java - 如何使用java 8将一组对象分组到排序列表中？

Question

我想获取一组对象（在这种情况下为ObjectInstance），我想按一个属性对它们进行分组，并将结果列表按另一个属性排序。

Set<ObjectInstance> beans = server.queryMBeans(null, null);
Map<String, List<String>> beansByDomain = beans.stream()
            .collect(groupingBy( (ObjectInstance oi) -> oi.getObjectName().getDomain(),
                                mapping((ObjectInstance oi) -> oi.getObjectName().getCanonicalKeyPropertyListString(),
                                toList() )));

上面的表达式创建了正确的数据结构：Map，其中键是ObjectInstance 对象的域，值是属性列表的列表。我想要的是现在对列表进行排序，以确保它们按字母顺序排列。有没有办法在同一个表达式中做到这一点？

一个想法是在.stream() 之后添加.sort()，但这真的可以保证工作吗？

Answer 1

使用collectingAndThen:

List<String> beansByDomain = beans.stream()
        .collect(groupingBy( (ObjectInstance oi) -> oi.getObjectName().getDomain(),
                            mapping((ObjectInstance oi) -> oi.getObjectName().getCanonicalKeyPropertyListString(),
                            collectingAndThen(toList(), (l -> l.stream().sorted().collect(toList()))) )));

您可以提取 Collector 以使代码更具可读性：

public static <T> Collector<T,?,List<T>> toSortedList() {
    return Collectors.collectingAndThen(Collectors.toList(), 
                                       l -> l.stream().sorted().collect(toList()));
}

 List<String> beansByDomain = beans.stream()
        .collect(groupingBy( (ObjectInstance oi) -> oi.getObjectName().getDomain(),
                            mapping((ObjectInstance oi) -> oi.getObjectName().getCanonicalKeyPropertyListString(),
                                    toSortedList())));

Answer 2

当然，您可以在收集之前对整个流进行排序：

Map<String, List<String>> beansByDomain = beans.stream()
        .map(ObjectInstance::getObjectName)
        .sorted(Comparator.comparing(ObjectName::getCanonicalKeyPropertyListString))
        .collect(groupingBy(ObjectName::getDomain,
                            mapping(ObjectName::getCanonicalKeyPropertyListString,
                            toList() )));

请注意，我添加了.map(ObjectInstance::getObjectName) 步骤，因为您不需要来自ObjectInstance 的任何其他内容。这会很好地工作，但我无法预测它是否比单独排序每个结果列表更快。

如果您更喜欢单独的toSortingList() 收集器（如@JeanLogeart 答案），可以这样优化：

public static <T extends Comparable<T>> Collector<T,?,List<T>> toSortedList() {
    return collectingAndThen(toCollection(ArrayList::new),
                    (List<T> l) -> {l.sort(Comparator.naturalOrder()); return l;});
}

这里我们明确收集到ArrayList（toList() 也这样做，但不能保证），然后在不额外复制的情况下对结果列表进行就地排序（使用stream().sorted().collect(toList()) 你复制整个列表内容至少两次)。另请注意，<T> 参数必须声明为extends Comparable<T>。否则你可能会错误地将这个收集器用于不可比较的类型，这会编译得很好，但会导致运行时错误。