如何把一个集合的数据 根据父级id 分类为Map<Long,Liset<User>>
时间: 2024-03-10 20:43:14 浏览: 199
idistance之大量数据分类
你可以使用Java 8的Stream API来实现这个功能。具体的实现方法如下所示:
```java
public Map<Long, List<User>> groupByParentId(List<User> users) {
return users.stream()
.collect(Collectors.groupingBy(User::getParentId));
}
```
其中,`User`是代表用户的实体类,包含`id`和`parentId`两个属性。`groupByParentId`方法接收一个`List<User>`类型的参数,返回一个`Map<Long, List<User>>`类型的结果。`collect`方法结合`groupingBy`方法可以实现按照`User`对象的`parentId`属性进行分组,并且将分组结果存储到`Map`中返回。
需要注意的是,如果某个`User`对象的`parentId`属性为`null`,那么它会被分到`Map`中的`null`键对应的列表中。如果你希望将这些对象分到其他键对应的列表中,可以使用`groupingBy`方法的第二个参数指定一个默认的键。例如,你可以使用下面的代码将所有`parentId`为`null`的对象分到`-1`键对应的列表中:
```java
public Map<Long, List<User>> groupByParentId(List<User> users) {
return users.stream()
.collect(Collectors.groupingBy(User::getParentId, Collectors.toList(), () -> new HashMap<Long, List<User>>(){{put(-1L, new ArrayList<>());}}));
}
```
这里使用了一个匿名内部类来创建一个只包含`-1`键对应的空列表的`HashMap`对象,并将其作为第三个参数传递给`groupingBy`方法。
阅读全文