使用Java8流对不同类的对象进行分组

Question

我有下面的课

class Pojo1 {
    pojo2 item;
}

class pojo2 {
    pojo3 item;
}

class pojo3 {
   pojo4 item;
}

Map<pojo4,Map<pojo3,List<pojo2>>>  k=new HashMap()<>;

   (pojo4 - some property)
   (pojo3 - some property)
   (pojo2- some property)

Answer 1

首先，提到的POJO类POJO4和POJO3需要实现hashcode和equals，以便用作映射键以及属性的相关getter。

转换应该使用公共collectors.Groupingby/collectors.Mapping收集器，并且需要将初始重新映射到两个对象的纯容器，例如Pair。

下面的示例使用Java11map.entry作为中间容器。

static Map<Pojo4, <Map<Pojo3, List<Pojo2>>>> convert(Stream<Pojo1> ones) {
    return ones              // Stream<Pojo1>
        .map(Pojo1::getItem) // Stream<Pojo2>
        .map(p2 -> Map.entry(p2, p2.getItem())) // Stream<Map.Entry<Pojo2, Pojo3>>
        .collect(Collectors.groupingBy(
            me -> me.getValue().getItem(), // key: Pojo4
            Collectors.groupingBy(
                me -> me.getValue(), // key: Pojo3
                Collectors.mapping(
                    Map.Entry::getKey, Collectors.toList()
                )
            )
        ));
}

假设pojo2、pojo3、pojo4有gettergetfield，以便对属性进行排序，则提供以下更新：

static Map<Pojo4, <Map<Pojo3, List<Pojo2>>>> convert(Stream<Pojo1> ones) {
    return ones
        .map(Pojo1::getItem) // Stream<Pojo2>
        .map(p2 -> Map.entry(p2, p2.getItem())) // Stream<Map.Entry<Pojo2, Pojo3>>
        .collect(Collectors.groupingBy(
            me -> me.getValue().getItem(), // key: Pojo4
            () -> new TreeMap(Comparator.comparing(Pojo4::getField)),
            Collectors.groupingBy(
                me -> me.getValue(), // key: Pojo3
                () -> new TreeMap(Comparator.comparing(Pojo3::getField)),
                Collectors.mapping(
                    Map.Entry::getKey, Collectors.collectingAndThen(
                        Collectors.toList(), // List<Pojo2>
                        list -> {
                            list.sort(Comparator.comparing(Pojo2::getField)); 
                            return list;
                        }
                    )
                )
            )
        ));
}

免责声明：上面的代码示例没有经过测试，因此可能需要额外的调整。