SpringBoot业务层开发常见问题

川长思鸟来 2024-02-05 15:02 49阅读 0赞

## **1. 目录层级数据返回问题** ##

获取到一堆数据，诸如这种，需要向前端返回带层级的数据，从数据库获取数据的方式为递归sql：

with recursive t1 as(
    	select * from course_category where id = '1'
    	union ALL
    	select t2.* from course_category t2 join t1 where t1.id = t2.parentid
    )
    select * from t1

![b446f8dd6e7047deb248317c97a6aa17.png][]

需要向前端返回的数据为List<CourseCategoryTreeDto>

CourseCategoryTreeDto：

![98e88cdb8ce34b8ea078a3c615660e7d.png][]

包含当前子节点的列表的父节点信息和一个子节点列表：

对这些数据应该进行如下操作：

@Override
        public List<CourseCategoryTreeDto> queryTreeNodes(String id) {
            List<CourseCategoryTreeDto> courseCategoryTreeDtos = courseCategoryMapper.selectTreeNodes(id);
            //先将list转成map，key是节点的id，value是CourseCategoryTreeDto对象，目的就是方便从map中获取节点,对map进行过滤，不需要根节点1，也即.filter()
            Map<String, CourseCategoryTreeDto> idDtoMap = courseCategoryTreeDtos.stream().filter(item -> !item.getId().equals(id)).collect(Collectors.toMap(CourseCategory::getId, value -> value, (key1, key2) -> key2));
            //new出需要返回的list
            List<CourseCategoryTreeDto> categoryList = new ArrayList<>();
            //从头遍历List<CourseCategoryTreeDto>，一边遍历一边找子节点放在父节点的childrenTreeNodes属性中
            courseCategoryTreeDtos.stream().filter(item -> !item.getId().equals(id)).forEach(item -> {
                if(item.getParentid().equals(id)){
                    categoryList.add(item);
                }
                CourseCategoryTreeDto courseCategoryTreeDto = idDtoMap.get(item.getParentid());
                if(courseCategoryTreeDto!=null){
                    if(courseCategoryTreeDto.getChildrenTreeNodes() == null){
                        courseCategoryTreeDto.setChildrenTreeNodes(new ArrayList<>());
                    }
                    courseCategoryTreeDto.getChildrenTreeNodes().add(item);
                }
    
            });
    
            return categoryList;
        }

使用stream操作，支持多级

[b446f8dd6e7047deb248317c97a6aa17.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/05/e2bfbb3fc77444fc9d9b061c40a2b535.png
[98e88cdb8ce34b8ea078a3c615660e7d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/05/39b822d33c464cd6b0b3477bffb33a41.png