另一个不同对象内的对象的流分组依据

Question

我有以下三个对象：

public class A {
     private int a;
     private ...
     private B b;
    //getters setters for B
}

public class B {
     private String name;
     private ... 
     private C c;
    //getters setters for C
}

public class C {
     private Long id;
     private ... 
}

我在每个A 和C 的每个B 中都有一个List<A> 和B's 对象。

我想在地图Map<C,List<A>> 中按 C.id 对列表进行分组

如何使用Stream 的groupingBy 方法实现这一点？

Answer 1

Map<Long, List<A>> resultSet = 
       myList.stream()
             .collect(Collectors.groupingBy(e -> e.getB().getC().getId()));

编辑：

由于您希望 Map<C, List<A>> 作为结果集的接收器类型，因此您需要在 C 类中覆盖 equals 和 hashcode，如下所示：

class C {
     public Long getId() {
         return id;
     }

     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;

         C c = (C) o;

         return id != null ? id.equals(c.id) : c.id == null;
     }

     @Override
     public int hashCode() {
         return id != null ? id.hashCode() : 0;
     }

     private Long id;
}

现在，你可以这样做了：

Map<C,List<A>> resultSet = 
     myList.stream()
           .collect(Collectors.groupingBy(e -> e.getB().getC()));

这意味着具有相同C.id 的所有A 对象将位于同一个存储桶（即组）中。