对对象列表进行排序和分组

Question

我有一个程序对象列表如下

Procedure1  01/01/2020
Procedure2  03/01/2020
Procedure3  03/01/2020
Procedure1  04/01/2020
Procedure5  05/01/2020, 02/01/2020
Procedure2  06/01/2020

我的程序类就像

Class Procedure {
    List<Date> procedureDate;
    String procedureName;
}

我想根据以下条件对对象进行排序和分组。

1. 所有程序都应根据程序名称进行分组。
2. 程序必须按程序日期的降序排列。 [日期列表中的第一个元素，即procedureDate.get[0]]
3. 组合在一起的相同程序应按日期降序排列。

最终结果一定是，

Procedure2  06/01/2020
Procedure2  03/01/2020

Procedure5  05/01/2020, 02/01/2020

Procedure1  04/01/2020
Procedure1  01/01/2020

Procedure3  03/01/2020

我能够使用 Comparator 和旧的 java 代码来实现这一点。 是否可以使用 java8 流、收集器和分组来实现相同的目标？

Answer 1

这是一个非常有趣的问题。 解决方案并不像看起来那么容易。 您必须将解决方案分为多个步骤：

1. 根据List<Date>的第一个日期获取每个分组的procedureName的最大值。
2. 根据第一步中创建的Map<String, Date中的最大Date值比较Procedure实例。
3. 如果它们相等，则通过名称区分它们（例如两次Procedure 2 ）。
4. 如果它们仍然相等，则根据它们的实际第一个日期对Procedure实例进行排序。

这是演示： https : //www.jdoodle.com/iembed/v0/Te 。

第1步

List<Procedure> procedures = ...

Map<String, Date> map = procedures.stream().collect(
    Collectors.collectingAndThen(
        Collectors.groupingBy(
            Procedure::getProcedureName,
            Collectors.maxBy(Comparator.comparing(s -> s.getProcedureDate().get(0)))),
    s -> s.entrySet().stream()
        .filter(e -> e.getValue().isPresent())
        .collect(Collectors.toMap(
              Map.Entry::getKey,
              e -> e.getValue().get().getProcedureDate().get(0)))));

..解释说：有一个简单的方式来获得一个Procedure通过分组最大的第一次约会procedureName 。

Map<String, Optional<Procedure>> mapOfOptionalProcedures = procedures.stream()
    .collect(Collectors.groupingBy(
             Procedure::getProcedureName,
             Collectors.maxBy(Comparator.comparing(o -> o.getProcedureDate().get(0)))));

但是，返回的结构有点笨拙（ Map<String, Optional<Procedure>> ），为了使其有用并直接返回Date ，需要额外的下游收集器Collectors::collectingAndThen使用Function作为结果映射器：

Map<String, Date> map = procedures.stream().collect(
    Collectors.collectingAndThen(
        /* grouping part */,
        s -> s.entrySet().stream()
            .filter(e -> e.getValue().isPresent())
            .collect(Collectors.toMap(
                    Map.Entry::getKey,
                    e -> e.getValue().get().getProcedureDate().get(0)))));

...这实际上是第一个片段。

步骤 2、3 和 4

基本上，按每个组的最大日期排序。 然后按名称排序，最后按实际的第一个日期排序。

Collections.sort(
    procedures,
    (l, r) -> {
        int dates = map.get(r.getProcedureName()).compareTo(map.get(l.getProcedureName()));
        if (dates == 0) {
             int names =  l.getProcedureName().compareTo(r.getProcedureName());
             if (names == 0) {
                 return r.getProcedureDate().get(0).compareTo(l.getProcedureDate().get(0));
             } else return names;
        } else return dates;
    }
);

排序结果

根据您的问题使用已弃用的java.util.Date ，排序procedures将具有排序项目，如您预期的片段（我已经覆盖了Procedure::toString方法）

@Override
public String toString() {
     return procedureName + " " + procedureDate;
}

Procedure2 [Mon Jan 06 00:00:00 CET 2020]
Procedure2 [Fri Jan 03 00:00:00 CET 2020]
Procedure5 [Sun Jan 05 00:00:00 CET 2020, Thu Jan 02 00:00:00 CET 2020]
Procedure1 [Sat Jan 04 00:00:00 CET 2020]
Procedure1 [Wed Jan 01 00:00:00 CET 2020]
Procedure3 [Fri Jan 03 00:00:00 CET 2020]

Answer 2

我的想法来自基于 map-reduce 的函数式编程。 你可以看到 groupBy/collect 实际上是一种 reduce 形式，这个问题可以更好地“合并”而不是使用 Stream 的 groupBy 功能。 这是我在纯 Stream 中的实现。

List<Procedure> a = List.of(
    new Procedure(...),
    ...

)


List<Procedure> b = a.stream().map((p)-> {                    // Prepare for reduce by create Map for each object
        Map<String,Procedure> mapP = new HashMap<>();
        mapP.put(p.getProcedureName(),p)
        return mapP
    }).reduce((p,q)->{                                         //Use reduce to merge
        q.entrySet().stream().forEach((qq)-> {
            if (p.containsKey(qq.getKey())) {
                p.get(qq.getKey()).setProcedureDate(
                    new ArrayList<Date>(
                        Stream.concat(
                            p.get(qq.getKey()).getProcedureDate().stream(),
                            qq.getValue().getProcedureDate().stream())
                        .collect(Collectors.toSet()))
                );
            } else {
                p.put(qq.getKey(), qq.getValue());
            }

        })

        return p;
    }).get().values().stream().map(p-> {                          //sort date inside object
            p.setProcedureDate(p.getProcedureDate().stream().sorted().collect(Collectors.toList()))
            return p;
        }
    ).sorted((x,y)->                                         //sort object by the first date

        x.procedureDate.get(0).compareTo(y.procedureDate.get(0))

    ).collect(Collectors.toList());