当我们要对数据的父子关系进行树状结构整理时,一般我们会想到使用递归进行数据组装,但是这样写相对来说是比较麻烦的了,或者如果你的java架构是hibernate的时候可以直接通过注解实现,好了现在说下stream是如何实现树状结构的
本例子是以 CategoryId 和 pCategoryId 为父子关系
查出父子关系的所有数据 pidList
//以pid为Key进行分组存入Map
Map<Long,List> pidListMap = pidList.stream().collect(Collectors.groupingBy(CategoryListVO::getpCategoryId,LinkedHashMap::new,Collectors.toList()));
pidList.stream().forEach(item->{
item.setChildren(pidListMap.get(item.getCategoryId())==null?new ArrayList<>():pidListMap.get(item.getCategoryId()));
});
//取出顶层节点的对象,本例顶层节点的"PID"为0,注意是PID
// Object obj = JSONArray.toJSON(pidListMap.get(0l));
// String json = obj.toString();
// System.out.println(json);
pidListMap.get(0l)==null?new ArrayList<>():pidListMap.get(0l);