Java 如何通过某个最大值过滤流

发布于03月30日

假设我有一个如下对象流.

class MyObject {
    private Some some; // Comparable
    private Other value
}

有没有一个成语在一条链中做以下事情？

给定Stream<MyObject>，
查找最大值some.
map value，其some为最大值.

犹如，

final List<MyObject> list = getList();
final Some max = list.stream().max(Comparator.naturalOrder())
final List<Other> list = list.stream()
        .filter(e -> Objects.equals(e.some, max)
        .map(e -> e.getValue()).collect(toList());

推荐答案

如果必须一次性完成，可以编写一个自定义收集器，将流减少到max元素列表中.这是一个基于Stuart Marks this answer的实现.

List<MyObject> maxList = list.stream()
                             .collect(maxList(Comparator.comparing(MyObject::getSome)));

static <T> Collector<T,?,List<T>> maxList(Comparator<? super T> comp) {
    return Collector.of(
        ArrayList::new,
        (list, t) -> {
            int c;
            if (list.isEmpty() || (c = comp.compare(t, list.get(0))) == 0) {
                list.add(t);
            } else if (c > 0) {
                list.clear();
                list.add(t);
            }
        },
        (list1, list2) -> {
            if (list1.isEmpty()) {
                return list2;
            } 
            if (list2.isEmpty()) {
                return list1;
            }
            int r = comp.compare(list1.get(0), list2.get(0));
            if (r < 0) {
                return list2;
            } else if (r > 0) {
                return list1;
            } else {
                list1.addAll(list2);
                return list1;
            }
        });
}

收集器将维护一个用于结果的ArrayList，并将每个元素累加到其中，判断该元素与当前列表的第一个元素的比较情况.第c = comp.compare(t, list.get(0))) == 0部分将判断元素是否具有相同的最大值，如果是，则将其添加到列表中.